]> git.ipfire.org Git - thirdparty/squid.git/blobdiff - helpers/negotiate_auth/kerberos/negotiate_kerberos_auth_test.cc
Boilerplate: update copyright blurbs on Squid helpers
[thirdparty/squid.git] / helpers / negotiate_auth / kerberos / negotiate_kerberos_auth_test.cc
index bc74bc6cfc6c3ba2015d6c66ab0828da27ce1003..455edf71b15fbd02af9da0d16ec2373d35236aa2 100644 (file)
@@ -1,3 +1,11 @@
+/*
+ * Copyright (C) 1996-2014 The Squid Software Foundation and contributors
+ *
+ * Squid software is distributed under GPLv2+ license and includes
+ * contributions from numerous individuals and organizations.
+ * Please see the COPYING and CONTRIBUTORS files for details.
+ */
+
 /*
  * -----------------------------------------------------------------------------
  *
  *
  * -----------------------------------------------------------------------------
  */
-/*
- * Hosted at http://sourceforge.net/projects/squidkerbauth
- */
 
-#include "config.h"
+#include "squid.h"
 
 #if HAVE_GSSAPI
 
-#if HAVE_STRING_H
-#include <string.h>
-#endif
-#if HAVE_STDIO_H
-#include <stdio.h>
-#endif
-#if HAVE_STDLIB_H
-#include <stdlib.h>
-#endif
+#include <cerrno>
+#include <cstring>
+#include <ctime>
 #if HAVE_NETDB_H
 #include <netdb.h>
 #endif
 #if HAVE_UNISTD_H
 #include <unistd.h>
 #endif
-#if HAVE_TIME_H
-#include <time.h>
-#endif
-#if HAVE_SYS_TIME_H
-#include <sys/time.h>
-#endif
-#if HAVE_ERRNO_H
-#include <errno.h>
-#endif
 
 #include "base64.h"
 #include "util.h"
 
-#if HAVE_HEIMDAL_KERBEROS
+#if USE_HEIMDAL_KRB5
 #if HAVE_GSSAPI_GSSAPI_H
 #include <gssapi/gssapi.h>
 #elif HAVE_GSSAPI_H
 #include <gssapi.h>
-#endif /* HAVE_GSSAPI_GSSAPI_H */
-#define gss_nt_service_name GSS_C_NT_HOSTBASED_SERVICE
-#else /* HAVE_HEIMDAL_KERBEROS */
+#endif
+#elif USE_GNUGSS
+#if HAVE_GSS_H
+#include <gss.h>
+#endif
+#else
 #if HAVE_GSSAPI_GSSAPI_H
 #include <gssapi/gssapi.h>
 #elif HAVE_GSSAPI_H
 #include <gssapi.h>
-#endif /* HAVE_GSSAPI_GSSAPI_H */
+#endif
 #if HAVE_GSSAPI_GSSAPI_KRB5_H
 #include <gssapi/gssapi_krb5.h>
-#endif /* HAVE_GSSAPI_GSSAPI_KRB5_H */
+#endif
 #if HAVE_GSSAPI_GSSAPI_GENERIC_H
 #include <gssapi/gssapi_generic.h>
-#endif /* HAVE_GSSAPI_GSSAPI_GENERIC_H */
-#endif /* HAVE_HEIMDAL_KERBEROS */
+#endif
+#if HAVE_GSSAPI_GSSAPI_EXT_H
+#include <gssapi/gssapi_ext.h>
+#endif
+#endif
+
+#ifndef gss_nt_service_name
+#define gss_nt_service_name GSS_C_NT_HOSTBASED_SERVICE
+#endif
 
 static const char *LogTime(void);
 
 int check_gss_err(OM_uint32 major_status, OM_uint32 minor_status,
-    const char *function);
+                  const char *function);
 
 const char *squid_kerb_proxy_auth(char *proxy);
 
@@ -97,70 +97,65 @@ LogTime()
 
     gettimeofday(&now, NULL);
     if (now.tv_sec != last_t) {
-       tm = localtime(&now.tv_sec);
-       strftime(buf, 127, "%Y/%m/%d %H:%M:%S", tm);
-       last_t = now.tv_sec;
+        tm = localtime((const time_t *) &now.tv_sec);
+        strftime(buf, 127, "%Y/%m/%d %H:%M:%S", tm);
+        last_t = now.tv_sec;
     }
     return buf;
 }
 
