cifs.upcall: clean up flag handling
[kai/samba.git] / client / cifs.upcall.c
index da6b1b8d430ee1ad22725a8c22d4f9010165e5d8..904ec8e1c1c35ad8d67456b0ca2b6fe53e2d6ae4 100644 (file)
@@ -30,13 +30,13 @@ create dns_resolver * * /usr/local/sbin/cifs.upcall %k
 
 #include "cifs_spnego.h"
 
-const char *CIFSSPNEGO_VERSION = "1.2";
+const char *CIFSSPNEGO_VERSION = "1.3";
 static const char *prog = "cifs.upcall";
-typedef enum _secType {
+typedef enum _sectype {
        NONE = 0,
        KRB5,
        MS_KRB5
-} secType_t;
+} sectype_t;
 
 /*
  * given a process ID, get the value of the KRB5CCNAME environment variable
@@ -113,7 +113,7 @@ get_krb5_ccname(pid_t pid)
  *     sess_key-       pointer for SessionKey data to be stored
  *
  * ret: 0 - success, others - failure
-*/
+ */
 static int
 handle_krb5_mech(const char *oid, const char *principal, DATA_BLOB *secblob,
                 DATA_BLOB *sess_key, const char *ccname)
@@ -156,9 +156,16 @@ handle_krb5_mech(const char *oid, const char *principal, DATA_BLOB *secblob,
 #define DKD_HAVE_PID           0x40
 #define DKD_MUSTHAVE_SET (DKD_HAVE_HOSTNAME|DKD_HAVE_VERSION|DKD_HAVE_SEC)
 
-static int
-decode_key_description(const char *desc, int *ver, secType_t *sec,
-                          char **hostname, uid_t *uid, pid_t *pid)
+static struct decoded_args {
+       int             ver;
+       char            *hostname;
+       uid_t           uid;
+       pid_t           pid;
+       sectype_t       sec;
+};
+
+static unsigned int
+decode_key_description(const char *desc, struct decoded_args *arg)
 {
        int retval = 0;
        char *pos;
@@ -169,15 +176,15 @@ decode_key_description(const char *desc, int *ver, secType_t *sec,
                if (strncmp(tkn, "host=", 5) == 0) {
                        int len;
 
-                       if (pos == NULL) {
+                       if (pos == NULL)
                                len = strlen(tkn);
-                       } else {
+                       else
                                len = pos - tkn;
-                       }
+
                        len -= 4;
-                       SAFE_FREE(*hostname);
-                       *hostname = SMB_XMALLOC_ARRAY(char, len);
-                       strlcpy(*hostname, tkn + 5, len);
+                       SAFE_FREE(arg->hostname);
+                       arg->hostname = SMB_XMALLOC_ARRAY(char, len);
+                       strlcpy(arg->hostname, tkn + 5, len);
                        retval |= DKD_HAVE_HOSTNAME;
                } else if (strncmp(tkn, "ipv4=", 5) == 0) {
                        /* BB: do we need it if we have hostname already? */
@@ -185,7 +192,7 @@ decode_key_description(const char *desc, int *ver, secType_t *sec,
                        /* BB: do we need it if we have hostname already? */
                } else if (strncmp(tkn, "pid=", 4) == 0) {
                        errno = 0;
-                       *pid = strtol(tkn + 4, NULL, 0);
+                       arg->pid = strtol(tkn + 4, NULL, 0);
                        if (errno != 0) {
                                syslog(LOG_ERR, "Invalid pid format: %s",
                                       strerror(errno));
@@ -196,14 +203,14 @@ decode_key_description(const char *desc, int *ver, secType_t *sec,
                } else if (strncmp(tkn, "sec=", 4) == 0) {
                        if (strncmp(tkn + 4, "krb5", 4) == 0) {
                                retval |= DKD_HAVE_SEC;
-                               *sec = KRB5;
+                               arg->sec = KRB5;
                        } else if (strncmp(tkn + 4, "mskrb5", 6) == 0) {
                                retval |= DKD_HAVE_SEC;
-                               *sec = MS_KRB5;
+                               arg->sec = MS_KRB5;
                        }
                } else if (strncmp(tkn, "uid=", 4) == 0) {
                        errno = 0;
-                       *uid = strtol(tkn + 4, NULL, 16);
+                       arg->uid = strtol(tkn + 4, NULL, 16);
                        if (errno != 0) {
                                syslog(LOG_ERR, "Invalid uid format: %s",
                                       strerror(errno));
@@ -213,7 +220,7 @@ decode_key_description(const char *desc, int *ver, secType_t *sec,
                        }
                } else if (strncmp(tkn, "ver=", 4) == 0) {      /* if version */
                        errno = 0;
-                       *ver = strtol(tkn + 4, NULL, 16);
+                       arg->ver = strtol(tkn + 4, NULL, 16);
                        if (errno != 0) {
                                syslog(LOG_ERR, "Invalid version format: %s",
                                       strerror(errno));
@@ -257,11 +264,11 @@ cifs_resolver(const key_serial_t key, const char *key_descr)
        }
 
        /* conver ip to string form */
-       if (addr->ai_family == AF_INET) {
+       if (addr->ai_family == AF_INET)
                p = &(((struct sockaddr_in *)addr->ai_addr)->sin_addr);
-       } else {
+       else
                p = &(((struct sockaddr_in6 *)addr->ai_addr)->sin6_addr);
-       }
+
        if (!inet_ntop(addr->ai_family, p, ip, sizeof(ip))) {
                syslog(LOG_ERR, "%s: inet_ntop: %s", __func__, strerror(errno));
                freeaddrinfo(addr);
@@ -284,8 +291,8 @@ cifs_resolver(const key_serial_t key, const char *key_descr)
 static void
 usage(void)
 {
-       syslog(LOG_INFO, "Usage: %s [-c] [-v] key_serial", prog);
-       fprintf(stderr, "Usage: %s [-c] [-v] key_serial\n", prog);
+       syslog(LOG_INFO, "Usage: %s [-v] key_serial", prog);
+       fprintf(stderr, "Usage: %s [-v] key_serial\n", prog);
 }
 
 int main(const int argc, char *const argv[])
@@ -293,33 +300,28 @@ int main(const int argc, char *const argv[])
        struct cifs_spnego_msg *keydata = NULL;
        DATA_BLOB secblob = data_blob_null;
        DATA_BLOB sess_key = data_blob_null;
-       secType_t sectype = NONE;
        key_serial_t key = 0;
        size_t datalen;
+       unsigned int have;
        long rc = 1;
-       uid_t uid = 0;
-       pid_t pid = 0;
-       int kernel_upcall_version = 0;
-       int c, use_cifs_service_prefix = 0;
-       char *buf, *ccname = NULL, *hostname = NULL;
+       int c;
+       char *buf, *princ, *ccname = NULL;
+       struct decoded_args arg = { };
        const char *oid;
 
        openlog(prog, 0, LOG_DAEMON);
 
        while ((c = getopt(argc, argv, "cv")) != -1) {
                switch (c) {
-               case 'c':{
-                       use_cifs_service_prefix = 1;
+               case 'c':
+                       /* legacy option -- skip it */
                        break;
-                       }
-               case 'v':{
+               case 'v':
                        printf("version: %s\n", CIFSSPNEGO_VERSION);
                        goto out;
-                       }
-               default:{
+               default:
                        syslog(LOG_ERR, "unknown option: %c", c);
                        goto out;
-                       }
                }
        }
 
@@ -354,77 +356,72 @@ int main(const int argc, char *const argv[])
                goto out;
        }
 
-       rc = decode_key_description(buf, &kernel_upcall_version, &sectype,
-                                   &hostname, &uid, &pid);
-       if ((rc & DKD_MUSTHAVE_SET) != DKD_MUSTHAVE_SET) {
+       have = decode_key_description(buf, &arg);
+       SAFE_FREE(buf);
+       if ((have & DKD_MUSTHAVE_SET) != DKD_MUSTHAVE_SET) {
                syslog(LOG_ERR, "unable to get necessary params from key "
-                               "description (0x%x)", rc);
+                               "description (0x%x)", have);
                rc = 1;
-               SAFE_FREE(buf);
                goto out;
        }
-       SAFE_FREE(buf);
 
-       if (kernel_upcall_version > CIFS_SPNEGO_UPCALL_VERSION) {
+       if (arg.ver > CIFS_SPNEGO_UPCALL_VERSION) {
                syslog(LOG_ERR, "incompatible kernel upcall version: 0x%x",
-                               kernel_upcall_version);
+                               arg.ver);
                rc = 1;
                goto out;
        }
 
-       if (rc & DKD_HAVE_PID)
-               ccname = get_krb5_ccname(pid);
-
-       if (rc & DKD_HAVE_UID) {
-               rc = setuid(uid);
+       if (have & DKD_HAVE_UID) {
+               rc = setuid(arg.uid);
                if (rc == -1) {
                        syslog(LOG_ERR, "setuid: %s", strerror(errno));
                        goto out;
                }
        }
 
+       if (have & DKD_HAVE_PID)
+               ccname = get_krb5_ccname(arg.pid);
+
        // do mech specific authorization
-       switch (sectype) {
+       switch (arg.sec) {
        case MS_KRB5:
-       case KRB5:{
-                       char *princ;
-                       size_t len;
-
-                       /* for "cifs/" service name + terminating 0 */
-                       len = strlen(hostname) + 5 + 1;
-                       princ = SMB_XMALLOC_ARRAY(char, len);
-                       if (!princ) {
-                               rc = 1;
-                               break;
-                       }
-                       if (use_cifs_service_prefix) {
-                               strlcpy(princ, "cifs/", len);
-                       } else {
-                               strlcpy(princ, "host/", len);
-                       }
-                       strlcpy(princ + 5, hostname, len - 5);
-
-                       if (sectype == MS_KRB5)
-                               oid = OID_KERBEROS5_OLD;
-                       else
-                               oid = OID_KERBEROS5;
-
-                       rc = handle_krb5_mech(oid, princ, &secblob, &sess_key,
-                                             ccname);
-                       SAFE_FREE(princ);
-                       break;
-               }
-       default:{
-                       syslog(LOG_ERR, "sectype: %d is not implemented",
-                              sectype);
+       case KRB5:
+               /* for "cifs/" service name + terminating 0 */
+               datalen = strlen(arg.hostname) + 5 + 1;
+               princ = SMB_XMALLOC_ARRAY(char, datalen);
+               if (!princ) {
                        rc = 1;
                        break;
                }
+
+               if (arg.sec == MS_KRB5)
+                       oid = OID_KERBEROS5_OLD;
+               else
+                       oid = OID_KERBEROS5;
+
+               /*
+                * try getting a cifs/ principal first and then fall back to
+                * getting a host/ principal if that doesn't work.
+                */
+               strlcpy(princ, "cifs/", datalen);
+               strlcpy(princ + 5, arg.hostname, datalen - 5);
+               rc = handle_krb5_mech(oid, princ, &secblob, &sess_key, ccname);
+               if (rc) {
+                       memcpy(princ, "host/", 5);
+                       rc = handle_krb5_mech(oid, princ, &secblob, &sess_key,
+                                               ccname);
+               }
+               SAFE_FREE(princ);
+               break;
+       default:
+               syslog(LOG_ERR, "sectype: %d is not implemented", arg.sec);
+               rc = 1;
+               break;
        }
 
-       if (rc) {
+       if (rc)
                goto out;
-       }
 
        /* pack SecurityBLob and SessionKey into downcall packet */
        datalen =
@@ -434,7 +431,7 @@ int main(const int argc, char *const argv[])
                rc = 1;
                goto out;
        }
-       keydata->version = kernel_upcall_version;
+       keydata->version = arg.ver;
        keydata->flags = 0;
        keydata->sesskey_len = sess_key.length;
        keydata->secblob_len = secblob.length;
@@ -463,7 +460,7 @@ out:
        data_blob_free(&secblob);
        data_blob_free(&sess_key);
        SAFE_FREE(ccname);
-       SAFE_FREE(hostname);
+       SAFE_FREE(arg.hostname);
        SAFE_FREE(keydata);
        return rc;
 }