libopenbsd/auth_userokay.c

   1 /*
   2  * Copyright (c) 2015 Nathan Holstein <nathan.holstein@gmail.com>
   3  *
   4  * Permission to use, copy, modify, and distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  *
   8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15  */
  16
  17 #include <sys/types.h>
  18 #include <err.h>
  19 #include <errno.h>
  20 #include <pwd.h>
  21 #include <readpassphrase.h>
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #include <string.h>
  25
  26 #include <security/pam_appl.h>
  27
  28 #include "openbsd.h"
  29
  30 #define PAM_SERVICE "sudo"
  31
  32 #define __UNUSED __attribute__ ((unused))
  33
  34 static char *
  35 pam_prompt(const char *msg, int echo_on)
  36 {
  37         char buf[PAM_MAX_RESP_SIZE];
  38         int flags = RPP_REQUIRE_TTY | (echo_on ? RPP_ECHO_ON : RPP_ECHO_OFF);
  39         char *ret = readpassphrase(msg, buf, sizeof(buf), flags);
  40         if (ret)
  41                 ret = strdup(ret);
  42         explicit_bzero(buf, sizeof(buf));
  43         return ret;
  44 }
  45
  46 static int
  47 pam_conv(int nmsgs, const struct pam_message **msgs,
  48                 struct pam_response **rsps, __UNUSED void *ptr)
  49 {
  50         int i, style;
  51         struct pam_response *rsp;
  52
  53         if (!(rsp = calloc(nmsgs, sizeof(struct pam_response))))
  54                 errx(1, "couldn't malloc pam_response");
  55         *rsps = rsp;
  56
  57         for (i = 0; i < nmsgs; i++) {
  58                 switch (style = msgs[i]->msg_style) {
  59                 case PAM_PROMPT_ECHO_OFF:
  60                 case PAM_PROMPT_ECHO_ON:
  61                         rsp[i].resp = pam_prompt(msgs[i]->msg,
  62                                         style == PAM_PROMPT_ECHO_ON);
  63                         break;
  64
  65                 case PAM_ERROR_MSG:
  66                 case PAM_TEXT_INFO:
  67                         fprintf(style == PAM_ERROR_MSG ? stderr : stdout,
  68                                         "%s\n", msgs[i]->msg);
  69                         break;
  70
  71                 default:
  72                         errx(1, "invalid PAM msg_style %d", style);
  73                 }
  74         }
  75
  76         return PAM_SUCCESS;
  77 }
  78
  79 int
  80 auth_userokay(char *name, char *style, char *type, char *password)
  81 {
  82         static const struct pam_conv conv = {
  83                 .conv = pam_conv,
  84                 .appdata_ptr = NULL,
  85         };
  86
  87         int ret, auth;
  88         pam_handle_t *pamh = NULL;
  89
  90         if (!name)
  91                 return 0;
  92         if (style || type || password)
  93                 errx(1, "auth_userokay(name, NULL, NULL, NULL)!\n");
  94
  95         ret = pam_start(PAM_SERVICE, name, &conv, &pamh);
  96         if (ret != PAM_SUCCESS)
  97                 errx(1, "pam_start(\"%s\", \"%s\", ?, ?): failed\n",
  98                                 PAM_SERVICE, name);
  99
 100         auth = pam_authenticate(pamh, 0);
 101
 102         ret = pam_close_session(pamh, 0);
 103         if (ret != PAM_SUCCESS)
 104                 errx(1, "pam_close_session(): %s\n", pam_strerror(pamh, ret));
 105
 106         return auth == PAM_SUCCESS;
 107 }
 108