-#ifdef HAVE_SPNEGO
 #ifndef gss_mech_spnego
-static gss_OID_desc _gss_mech_spnego =
-    { 6, (void *) "\x2b\x06\x01\x05\x05\x02" };
+static gss_OID_desc _gss_mech_spnego = {6, (void *) "\x2b\x06\x01\x05\x05\x02"};
 gss_OID gss_mech_spnego = &_gss_mech_spnego;
 #endif
-#endif
 
 int
 check_gss_err(OM_uint32 major_status, OM_uint32 minor_status,
-    const char *function)
+              const char *function)
 {
     if (GSS_ERROR(major_status)) {
-       OM_uint32 maj_stat, min_stat;
-       OM_uint32 msg_ctx = 0;
-       gss_buffer_desc status_string;
-       char buf[1024];
-       size_t len;
-
-       len = 0;
-       msg_ctx = 0;
-       while (!msg_ctx) {
-           /* convert major status code (GSS-API error) to text */
-           maj_stat = gss_display_status(&min_stat, major_status,
-               GSS_C_GSS_CODE, GSS_C_NULL_OID, &msg_ctx, &status_string);
-           if (maj_stat == GSS_S_COMPLETE) {
-               if (sizeof(buf) > len + status_string.length + 1) {
-                   sprintf(buf + len, "%s", (char *) status_string.value);
-                   len += status_string.length;
-               }
-               gss_release_buffer(&min_stat, &status_string);
-               break;
-           }
-           gss_release_buffer(&min_stat, &status_string);
-       }
-       if (sizeof(buf) > len + 2) {
-           sprintf(buf + len, "%s", ". ");
-           len += 2;
-       }
-       msg_ctx = 0;
-       while (!msg_ctx) {
-           /* convert minor status code (underlying routine error) to text */
-           maj_stat = gss_display_status(&min_stat, minor_status,
-               GSS_C_MECH_CODE, GSS_C_NULL_OID, &msg_ctx, &status_string);
-           if (maj_stat == GSS_S_COMPLETE) {
-               if (sizeof(buf) > len + status_string.length) {
-                   sprintf(buf + len, "%s", (char *) status_string.value);
-                   len += status_string.length;
-               }
-               gss_release_buffer(&min_stat, &status_string);
-               break;
-           }
-           gss_release_buffer(&min_stat, &status_string);
-       }
-       fprintf(stderr, "%s| %s: %s failed: %s\n", LogTime(), PROGRAM, function,
-           buf);
-       return (1);
+        OM_uint32 maj_stat, min_stat;
+        OM_uint32 msg_ctx = 0;
+        gss_buffer_desc status_string;
+        char buf[1024];
+        size_t len;
+
+        len = 0;
+        msg_ctx = 0;
+        do {
+            /* convert major status code (GSS-API error) to text */
+            maj_stat = gss_display_status(&min_stat, major_status,
+                                          GSS_C_GSS_CODE, GSS_C_NULL_OID, &msg_ctx, &status_string);
+            if (maj_stat == GSS_S_COMPLETE && status_string.length > 0) {
+                if (sizeof(buf) > len + status_string.length + 1) {
+                    snprintf(buf + len, (sizeof(buf) - len), "%s", (char *) status_string.value);
+                    len += status_string.length;
+                }
+            } else
+                msg_ctx = 0;
+            gss_release_buffer(&min_stat, &status_string);
+        } while (msg_ctx);
+        if (sizeof(buf) > len + 2) {
+            snprintf(buf + len, (sizeof(buf) - len), "%s", ". ");
+            len += 2;
+        }
+        msg_ctx = 0;
+        do {
+            /* convert minor status code (underlying routine error) to text */
+            maj_stat = gss_display_status(&min_stat, minor_status,
+                                          GSS_C_MECH_CODE, GSS_C_NULL_OID, &msg_ctx, &status_string);
+            if (maj_stat == GSS_S_COMPLETE && status_string.length > 0) {
+                if (sizeof(buf) > len + status_string.length) {
+                    snprintf(buf + len, (sizeof(buf) - len), "%s", (char *) status_string.value);
+                    len += status_string.length;
+                }
+            } else
+                msg_ctx = 0;
+            gss_release_buffer(&min_stat, &status_string);
+        } while (msg_ctx);
+        fprintf(stderr, "%s| %s: %s failed: %s\n", LogTime(), PROGRAM, function,
+                buf);
+        return (1);
     }
     return (0);
 }
