123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275 |
- /* SLiM - Simple Login Manager
- Copyright (C) 2007 Martin Parm
- This program is free software; you can redistribute it and/or modify
- it under the terms of the GNU General Public License as published by
- the Free Software Foundation; either version 2 of the License, or
- (at your option) any later version.
- */
- #include <string>
- #include <iostream>
- #include "PAM.h"
- namespace PAM {
- Exception::Exception(pam_handle_t* _pam_handle,
- const std::string& _func_name,
- int _errnum):
- errnum(_errnum),
- errstr(pam_strerror(_pam_handle, _errnum)),
- func_name(_func_name)
- {}
- Exception::~Exception(void){}
- Auth_Exception::Auth_Exception(pam_handle_t* _pam_handle,
- const std::string& _func_name,
- int _errnum):
- Exception(_pam_handle, _func_name, _errnum){}
- Cred_Exception::Cred_Exception(pam_handle_t* _pam_handle,
- const std::string& _func_name,
- int _errnum):
- Exception(_pam_handle, _func_name, _errnum){}
- int Authenticator::_end(void){
- int result=pam_end(pam_handle, last_result);
- pam_handle=0;
- return result;
- }
- Authenticator::Authenticator(conversation* conv, void* data):
- pam_handle(0),
- last_result(PAM_SUCCESS)
- {
- pam_conversation.conv=conv;
- pam_conversation.appdata_ptr=data;
- }
- Authenticator::~Authenticator(void){
- if (pam_handle) _end();
- }
- void Authenticator::start(const std::string& service){
- switch((last_result=pam_start(service.c_str(), NULL, &pam_conversation, &pam_handle))){
- default:
- throw Exception(pam_handle, "pam_start()", last_result);
- case PAM_SUCCESS:
- break;
- }
- return;
- }
- void Authenticator::end(void){
- switch((last_result=_end())){
- default:
- throw Exception(pam_handle, "pam_end()", last_result);
- case PAM_SUCCESS:
- break;
- }
- return;
- }
- void Authenticator::set_item(const Authenticator::ItemType item, const void* value){
- switch((last_result=pam_set_item(pam_handle, item, value))){
- default:
- _end();
- throw Exception(pam_handle, "pam_set_item()", last_result);
- case PAM_SUCCESS:
- break;
- }
- return;
- }
- const void* Authenticator::get_item(const Authenticator::ItemType item){
- const void* data;
- switch ((last_result=pam_get_item(pam_handle, item, &data))){
- default:
- case PAM_SYSTEM_ERR:
- #ifdef __LIBPAM_VERSION
- case PAM_BAD_ITEM:
- #endif
- _end();
- throw Exception(pam_handle, "pam_get_item()", last_result);
- case PAM_PERM_DENIED: // The value of item was NULL
- case PAM_SUCCESS:
- break;
- }
- return data;
- }
- #ifdef __LIBPAM_VERSION
- void Authenticator::fail_delay(const unsigned int micro_sec){
- switch((last_result=pam_fail_delay(pam_handle, micro_sec))){
- default:
- _end();
- throw Exception(pam_handle, "fail_delay()", last_result);
- case PAM_SUCCESS:
- break;
- }
- return;
- }
- #endif
- void Authenticator::authenticate(void){
- switch((last_result=pam_authenticate(pam_handle, 0))){
- default:
- case PAM_ABORT:
- case PAM_AUTHINFO_UNAVAIL:
- _end();
- throw Exception(pam_handle, "pam_authenticate()", last_result);
- case PAM_USER_UNKNOWN:
- case PAM_MAXTRIES:
- case PAM_CRED_INSUFFICIENT:
- case PAM_AUTH_ERR:
- throw Auth_Exception(pam_handle, "pam_authentication()", last_result);
- case PAM_SUCCESS:
- break;
- }
- switch((last_result=pam_acct_mgmt(pam_handle, PAM_SILENT))){
- // The documentation and implementation of Linux PAM differs:
- // PAM_NEW_AUTHTOKEN_REQD is described in the documentation but
- // don't exists in the actual implementation. This issue needs
- // to be fixes at some point.
- default:
- //case PAM_NEW_AUTHTOKEN_REQD:
- case PAM_ACCT_EXPIRED:
- case PAM_USER_UNKNOWN:
- _end();
- throw Exception(pam_handle, "pam_acct_mgmt()", last_result);
-
- case PAM_AUTH_ERR:
- case PAM_PERM_DENIED:
- throw Auth_Exception(pam_handle, "pam_acct_mgmt()", last_result);
- case PAM_SUCCESS:
- break;
- };
- return;
- }
- void Authenticator::open_session(void){
- switch((last_result=pam_setcred(pam_handle, PAM_ESTABLISH_CRED))){
- default:
- case PAM_CRED_ERR:
- case PAM_CRED_UNAVAIL:
- _end();
- throw Exception(pam_handle, "pam_setcred()", last_result);
- case PAM_CRED_EXPIRED:
- case PAM_USER_UNKNOWN:
- throw Cred_Exception(pam_handle, "pam_setcred()", last_result);
- case PAM_SUCCESS:
- break;
- }
- switch((last_result=pam_open_session(pam_handle, 0))){
- // The documentation and implementation of Linux PAM differs:
- // PAM_SESSION_ERROR is described in the documentation but
- // don't exists in the actual implementation. This issue needs
- // to be fixes at some point.
- default:
- //case PAM_SESSION_ERROR:
- pam_setcred(pam_handle, PAM_DELETE_CRED);
- _end();
- throw Exception(pam_handle, "pam_open_session()", last_result);
- case PAM_SUCCESS:
- break;
- };
- return;
- }
- void Authenticator::close_session(void){
- switch((last_result=pam_close_session(pam_handle, 0))){
- // The documentation and implementation of Linux PAM differs:
- // PAM_SESSION_ERROR is described in the documentation but
- // don't exists in the actual implementation. This issue needs
- // to be fixes at some point.
- default:
- //case PAM_SESSION_ERROR:
- pam_setcred(pam_handle, PAM_DELETE_CRED);
- _end();
- throw Exception(pam_handle, "pam_close_session", last_result);
- case PAM_SUCCESS:
- break;
- };
- switch((last_result=pam_setcred(pam_handle, PAM_DELETE_CRED))){
- default:
- case PAM_CRED_ERR:
- case PAM_CRED_UNAVAIL:
- case PAM_CRED_EXPIRED:
- case PAM_USER_UNKNOWN:
- _end();
- throw Exception(pam_handle, "pam_setcred()", last_result);
- case PAM_SUCCESS:
- break;
- }
- return;
- }
- void Authenticator::setenv(const std::string& key, const std::string& value){
- std::string name_value = key+"="+value;
- switch((last_result=pam_putenv(pam_handle, name_value.c_str()))){
- default:
- case PAM_PERM_DENIED:
- case PAM_ABORT:
- case PAM_BUF_ERR:
- #ifdef __LIBPAM_VERSION
- case PAM_BAD_ITEM:
- #endif
- _end();
- throw Exception(pam_handle, "pam_putenv()", last_result);
- case PAM_SUCCESS:
- break;
- };
- return;
- }
- void Authenticator::delenv(const std::string& key){
- switch((last_result=pam_putenv(pam_handle, key.c_str()))){
- default:
- case PAM_PERM_DENIED:
- case PAM_ABORT:
- case PAM_BUF_ERR:
- #ifdef __LIBPAM_VERSION
- case PAM_BAD_ITEM:
- #endif
- _end();
- throw Exception(pam_handle, "pam_putenv()", last_result);
- case PAM_SUCCESS:
- break;
- };
- return;
- }
- const char* Authenticator::getenv(const std::string& key){
- return pam_getenv(pam_handle, key.c_str());
- }
- char** Authenticator::getenvlist(void){
- return pam_getenvlist(pam_handle);
- }
- };
- std::ostream& operator<<( std::ostream& os, const PAM::Exception& e){
- os << e.func_name << ": " << e.errstr;
- return os;
- }
|