cifs.upcall: switch to getopt_long
[samba.git] / client / cifs.upcall.c
index 926ec2064d7794953e8fbc8dced8d41d7e67e45c..f06d563959b9806690cac7b76e631030a54313a6 100644 (file)
@@ -27,16 +27,17 @@ create dns_resolver * * /usr/local/sbin/cifs.upcall %k
 
 #include "includes.h"
 #include <keyutils.h>
+#include <getopt.h>
 
 #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
@@ -150,16 +151,24 @@ handle_krb5_mech(const char *oid, const char *principal, DATA_BLOB *secblob,
 #define DKD_HAVE_HOSTNAME      0x1
 #define DKD_HAVE_VERSION       0x2
 #define DKD_HAVE_SEC           0x4
-#define DKD_HAVE_IPV4          0x8
-#define DKD_HAVE_IPV6          0x10
-#define DKD_HAVE_UID           0x20
-#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)
+#define DKD_HAVE_IP            0x8
+#define DKD_HAVE_UID           0x10
+#define DKD_HAVE_PID           0x20
+#define DKD_MUSTHAVE_SET (DKD_HAVE_IP|DKD_HAVE_VERSION|DKD_HAVE_SEC)
+
+static struct decoded_args {
+       int             ver;
+       char            *hostname;
+       char            *ip;
+       uid_t           uid;
+       pid_t           pid;
+       sectype_t       sec;
+};
+
+static unsigned int
+decode_key_description(const char *desc, struct decoded_args *arg)
 {
+       int len;
        int retval = 0;
        char *pos;
        const char *tkn = desc;
@@ -167,7 +176,6 @@ decode_key_description(const char *desc, int *ver, secType_t *sec,
        do {
                pos = index(tkn, ';');
                if (strncmp(tkn, "host=", 5) == 0) {
-                       int len;
 
                        if (pos == NULL)
                                len = strlen(tkn);
@@ -175,17 +183,25 @@ decode_key_description(const char *desc, int *ver, secType_t *sec,
                                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? */
-               } else if (strncmp(tkn, "ipv6=", 5) == 0) {
-                       /* BB: do we need it if we have hostname already? */
+               } else if (!strncmp(tkn, "ip4=", 4) ||
+                          !strncmp(tkn, "ip6=", 4)) {
+                       if (pos == NULL)
+                               len = strlen(tkn);
+                       else
+                               len = pos - tkn;
+
+                       len -= 3;
+                       SAFE_FREE(arg->ip);
+                       arg->ip = SMB_XMALLOC_ARRAY(char, len);
+                       strlcpy(arg->ip, tkn + 4, len);
+                       retval |= DKD_HAVE_IP;
                } 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 +212,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 +229,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));
@@ -281,35 +297,93 @@ cifs_resolver(const key_serial_t key, const char *key_descr)
        return 0;
 }
 
+/*
+ * Older kernels sent IPv6 addresses without colons. Well, at least
+ * they're fixed-length strings. Convert these addresses to have colon
+ * delimiters to make getaddrinfo happy.
+ */
+static void
+convert_inet6_addr(const char *from, char *to)
+{
+       int i = 1;
+
+       while (*from) {
+               *to++ = *from++;
+               if (!(i++ % 4) && *from)
+                       *to++ = ':';
+       }
+       *to = 0;
+}
+
+static int
+ip_to_fqdn(const char *addrstr, char *host, size_t hostlen)
+{
+       int rc;
+       struct addrinfo hints = { .ai_flags = AI_NUMERICHOST };
+       struct addrinfo *res;
+       const char *ipaddr = addrstr;
+       char converted[INET6_ADDRSTRLEN + 1];
+
+       if ((strlen(ipaddr) > INET_ADDRSTRLEN) && !strchr(ipaddr, ':')) {
+               convert_inet6_addr(ipaddr, converted);
+               ipaddr = converted;
+       }
+
+       rc = getaddrinfo(ipaddr, NULL, &hints, &res);
+       if (rc) {
+               syslog(LOG_DEBUG, "%s: failed to resolve %s to "
+                       "ipaddr: %s", __func__, ipaddr,
+               rc == EAI_SYSTEM ? strerror(errno) : gai_strerror(rc));
+               return rc;
+       }
+
+       rc = getnameinfo(res->ai_addr, res->ai_addrlen, host, hostlen,
+                        NULL, 0, NI_NAMEREQD);
+       freeaddrinfo(res);
+       if (rc) {
+               syslog(LOG_DEBUG, "%s: failed to resolve %s to fqdn: %s",
+                       __func__, ipaddr,
+                       rc == EAI_SYSTEM ? strerror(errno) : gai_strerror(rc));
+               return rc;
+       }
+
+       syslog(LOG_DEBUG, "%s: resolved %s to %s", __func__, ipaddr, host);
+       return 0;
+}
+
 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);
 }
 
+const struct option long_options[] = {
+       { "version",    0, NULL, 'v' },
+       { NULL,         0, NULL, 0 }
+};
+
 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, *princ, *ccname = NULL, *hostname = NULL;
+       int c;
+       char *buf, *princ, *ccname = NULL;
+       char hostbuf[NI_MAXHOST];
+       struct decoded_args arg = { };
        const char *oid;
 
        openlog(prog, 0, LOG_DAEMON);
 
-       while ((c = getopt(argc, argv, "cv")) != -1) {
+       while ((c = getopt_long(argc, argv, "cv", long_options, NULL)) != -1) {
                switch (c) {
                case 'c':
-                       use_cifs_service_prefix = 1;
+                       /* legacy option -- skip it */
                        break;
                case 'v':
                        printf("version: %s\n", CIFSSPNEGO_VERSION);
@@ -351,64 +425,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);
+
+       if (have & DKD_HAVE_IP) {
+               rc = ip_to_fqdn(arg.ip, hostbuf, sizeof(hostbuf));
+               if (rc)
+                       goto out;
+       }
+
        // do mech specific authorization
-       switch (sectype) {
+       switch (arg.sec) {
        case MS_KRB5:
        case KRB5:
                /* for "cifs/" service name + terminating 0 */
-               datalen = strlen(hostname) + 5 + 1;
+               datalen = strnlen(hostbuf, sizeof(hostbuf)) + 5 + 1;
                princ = SMB_XMALLOC_ARRAY(char, datalen);
                if (!princ) {
                        rc = 1;
                        break;
                }
 
-               if (use_cifs_service_prefix)
-                       strlcpy(princ, "cifs/", datalen);
-               else
-                       strlcpy(princ, "host/", datalen);
-
-               strlcpy(princ + 5, hostname, datalen - 5);
-
-               if (sectype == MS_KRB5)
+               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, hostbuf, 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", sectype);
+               syslog(LOG_ERR, "sectype: %d is not implemented", arg.sec);
                rc = 1;
                break;
        }
@@ -424,7 +506,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;
@@ -453,7 +535,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;
 }