r4088: Get medieval on our ass about malloc.... :-). Take control of all our allocation
[tprouty/samba.git] / source / utils / net_ads.c
index 9a824e2662f312789a58e77a4356cabb24d17a01..72e8f70a7d86e4e7639550f48d26072df911d7ca 100644 (file)
@@ -21,7 +21,7 @@
 */
 
 #include "includes.h"
-#include "../utils/net.h"
+#include "utils/net.h"
 
 #ifdef HAVE_ADS
 
@@ -44,9 +44,9 @@ int net_ads_usage(int argc, const char **argv)
 "\n\tdump the machine account details to stdout\n"
 "\nnet ads lookup"\
 "\n\tperform a CLDAP search on the server\n"
-"\nnet ads password <username@realm> -Uadmin_username@realm%%admin_pass"\
+"\nnet ads password <username@realm> <password> -Uadmin_username@realm%%admin_pass"\
 "\n\tchange a user's password using an admin account"\
-"\n\t(note: use realm in UPPERCASE)\n"\
+"\n\t(note: use realm in UPPERCASE, prompts if password is obmitted)\n"\
 "\nnet ads changetrustpw"\
 "\n\tchange the trust account password of this machine in the AD tree\n"\
 "\nnet ads printer [info | publish | remove] <printername> <servername>"\
@@ -55,6 +55,8 @@ int net_ads_usage(int argc, const char **argv)
 "\n\tperform a raw LDAP search and dump the results\n"
 "\nnet ads dn"\
 "\n\tperform a raw LDAP search and dump attributes of a particular DN\n"
+"\nnet ads keytab"\
+"\n\tcreates and updates the kerberos system keytab file\n"
                );
        return -1;
 }
@@ -68,16 +70,19 @@ static int net_ads_lookup(int argc, const char **argv)
 {
        ADS_STRUCT *ads;
 
-       ads = ads_init(NULL, NULL, opt_host);
+       ads = ads_init(NULL, opt_target_workgroup, opt_host);
        if (ads) {
                ads->auth.flags |= ADS_AUTH_NO_BIND;
        }
 
        ads_connect(ads);
 
-       if (!ads || !ads->config.realm) {
+       if (!ads) {
                d_printf("Didn't find the cldap server!\n");
                return -1;
+       } if (!ads->config.realm) {
+               ads->config.realm = opt_target_workgroup;
+               ads->ldap_port = 389;
        }
 
        return ads_cldap_netlogon(ads);
@@ -89,7 +94,7 @@ static int net_ads_info(int argc, const char **argv)
 {
        ADS_STRUCT *ads;
 
-       ads = ads_init(NULL, NULL, opt_host);
+       ads = ads_init(NULL, opt_target_workgroup, opt_host);
 
        if (ads) {
                ads->auth.flags |= ADS_AUTH_NO_BIND;
@@ -109,8 +114,8 @@ static int net_ads_info(int argc, const char **argv)
        d_printf("LDAP port: %d\n", ads->ldap_port);
        d_printf("Server time: %s\n", http_timestring(ads->config.current_time));
 
-       d_printf("KDC server: %s\n", ads->auth.kdc_server );
-       d_printf("Server time offset: %d\n", ads->auth.time_offset );
+       d_printf("KDC server: %s\n", ads->auth.kdc_server );
+       d_printf("Server time offset: %d\n", ads->auth.time_offset );
 
        return 0;
 }
@@ -127,9 +132,14 @@ static ADS_STRUCT *ads_startup(void)
        ADS_STATUS status;
        BOOL need_password = False;
        BOOL second_time = False;
-       char *cp;
+       char *cp;
        
-       ads = ads_init(NULL, NULL, opt_host);
+       /* lp_realm() should be handled by a command line param, 
+          However, the join requires that realm be set in smb.conf
+          and compares our realm with the remote server's so this is
+          ok until someone needs more flexibility */
+          
+       ads = ads_init(lp_realm(), opt_target_workgroup, opt_host);
 
        if (!opt_user_name) {
                opt_user_name = "administrator";
@@ -140,9 +150,9 @@ static ADS_STRUCT *ads_startup(void)
        }
 
 retry:
-       if (!opt_password && need_password) {
+       if (!opt_password && need_password && !opt_machine_pass) {
                char *prompt;
-               asprintf(&prompt,"%s password: ", opt_user_name);
+               asprintf(&prompt,"%s's password: ", opt_user_name);
                opt_password = getpass(prompt);
                free(prompt);
        }
@@ -159,10 +169,10 @@ retry:
         * extract the realm and convert to upper case.
         * This is only used to establish the connection.
         */
-       if ((cp = strchr(ads->auth.user_name, '@'))!=0) {
+       if ((cp = strchr_m(ads->auth.user_name, '@'))!=0) {
                *cp++ = '\0';
                ads->auth.realm = smb_xstrdup(cp);
-               strupper(ads->auth.realm);
+               strupper_m(ads->auth.realm);
        }
 
        status = ads_connect(ads);
@@ -173,7 +183,7 @@ retry:
                        second_time = True;
                        goto retry;
                } else {
-                       DEBUG(1,("ads_connect: %s\n", ads_errstr(status)));
+                       DEBUG(0,("ads_connect: %s\n", ads_errstr(status)));
                        return NULL;
                }
        }
@@ -204,11 +214,12 @@ static int net_ads_workgroup(int argc, const char **argv)
 {
        ADS_STRUCT *ads;
        TALLOC_CTX *ctx;
-       char *workgroup;
+       const char *workgroup;
 
        if (!(ads = ads_startup())) return -1;
 
        if (!(ctx = talloc_init("net_ads_workgroup"))) {
+               ads_destroy(&ads);
                return -1;
        }
 
@@ -216,13 +227,14 @@ static int net_ads_workgroup(int argc, const char **argv)
                d_printf("Failed to find workgroup for realm '%s'\n", 
                         ads->config.realm);
                talloc_destroy(ctx);
+               ads_destroy(&ads);
                return -1;
        }
 
        d_printf("Workgroup: %s\n", workgroup);
 
        talloc_destroy(ctx);
-
+       ads_destroy(&ads);
        return 0;
 }
 
@@ -247,10 +259,10 @@ static BOOL usergrp_display(char *field, void **values, void *data_area)
        if (!values) /* must be new field, indicate string field */
                return True;
        if (StrCaseCmp(field, "sAMAccountName") == 0) {
-               disp_fields[0] = strdup((char *) values[0]);
+               disp_fields[0] = SMB_STRDUP((char *) values[0]);
        }
        if (StrCaseCmp(field, "description") == 0)
-               disp_fields[1] = strdup((char *) values[0]);
+               disp_fields[1] = SMB_STRDUP((char *) values[0]);
        return True;
 }
 
@@ -269,7 +281,9 @@ static int ads_user_add(int argc, const char **argv)
 
        if (argc < 1) return net_ads_user_usage(argc, argv);
        
-       if (!(ads = ads_startup())) return -1;
+       if (!(ads = ads_startup())) {
+               return -1;
+       }
 
        status = ads_find_user_acct(ads, &res, argv[0]);
 
@@ -283,6 +297,10 @@ static int ads_user_add(int argc, const char **argv)
                goto done;
        }
 
