20 #include "GreeterPrivate.h"
22 #include <QFutureInterface>
23 #include <QFutureWatcher>
25 #include <QtConcurrent>
27 #include <security/pam_appl.h>
32 class GreeterImpl :
public QObject
42 typedef QFutureInterface<QString> ResponseFuture;
45 explicit GreeterImpl(Greeter *parent, GreeterPrivate *greeterPrivate)
48 greeterPrivate(greeterPrivate),
51 qRegisterMetaType<QLightDM::GreeterImpl::ResponseFuture>(
"QLightDM::GreeterImpl::ResponseFuture");
53 connect(&futureWatcher, SIGNAL(finished()),
54 this, SLOT(finishPam()));
55 connect(
this, SIGNAL(showMessage(pam_handle *, QString, QLightDM::Greeter::MessageType)),
56 this, SLOT(handleMessage(pam_handle *, QString, QLightDM::Greeter::MessageType)));
58 connect(
this, SIGNAL(showPrompt(pam_handle *, QString, QLightDM::Greeter::PromptType, QLightDM::GreeterImpl::ResponseFuture)),
59 this, SLOT(handlePrompt(pam_handle *, QString, QLightDM::Greeter::PromptType, QLightDM::GreeterImpl::ResponseFuture)));
67 void start(QString username)
72 AppData *appData =
new AppData();
76 pam_conv conversation;
77 conversation.conv = converseWithPam;
78 conversation.appdata_ptr =
static_cast<void*
>(appData);
80 if (pam_start(
"lightdm", username.toUtf8(), &conversation, &pamHandle) == PAM_SUCCESS) {
81 appData->handle = pamHandle;
82 futureWatcher.setFuture(QtConcurrent::run(authenticateWithPam, pamHandle));
85 greeterPrivate->authenticated =
false;
86 Q_EMIT greeter->showMessage(
"Internal error: could not start PAM authentication", QLightDM::Greeter::MessageTypeError);
87 Q_EMIT greeter->authenticationComplete();
91 static int authenticateWithPam(pam_handle* pamHandle)
93 int pamStatus = pam_authenticate(pamHandle, 0);
94 if (pamStatus == PAM_SUCCESS) {
95 pamStatus = pam_acct_mgmt(pamHandle, 0);
97 if (pamStatus == PAM_NEW_AUTHTOK_REQD) {
98 pamStatus = pam_chauthtok(pamHandle, PAM_CHANGE_EXPIRED_AUTHTOK);
100 if (pamStatus == PAM_SUCCESS) {
101 pam_setcred(pamHandle, PAM_REINITIALIZE_CRED);
106 static int converseWithPam(
int num_msg,
const pam_message** msg,
107 pam_response** resp,
void* appdata_ptr)
112 auto* tmp_response =
static_cast<pam_response*
>(calloc(num_msg,
sizeof(pam_response)));
116 AppData *appData =
static_cast<AppData*
>(appdata_ptr);
117 GreeterImpl *impl = appData->impl;
118 pam_handle *handle = appData->handle;
121 QVector<ResponseFuture> responses;
123 for (count = 0; count < num_msg; ++count)
125 switch (msg[count]->msg_style)
127 case PAM_PROMPT_ECHO_ON:
129 QString message(msg[count]->msg);
130 responses.append(ResponseFuture());
131 responses.last().reportStarted();
132 Q_EMIT impl->showPrompt(handle, message, Greeter::PromptTypeQuestion, responses.last());
135 case PAM_PROMPT_ECHO_OFF:
137 QString message(msg[count]->msg);
138 responses.append(ResponseFuture());
139 responses.last().reportStarted();
140 Q_EMIT impl->showPrompt(handle, message, Greeter::PromptTypeSecret, responses.last());
145 QString message(msg[count]->msg);
146 Q_EMIT impl->showMessage(handle, message, Greeter::MessageTypeInfo);
151 QString message(msg[count]->msg);
152 Q_EMIT impl->showMessage(handle, message, Greeter::MessageTypeError);
159 bool raise_error =
false;
161 for (
auto &response : responses)
163 pam_response* resp_item = &tmp_response[i++];
164 resp_item->resp_retcode = 0;
165 resp_item->resp = strdup(response.future().result().toUtf8());
167 if (!resp_item->resp)
178 for (
int i = 0; i < count; ++i)
179 free(tmp_response[i].resp);
186 *resp = tmp_response;
192 bool respond(QString response)
194 if (!futures.isEmpty()) {
195 futures.dequeue().reportFinished(&response);
203 void showMessage(pam_handle *handle, QString text, QLightDM::Greeter::MessageType type);
204 void showPrompt(pam_handle *handle, QString text, QLightDM::Greeter::PromptType type, QLightDM::GreeterImpl::ResponseFuture response);
209 if (pamHandle ==
nullptr) {
213 int pamStatus = futureWatcher.result();
215 pam_end(pamHandle, pamStatus);
218 greeterPrivate->authenticated = (pamStatus == PAM_SUCCESS);
219 Q_EMIT greeter->authenticationComplete();
222 void handleMessage(pam_handle *handle, QString text, QLightDM::Greeter::MessageType type)
224 if (handle != pamHandle)
227 Q_EMIT greeter->showMessage(text, type);
230 void handlePrompt(pam_handle *handle, QString text, QLightDM::Greeter::PromptType type, QLightDM::GreeterImpl::ResponseFuture future)
232 if (handle != pamHandle) {
233 future.reportResult(QString());
234 future.reportFinished();
238 futures.enqueue(future);
239 Q_EMIT greeter->showPrompt(text, type);
246 if (pamHandle !=
nullptr) {
247 pam_handle *handle = pamHandle;
249 while (respond(QString()));
250 pam_end(handle, PAM_CONV_ERR);
255 GreeterPrivate *greeterPrivate;
256 pam_handle* pamHandle;
257 QFutureWatcher<int> futureWatcher;
258 QQueue<ResponseFuture> futures;
261 GreeterPrivate::GreeterPrivate(Greeter* parent)
262 : authenticated(false),
263 authenticationUser(),
264 m_impl(new GreeterImpl(parent, this)),
269 void GreeterPrivate::handleAuthenticate()
271 m_impl->start(authenticationUser);
274 void GreeterPrivate::handleRespond(
const QString &response)
276 m_impl->respond(response);
281 #include "GreeterPrivate.moc"