@@ -180,44 +175,37 @@ squid_kerb_proxy_auth(char *proxy)
     setbuf(stdin, NULL);
 
     if (!proxy) {
-       fprintf(stderr, "%s| %s: Error: No proxy server name\n", LogTime(),
-           PROGRAM);
-       return NULL;
+        fprintf(stderr, "%s| %s: Error: No proxy server name\n", LogTime(),
+                PROGRAM);
+        return NULL;
     }
-
     service.value = xmalloc(strlen("HTTP") + strlen(proxy) + 2);
-    snprintf((char*)service.value, strlen("HTTP") + strlen(proxy) + 2, "%s@%s", "HTTP", proxy);
+    snprintf((char *) service.value, strlen("HTTP") + strlen(proxy) + 2, "%s@%s", "HTTP", proxy);
     service.length = strlen((char *) service.value);
 
     major_status = gss_import_name(&minor_status, &service,
-       gss_nt_service_name, &server_name);
+                                   gss_nt_service_name, &server_name);
 
     if (check_gss_err(major_status, minor_status, "gss_import_name()"))
-       goto cleanup;
+        goto cleanup;
 
     major_status = gss_init_sec_context(&minor_status,
-       GSS_C_NO_CREDENTIAL, &gss_context, server_name,
-#ifdef HAVE_SPNEGO
-       gss_mech_spnego,
-#else
-       0,
-#endif
-       0,
-       0,
-       GSS_C_NO_CHANNEL_BINDINGS,
-       &input_token, NULL, &output_token, NULL, NULL);
+                                        GSS_C_NO_CREDENTIAL, &gss_context, server_name,
+                                        gss_mech_spnego,
+                                        0,
+                                        0,
+                                        GSS_C_NO_CHANNEL_BINDINGS,
+                                        &input_token, NULL, &output_token, NULL, NULL);
 
     if (check_gss_err(major_status, minor_status, "gss_init_sec_context()"))
-       goto cleanup;
+        goto cleanup;
 
     if (output_token.length) {
-       token = (char*)xmalloc(ska_base64_encode_len(output_token.length));
-       ska_base64_encode(token, (const char *) output_token.value,
-           ska_base64_encode_len(output_token.length), output_token.length);
+        token = (char *) xmalloc((size_t)base64_encode_len((int)output_token.length));
+        base64_encode_str(token, base64_encode_len((int)output_token.length),
+                          (const char *) output_token.value, (int)output_token.length);
     }
-
-
-  cleanup:
+cleanup:
     gss_delete_sec_context(&minor_status, &gss_context, NULL);
     gss_release_buffer(&minor_status, &service);
     gss_release_buffer(&minor_status, &input_token);
@@ -230,35 +218,36 @@ squid_kerb_proxy_auth(char *proxy)
 int
 main(int argc, char *argv[])
 {
-
     const char *Token;
     int count;
 
     if (argc < 2) {
-       fprintf(stderr, "%s| %s: Error: No proxy server name given\n",
-           LogTime(), PROGRAM);
-       exit(99);
+        fprintf(stderr, "%s| %s: Error: No proxy server name given\n",
+                LogTime(), PROGRAM);
+        return 99;
     }
     if (argc == 3) {
-       count = atoi(argv[2]);
-       while (count > 0) {
-           Token = (const char *) squid_kerb_proxy_auth(argv[1]);
-           fprintf(stdout, "YR %s\n", Token ? Token : "NULL");
-           count--;
-       }
-       fprintf(stdout, "QQ\n");
+        count = atoi(argv[2]);
+        while (count > 0) {
+            Token = (const char *) squid_kerb_proxy_auth(argv[1]);
+            fprintf(stdout, "YR %s\n", Token ? Token : "NULL");
+            --count;
+        }
+        fprintf(stdout, "QQ\n");
     } else {
-       Token = (const char *) squid_kerb_proxy_auth(argv[1]);
-       fprintf(stdout, "Token: %s\n", Token ? Token : "NULL");
+        Token = (const char *) squid_kerb_proxy_auth(argv[1]);
+        fprintf(stdout, "Token: %s\n", Token ? Token : "NULL");
     }
 
-    exit(0);
+    return 0;
 }
+
 #else
-#include <stdlib.h>
+#include <cstdlib>
 int
 main(int argc, char *argv[])
 {
-  exit(-1);
+    return -1;
 }
+
 #endif /* HAVE_GSSAPI */