+       if (opt_container == NULL) {
+               opt_container = ads_default_ou_string(ads, WELL_KNOWN_GUID_USERS);
+       }
+
        status = ads_add_user_acct(ads, argv[0], opt_container, opt_comment);
 
        if (!ADS_ERR_OK(status)) {
@@ -337,13 +355,18 @@ static int ads_user_info(int argc, const char **argv)
        char **grouplist;
        char *escaped_user = escape_ldap_string_alloc(argv[0]);
 
-       if (argc < 1) return net_ads_user_usage(argc, argv);
+       if (argc < 1) {
+               return net_ads_user_usage(argc, argv);
+       }
        
-       if (!(ads = ads_startup())) return -1;
+       if (!(ads = ads_startup())) {
+               return -1;
+       }
 
        if (!escaped_user) {
                d_printf("ads_user_info: failed to escape user %s\n", argv[0]);
-               return -1;
+               ads_destroy(&ads);
+               return -1;
        }
 
        asprintf(&searchstring, "(sAMAccountName=%s)", escaped_user);
@@ -352,6 +375,7 @@ static int ads_user_info(int argc, const char **argv)
 
        if (!ADS_ERR_OK(rc)) {
                d_printf("ads_search: %s\n", ads_errstr(rc));
+               ads_destroy(&ads);
                return -1;
        }
        
@@ -369,7 +393,6 @@ static int ads_user_info(int argc, const char **argv)
        }
        
        ads_msgfree(ads, res);
-
        ads_destroy(&ads);
        return 0;
 }
@@ -381,13 +404,18 @@ static int ads_user_delete(int argc, const char **argv)
        void *res;
        char *userdn;
 
-       if (argc < 1) return net_ads_user_usage(argc, argv);
+       if (argc < 1) {
+               return net_ads_user_usage(argc, argv);
+       }
        
-       if (!(ads = ads_startup())) return -1;
+       if (!(ads = ads_startup())) {
+               return -1;
+       }
 
        rc = ads_find_user_acct(ads, &res, argv[0]);
        if (!ADS_ERR_OK(rc)) {
                DEBUG(0, ("User %s does not exist\n", argv[0]));
+               ads_destroy(&ads);
                return -1;
        }
        userdn = ads_get_dn(ads, res);
@@ -396,10 +424,12 @@ static int ads_user_delete(int argc, const char **argv)
        ads_memfree(ads, userdn);
        if (!ADS_ERR_OK(rc)) {
                d_printf("User %s deleted\n", argv[0]);
+               ads_destroy(&ads);
                return 0;
        }
        d_printf("Error deleting user %s: %s\n", argv[0], 
                 ads_errstr(rc));
+       ads_destroy(&ads);
        return -1;
 }
 
@@ -418,7 +448,9 @@ int net_ads_user(int argc, const char **argv)
        char *disp_fields[2] = {NULL, NULL};
        
        if (argc == 0) {
-               if (!(ads = ads_startup())) return -1;
+               if (!(ads = ads_startup())) {
+                       return -1;
+               }
 
                if (opt_long_list_entries)
                        d_printf("\nUser name             Comment"\
@@ -449,9 +481,13 @@ static int ads_group_add(int argc, const char **argv)
        void *res=NULL;
        int rc = -1;
 
-       if (argc < 1) return net_ads_group_usage(argc, argv);
+       if (argc < 1) {
+               return net_ads_group_usage(argc, argv);
+       }
        
-       if (!(ads = ads_startup())) return -1;
+       if (!(ads = ads_startup())) {
+               return -1;
+       }
 
        status = ads_find_user_acct(ads, &res, argv[0]);
 
@@ -466,6 +502,10 @@ static int ads_group_add(int argc, const char **argv)
                goto done;
        }
 
+       if (opt_container == NULL) {
+               opt_container = ads_default_ou_string(ads, WELL_KNOWN_GUID_USERS);
+       }
+
        status = ads_add_group_acct(ads, argv[0], opt_container, opt_comment);
 
        if (ADS_ERR_OK(status)) {
@@ -490,13 +530,18 @@ static int ads_group_delete(int argc, const char **argv)
        void *res;
        char *groupdn;
 
-       if (argc < 1) return net_ads_group_usage(argc, argv);
+       if (argc < 1) {
+               return net_ads_group_usage(argc, argv);
+       }
        
-       if (!(ads = ads_startup())) return -1;
+       if (!(ads = ads_startup())) {
+               return -1;
+       }
 
        rc = ads_find_user_acct(ads, &res, argv[0]);
        if (!ADS_ERR_OK(rc)) {
                DEBUG(0, ("Group %s does not exist\n", argv[0]));
+               ads_destroy(&ads);
                return -1;
        }
        groupdn = ads_get_dn(ads, res);
@@ -505,10 +550,12 @@ static int ads_group_delete(int argc, const char **argv)
        ads_memfree(ads, groupdn);
        if (!ADS_ERR_OK(rc)) {
                d_printf("Group %s deleted\n", argv[0]);
+               ads_destroy(&ads);
                return 0;
        }
        d_printf("Error deleting group %s: %s\n", argv[0], 
                 ads_errstr(rc));
+       ads_destroy(&ads);
        return -1;
 }
 
@@ -526,7 +573,9 @@ int net_ads_group(int argc, const char **argv)
        char *disp_fields[2] = {NULL, NULL};
 
        if (argc == 0) {
-               if (!(ads = ads_startup())) return -1;
+               if (!(ads = ads_startup())) {
+                       return -1;
+               }
 
                if (opt_long_list_entries)
                        d_printf("\nGroup name            Comment"\
@@ -550,21 +599,25 @@ static int net_ads_status(int argc, const char **argv)
        ADS_STATUS rc;
        void *res;
 
-       if (!(ads = ads_startup())) return -1;
+       if (!(ads = ads_startup())) {
+               return -1;
+       }
 
        rc = ads_find_machine_acct(ads, &res, global_myname());
        if (!ADS_ERR_OK(rc)) {
                d_printf("ads_find_machine_acct: %s\n", ads_errstr(rc));
+               ads_destroy(&ads);
                return -1;
        }
 
        if (ads_count_replies(ads, res) == 0) {
                d_printf("No machine account for '%s' found\n", global_myname());
+               ads_destroy(&ads);
                return -1;
        }
 
        ads_dump(ads, res);
-
+       ads_destroy(&ads);
        return 0;
 }
 
@@ -579,10 +632,7 @@ static int net_ads_leave(int argc, const char **argv)
        }
 
        if (!opt_password) {
-               char *user_name;
-               asprintf(&user_name, "%s$", global_myname());
-               opt_password = secrets_fetch_machine_password(opt_target_workgroup, NULL, NULL);
-               opt_user_name = user_name;
+               net_use_machine_password();
        }
 
        if (!(ads = ads_startup())) {
@@ -591,19 +641,19 @@ static int net_ads_leave(int argc, const char **argv)
 
        rc = ads_leave_realm(ads, global_myname());
        if (!ADS_ERR_OK(rc)) {
-           d_printf("Failed to delete host '%s' from the '%s' realm.\n", 
-                    global_myname(), ads->config.realm);
-           return -1;
+               d_printf("Failed to delete host '%s' from the '%s' realm.\n", 
+                       global_myname(), ads->config.realm);
+               ads_destroy(&ads);
+               return -1;
        }
 
        d_printf("Removed '%s' from realm '%s'\n", global_myname(), ads->config.realm);
-
+       ads_destroy(&ads);
        return 0;
 }
 
 static int net_ads_join_ok(void)
 {
-       char *user_name;
        ADS_STRUCT *ads = NULL;
 
        if (!secrets_init()) {
@@ -611,9 +661,7 @@ static int net_ads_join_ok(void)
                return -1;
        }
 
-       asprintf(&user_name, "%s$", global_myname());
-       opt_user_name = user_name;
-       opt_password = secrets_fetch_machine_password(opt_target_workgroup, NULL, NULL);
+       net_use_machine_password();
 
        if (!(ads = ads_startup())) {
                return -1;
@@ -648,16 +696,21 @@ int net_ads_join(int argc, const char **argv)
        ADS_STRUCT *ads;
        ADS_STATUS rc;
        char *password;
+       char *machine_account = NULL;
        char *tmp_password;
-       const char *org_unit = "Computers";
+       const char *org_unit = NULL;
        char *dn;
        void *res;
        DOM_SID dom_sid;
        char *ou_str;
        uint32 sec_channel_type = SEC_CHAN_WKSTA;
        uint32 account_type = UF_WORKSTATION_TRUST_ACCOUNT;
+       const char *short_domain_name = NULL;
+       TALLOC_CTX *ctx = NULL;
 
-       if (argc > 0) org_unit = argv[0];
+       if (argc > 0) {
+               org_unit = argv[0];
+       }
 
        if (!secrets_init()) {
                DEBUG(1,("Failed to initialise secrets database\n"));
@@ -665,67 +718,158 @@ int net_ads_join(int argc, const char **argv)
        }
 
        tmp_password = generate_random_str(DEFAULT_TRUST_ACCOUNT_PASSWORD_LENGTH);
-       password = strdup(tmp_password);
+       password = SMB_STRDUP(tmp_password);
 
-       if (!(ads = ads_startup())) return -1;
+       if (!(ads = ads_startup())) {
+               return -1;
+       }
 
-       ou_str = ads_ou_string(org_unit);
+       if (!*lp_realm()) {
+               d_printf("realm must be set in in smb.conf for ADS join to succeed.\n");
+               ads_destroy(&ads);
+               return -1;
+       }
+
+       if (strcmp(ads->config.realm, lp_realm()) != 0) {
+               d_printf("realm of remote server (%s) and realm in smb.conf (%s) DO NOT match.  Aborting join\n", ads->config.realm, lp_realm());
+               ads_destroy(&ads);
+               return -1;
+       }
+
+       ou_str = ads_ou_string(ads,org_unit);
        asprintf(&dn, "%s,%s", ou_str, ads->config.bind_path);
        free(ou_str);
 
        rc = ads_search_dn(ads, &res, dn, NULL);
        ads_msgfree(ads, res);
 
-       if (rc.error_type == ADS_ERROR_LDAP && rc.err.rc == LDAP_NO_SUCH_OBJECT) {
+       if (rc.error_type == ENUM_ADS_ERROR_LDAP && rc.err.rc == LDAP_NO_SUCH_OBJECT) {
                d_printf("ads_join_realm: organizational unit %s does not exist (dn:%s)\n", 
                         org_unit, dn);
+               ads_destroy(&ads);
                return -1;
        }
        free(dn);
 
        if (!ADS_ERR_OK(rc)) {
                d_printf("ads_join_realm: %s\n", ads_errstr(rc));
+               ads_destroy(&ads);
                return -1;
        }       
 
        rc = ads_join_realm(ads, global_myname(), account_type, org_unit);
        if (!ADS_ERR_OK(rc)) {
                d_printf("ads_join_realm: %s\n", ads_errstr(rc));
+               ads_destroy(&ads);
                return -1;
        }
 
        rc = ads_domain_sid(ads, &dom_sid);
        if (!ADS_ERR_OK(rc)) {
-               d_printf("ads_domain_sid: %s\n", ads_errstr(rc));
+               d_printf("ads_domain_sid: %s\n", ads_errstr(rc));       
+               ads_destroy(&ads);
+               return -1;
+       }
+
+       if (asprintf(&machine_account, "%s$", global_myname()) == -1) {
+               d_printf("asprintf failed\n");
+               ads_destroy(&ads);
                return -1;
        }
 
-       rc = ads_set_machine_password(ads, global_myname(), password);
+       rc = ads_set_machine_password(ads, machine_account, password);
        if (!ADS_ERR_OK(rc)) {
                d_printf("ads_set_machine_password: %s\n", ads_errstr(rc));
+               ads_destroy(&ads);
                return -1;
        }
-
+       
+       /* make sure we get the right workgroup */
+       
+       if ( !(ctx = talloc_init("net ads join")) ) {
+               d_printf("talloc_init() failed!\n");
+               ads_destroy(&ads);
+               return -1;
+       }
+       
+       rc = ads_workgroup_name(ads, ctx, &short_domain_name);
+       if ( ADS_ERR_OK(rc) ) {
+               if ( !strequal(lp_workgroup(), short_domain_name) ) {
+                       d_printf("The workgroup in smb.conf does not match the short\n");
+                       d_printf("domain name obtained from the server.\n");
+                       d_printf("Using the name [%s] from the server.\n", short_domain_name);
+                       d_printf("You should set \"workgroup = %s\" in smb.conf.\n", short_domain_name);
+               }
+       } else {
+               short_domain_name = lp_workgroup();
+       }
+       
+       d_printf("Using short domain name -- %s\n", short_domain_name);
+       
+       /*  HACK ALRET!  Store the sid and password under bother the lp_workgroup() 
+           value from smb.conf and the string returned from the server.  The former is
+           neede to bootstrap winbindd's first connection to the DC to get the real 
+           short domain name   --jerry */
+           
        if (!secrets_store_domain_sid(lp_workgroup(), &dom_sid)) {
                DEBUG(1,("Failed to save domain sid\n"));
+               ads_destroy(&ads);
                return -1;
        }
 
        if (!secrets_store_machine_password(password, lp_workgroup(), sec_channel_type)) {
                DEBUG(1,("Failed to save machine password\n"));
+               ads_destroy(&ads);
                return -1;
        }
 
-       d_printf("Joined '%s' to realm '%s'\n", global_myname(), ads->config.realm);
+#ifdef HAVE_KRB5
+       if (!kerberos_derive_salting_principal(machine_account)) {
+               DEBUG(1,("Failed to determine salting principal\n"));
+               ads_destroy(&ads);
+               return -1;
+       }
+
+       if (!kerberos_derive_cifs_salting_principals()) {
+               DEBUG(1,("Failed to determine salting principals\n"));
+               ads_destroy(&ads);
+               return -1;
+       }
+#endif
+
+       if (!secrets_store_domain_sid(short_domain_name, &dom_sid)) {
+               DEBUG(1,("Failed to save domain sid\n"));
+               ads_destroy(&ads);
+               return -1;
+       }
+
+       if (!secrets_store_machine_password(password, short_domain_name, sec_channel_type)) {
+               DEBUG(1,("Failed to save machine password\n"));
+               ads_destroy(&ads);
+               return -1;
+       }
+       
+       /* Now build the keytab, using the same ADS connection */
+       if (lp_use_kerberos_keytab() && ads_keytab_create_default(ads)) {
+               DEBUG(1,("Error creating host keytab!\n"));
+       }
 
-       free(password);
+       d_printf("Joined '%s' to realm '%s'\n", global_myname(), ads->config.realm);
 
+       SAFE_FREE(password);
+       SAFE_FREE(machine_account);
+       if ( ctx ) {
+               talloc_destroy(ctx);
+       }
+       ads_destroy(&ads);
        return 0;
 }
 
 int net_ads_printer_usage(int argc, const char **argv)
 {
        d_printf(
+"\nnet ads printer search <printer>"
+"\n\tsearch for a printer in the directory\n"
 "\nnet ads printer info <printer> <server>"
 "\n\tlookup info in directory for printer on server"
 "\n\t(note: printer defaults to \"*\", server defaults to local)\n"
@@ -738,6 +882,38 @@ int net_ads_printer_usage(int argc, const char **argv)
        return -1;
 }
 
+static int net_ads_printer_search(int argc, const char **argv)
+{
+       ADS_STRUCT *ads;
+       ADS_STATUS rc;
+       void *res = NULL;
+
+       if (!(ads = ads_startup())) {
+               return -1;
+       }
+
+       rc = ads_find_printers(ads, &res);
+
+       if (!ADS_ERR_OK(rc)) {
+               d_printf("ads_find_printer: %s\n", ads_errstr(rc));
+               ads_msgfree(ads, res);
+               ads_destroy(&ads);
+               return -1;
+       }
+
+       if (ads_count_replies(ads, res) == 0) {
+               d_printf("No results found\n");
+               ads_msgfree(ads, res);
+               ads_destroy(&ads);
+               return -1;
+       }
+
+       ads_dump(ads, res);
+       ads_msgfree(ads, res);
+       ads_destroy(&ads);
+       return 0;
+}
+
 static int net_ads_printer_info(int argc, const char **argv)
 {
        ADS_STRUCT *ads;
@@ -745,34 +921,41 @@ static int net_ads_printer_info(int argc, const char **argv)
        const char *servername, *printername;
        void *res = NULL;
 
-       if (!(ads = ads_startup())) return -1;
+       if (!(ads = ads_startup())) {
+               return -1;
+       }
 
-       if (argc > 0)
+       if (argc > 0) {
                printername = argv[0];
-       else
+       } else {
                printername = "*";
+       }
 
-       if (argc > 1)
+       if (argc > 1) {
                servername =  argv[1];
-       else
+       } else {
                servername = global_myname();
+       }
 
        rc = ads_find_printer_on_server(ads, &res, printername, servername);
 
        if (!ADS_ERR_OK(rc)) {
                d_printf("ads_find_printer_on_server: %s\n", ads_errstr(rc));
                ads_msgfree(ads, res);
+               ads_destroy(&ads);
                return -1;
        }
 
        if (ads_count_replies(ads, res) == 0) {
                d_printf("Printer '%s' not found\n", printername);
                ads_msgfree(ads, res);
+               ads_destroy(&ads);
                return -1;
        }
 
        ads_dump(ads, res);
        ads_msgfree(ads, res);
+       ads_destroy(&ads);
 
        return 0;
 }
@@ -786,7 +969,7 @@ static int net_ads_printer_publish(int argc, const char **argv)
 {
         ADS_STRUCT *ads;
         ADS_STATUS rc;
-       const char *servername;
+       const char *servername, *printername;
        struct cli_state *cli;
        struct in_addr          server_ip;
        NTSTATUS nt_status;
@@ -795,20 +978,23 @@ static int net_ads_printer_publish(int argc, const char **argv)
        char *prt_dn, *srv_dn, **srv_cn;
        void *res = NULL;
 
-       if (!(ads = ads_startup())) return -1;
+       if (!(ads = ads_startup())) {
+               return -1;
+       }
 
-       if (argc < 1)
+       if (argc < 1) {
                return net_ads_printer_usage(argc, argv);
+       }
        
-       if (argc == 2)
+       printername = argv[0];
+
+       if (argc == 2) {
                servername = argv[1];
-       else
+       } else {
                servername = global_myname();
+       }
                
-       ads_find_machine_acct(ads, &res, servername);
-       srv_dn = ldap_get_dn(ads->ld, res);
-       srv_cn = ldap_explode_dn(srv_dn, 1);
-       asprintf(&prt_dn, "cn=%s-%s,%s", srv_cn[0], argv[0], srv_dn);
+       /* Get printer data from SPOOLSS */
 
        resolve_name(servername, &server_ip, 0x20);
 
@@ -818,18 +1004,43 @@ static int net_ads_printer_publish(int argc, const char **argv)
                                        opt_user_name, opt_workgroup,
                                        opt_password ? opt_password : "", 
                                        CLI_FULL_CONNECTION_USE_KERBEROS, 
-                                       NULL);
+                                       Undefined, NULL);
+
+       if (NT_STATUS_IS_ERR(nt_status)) {
+               d_printf("Unable to open a connnection to %s to obtain data "
+                        "for %s\n", servername, printername);
+               ads_destroy(&ads);
+               return -1;
+       }
+
+       /* Publish on AD server */
+
+       ads_find_machine_acct(ads, &res, servername);
+
+       if (ads_count_replies(ads, res) == 0) {
+               d_printf("Could not find machine account for server %s\n", 
+                        servername);
+               ads_destroy(&ads);
+               return -1;
+       }
+
+       srv_dn = ldap_get_dn(ads->ld, res);
+       srv_cn = ldap_explode_dn(srv_dn, 1);
+
+       asprintf(&prt_dn, "cn=%s-%s,%s", srv_cn[0], printername, srv_dn);
 
        cli_nt_session_open(cli, PI_SPOOLSS);
-       get_remote_printer_publishing_data(cli, mem_ctx, &mods, argv[0]);
+       get_remote_printer_publishing_data(cli, mem_ctx, &mods, printername);
 
         rc = ads_add_printer_entry(ads, prt_dn, mem_ctx, &mods);
         if (!ADS_ERR_OK(rc)) {
                 d_printf("ads_publish_printer: %s\n", ads_errstr(rc));
+               ads_destroy(&ads);
                 return -1;
         }
  
         d_printf("published printer\n");
+       ads_destroy(&ads);
  
        return 0;
 }
@@ -842,27 +1053,33 @@ static int net_ads_printer_remove(int argc, const char **argv)
        char *prt_dn;
        void *res = NULL;
 
-       if (!(ads = ads_startup())) return -1;
+       if (!(ads = ads_startup())) {
+               return -1;
+       }
 
-       if (argc < 1)
+       if (argc < 1) {
                return net_ads_printer_usage(argc, argv);
+       }
 
-       if (argc > 1)
+       if (argc > 1) {
                servername = argv[1];
-       else
+       } else {
                servername = global_myname();
+       }
 
        rc = ads_find_printer_on_server(ads, &res, argv[0], servername);
 
        if (!ADS_ERR_OK(rc)) {
                d_printf("ads_find_printer_on_server: %s\n", ads_errstr(rc));
                ads_msgfree(ads, res);
+               ads_destroy(&ads);
                return -1;
        }
 
        if (ads_count_replies(ads, res) == 0) {
                d_printf("Printer '%s' not found\n", argv[1]);
                ads_msgfree(ads, res);
+               ads_destroy(&ads);
                return -1;
        }
 
@@ -873,15 +1090,18 @@ static int net_ads_printer_remove(int argc, const char **argv)
 
        if (!ADS_ERR_OK(rc)) {
                d_printf("ads_del_dn: %s\n", ads_errstr(rc));
+               ads_destroy(&ads);
                return -1;
        }
 
+       ads_destroy(&ads);
        return 0;
 }
 
 static int net_ads_printer(int argc, const char **argv)
 {
        struct functable func[] = {
+               {"SEARCH", net_ads_printer_search},
                {"INFO", net_ads_printer_info},
                {"PUBLISH", net_ads_printer_publish},
                {"REMOVE", net_ads_printer_remove},
@@ -894,118 +1114,123 @@ static int net_ads_printer(int argc, const char **argv)
 
 static int net_ads_password(int argc, const char **argv)
 {
-    ADS_STRUCT *ads;
-    const char *auth_principal = opt_user_name;
-    const char *auth_password = opt_password;
-    char *realm = NULL;
-    char *new_password = NULL;
-    char *c, *prompt;
-    const char *user;
-    ADS_STATUS ret;
-
-    if (opt_user_name == NULL || opt_password == NULL) {
-           d_printf("You must supply an administrator username/password\n");
-           return -1;
-    }
+       ADS_STRUCT *ads;
+       const char *auth_principal = opt_user_name;
+       const char *auth_password = opt_password;
+       char *realm = NULL;
+       char *new_password = NULL;
+       char *c, *prompt;
+       const char *user;
+       ADS_STATUS ret;
+
+       if (opt_user_name == NULL || opt_password == NULL) {
+               d_printf("You must supply an administrator username/password\n");
+               return -1;
+       }
 
-    
-    if (argc != 1) {
-           d_printf("ERROR: You must say which username to change password for\n");
-           return -1;
-    }
-
-    user = argv[0];
-    if (!strchr(user, '@')) {
-           asprintf(&c, "%s@%s", argv[0], lp_realm());
-           user = c;
-    }
-
-    use_in_memory_ccache();    
-    c = strchr(auth_principal, '@');
-    if (c) {
-           realm = ++c;
-    } else {
-           realm = lp_realm();
-    }
-
-    /* use the realm so we can eventually change passwords for users 
-    in realms other than default */
-    if (!(ads = ads_init(realm, NULL, NULL))) return -1;
-
-    /* we don't actually need a full connect, but it's the easy way to
-       fill in the KDC's addresss */
-    ads_connect(ads);
-    
-    if (!ads || !ads->config.realm) {
-           d_printf("Didn't find the kerberos server!\n");
-           return -1;
-    }
+       if (argc < 1) {
+               d_printf("ERROR: You must say which username to change password for\n");
+               return -1;
+       }
+
+       user = argv[0];
+       if (!strchr_m(user, '@')) {
+               asprintf(&c, "%s@%s", argv[0], lp_realm());
+               user = c;
+       }
 
-    asprintf(&prompt, "Enter new password for %s:", user);
+       use_in_memory_ccache();    
+       c = strchr_m(auth_principal, '@');
+       if (c) {
+               realm = ++c;
+       } else {
+               realm = lp_realm();
+       }
 
-    new_password = getpass(prompt);
+       /* use the realm so we can eventually change passwords for users 
+       in realms other than default */
+       if (!(ads = ads_init(realm, NULL, NULL))) {
+               return -1;
+       }
+
+       /* we don't actually need a full connect, but it's the easy way to
+               fill in the KDC's addresss */
+       ads_connect(ads);
+    
+       if (!ads || !ads->config.realm) {
+               d_printf("Didn't find the kerberos server!\n");
+               return -1;
+       }
+
+       if (argv[1]) {
+               new_password = (char *)argv[1];
+       } else {
+               asprintf(&prompt, "Enter new password for %s:", user);
+               new_password = getpass(prompt);
+               free(prompt);
+       }
 
-    ret = kerberos_set_password(ads->auth.kdc_server, auth_principal, 
+       ret = kerberos_set_password(ads->auth.kdc_server, auth_principal, 
                                auth_password, user, new_password, ads->auth.time_offset);
-    if (!ADS_ERR_OK(ret)) {
-       d_printf("Password change failed :-( ...\n");
-       ads_destroy(&ads);
-       free(prompt);
-       return -1;
-    }
+       if (!ADS_ERR_OK(ret)) {
+               d_printf("Password change failed :-( ...\n");
+               ads_destroy(&ads);
+               return -1;
+       }
 
-    d_printf("Password change for %s completed.\n", user);
-    ads_destroy(&ads);
-    free(prompt);
+       d_printf("Password change for %s completed.\n", user);
+       ads_destroy(&ads);
 
-    return 0;
+       return 0;
 }
 
-
 int net_ads_changetrustpw(int argc, const char **argv)
 {    
-    ADS_STRUCT *ads;
-    char *host_principal;
-    char *hostname;
-    ADS_STATUS ret;
-    char *user_name;
+       ADS_STRUCT *ads;
+       char *host_principal;
+       fstring my_name;
+       ADS_STATUS ret;
 
-    if (!secrets_init()) {
-           DEBUG(1,("Failed to initialise secrets database\n"));
-           return -1;
-    }
+       if (!secrets_init()) {
+               DEBUG(1,("Failed to initialise secrets database\n"));
+               return -1;
+       }
 
-    asprintf(&user_name, "%s$", global_myname());
-    opt_user_name = user_name;
+       net_use_machine_password();
 
-    opt_password = secrets_fetch_machine_password(opt_target_workgroup, NULL, NULL);
+       use_in_memory_ccache();
+
+       if (!(ads = ads_startup())) {
+               return -1;
+       }
 
-    use_in_memory_ccache();
+       fstrcpy(my_name, global_myname());
+       strlower_m(my_name);
+       asprintf(&host_principal, "%s@%s", my_name, ads->config.realm);
+       d_printf("Changing password for principal: HOST/%s\n", host_principal);
 
-    if (!(ads = ads_startup())) {
-           return -1;
-    }
+       ret = ads_change_trust_account_password(ads, host_principal);
 
-    hostname = strdup(global_myname());
-    strlower(hostname);
-    asprintf(&host_principal, "%s@%s", hostname, ads->config.realm);
-    SAFE_FREE(hostname);
-    d_printf("Changing password for principal: HOST/%s\n", host_principal);
+       if (!ADS_ERR_OK(ret)) {
+               d_printf("Password change failed :-( ...\n");
+               ads_destroy(&ads);
+               SAFE_FREE(host_principal);
+               return -1;
+       }
     
-    ret = ads_change_trust_account_password(ads, host_principal);
+       d_printf("Password change for principal HOST/%s succeeded.\n", host_principal);
+
+       if (lp_use_kerberos_keytab()) {
+               d_printf("Attempting to update system keytab with new password.\n");
+               if (ads_keytab_create_default(ads)) {
+                       d_printf("Failed to update system keytab.\n");
+               }
+       }
 
-    if (!ADS_ERR_OK(ret)) {
-       d_printf("Password change failed :-( ...\n");
        ads_destroy(&ads);
        SAFE_FREE(host_principal);
-       return -1;
-    }
-    
-    d_printf("Password change for principal HOST/%s succeeded.\n", host_principal);
-    ads_destroy(&ads);
-    SAFE_FREE(host_principal);
 
-    return 0;
+       return 0;
 }
 
 /*
@@ -1032,7 +1257,7 @@ static int net_ads_search(int argc, const char **argv)
 {
        ADS_STRUCT *ads;
        ADS_STATUS rc;
-       const char *exp;
+       const char *ldap_exp;
        const char **attrs;
        void *res = NULL;
 
@@ -1044,14 +1269,15 @@ static int net_ads_search(int argc, const char **argv)
                return -1;
        }
 
-       exp = argv[0];
+       ldap_exp = argv[0];
        attrs = (argv + 1);
 
        rc = ads_do_search_all(ads, ads->config.bind_path,
                               LDAP_SCOPE_SUBTREE,
-                              exp, attrs, &res);
+                              ldap_exp, attrs, &res);
        if (!ADS_ERR_OK(rc)) {
                d_printf("search failed: %s\n", ads_errstr(rc));
+               ads_destroy(&ads);
                return -1;
        }       
 
@@ -1111,6 +1337,7 @@ static int net_ads_dn(int argc, const char **argv)
                               "(objectclass=*)", attrs, &res);
        if (!ADS_ERR_OK(rc)) {
                d_printf("search failed: %s\n", ads_errstr(rc));
+               ads_destroy(&ads);
                return -1;
        }       
 
@@ -1125,6 +1352,86 @@ static int net_ads_dn(int argc, const char **argv)
        return 0;
 }
 
+static int net_ads_keytab_usage(int argc, const char **argv)
+{
+       d_printf(
+               "net ads keytab <COMMAND>\n"\
+"<COMMAND> can be either:\n"\
+"  CREATE    Creates a fresh keytab\n"\
+"  ADD       Adds new service principal\n"\
+"  FLUSH     Flushes out all keytab entries\n"\
+"  HELP      Prints this help message\n"\
+"The ADD command will take arguments, the other commands\n"\
+"will not take any arguments.   The arguments given to ADD\n"\
+"should be a list of principals to add.  For example, \n"\
+"   net ads keytab add srv1 srv2\n"\
+"will add principals for the services srv1 and srv2 to the\n"\
+"system's keytab.\n"\
+"\n"
+               );
+       return -1;
+}
+
+static int net_ads_keytab_flush(int argc, const char **argv)
+{
+       int ret;
+       ADS_STRUCT *ads;
+
+       if (!(ads = ads_startup())) {
+               return -1;
+       }
+       ret = ads_keytab_flush(ads);
+       ads_destroy(&ads);
+       return ret;
+}
+
+static int net_ads_keytab_add(int argc, const char **argv)
+{
+       int i;
+       int ret = 0;
+       ADS_STRUCT *ads;
+
+       d_printf("Processing principals to add...\n");
+       if (!(ads = ads_startup())) {
+               return -1;
+       }
+       for (i = 0; i < argc; i++) {
+               ret |= ads_keytab_add_entry(ads, argv[i]);
+       }
+       ads_destroy(&ads);
+       return ret;
+}
+
+static int net_ads_keytab_create(int argc, const char **argv)
+{
+       ADS_STRUCT *ads;
+       int ret;
+
+       if (!(ads = ads_startup())) {
+               return -1;
+       }
+       ret = ads_keytab_create_default(ads);
+       ads_destroy(&ads);
+       return ret;
+}
+
+int net_ads_keytab(int argc, const char **argv)
+{
+       struct functable func[] = {
+               {"CREATE", net_ads_keytab_create},
+               {"ADD", net_ads_keytab_add},
+               {"FLUSH", net_ads_keytab_flush},
+               {"HELP", net_ads_keytab_usage},
+               {NULL, NULL}
+       };
+
+       if (!lp_use_kerberos_keytab()) {
+               d_printf("\nWarning: \"use kerberos keytab\" must be set to \"true\" in order to \
+use keytab functions.\n");
+       }
+
+       return net_run_function(argc, argv, func, net_ads_keytab_usage);
+}
 
 int net_ads_help(int argc, const char **argv)
 {
@@ -1164,6 +1471,7 @@ int net_ads(int argc, const char **argv)
                {"DN", net_ads_dn},
                {"WORKGROUP", net_ads_workgroup},
                {"LOOKUP", net_ads_lookup},
+               {"KEYTAB", net_ads_keytab},
                {"HELP", net_ads_help},
                {NULL, NULL}
        };
@@ -1179,6 +1487,11 @@ static int net_ads_noads(void)
        return -1;
 }
 
+int net_ads_keytab(int argc, const char **argv)
+{
+       return net_ads_noads();
+}
+
 int net_ads_usage(int argc, const char **argv)
 {
        return net_ads_noads();