cleaning up conflicts between group code not yet committed and
[samba.git] / source / rpc_client / cli_reg.c
index 2f04facffd8f3cfd9da759abe10519229b50d974..5eccdc7826afa8070aadbb7ad626813bb4022868 100644 (file)
@@ -3,9 +3,9 @@
  *  Unix SMB/Netbios implementation.
  *  Version 1.9.
  *  RPC Pipe client / server routines
- *  Copyright (C) Andrew Tridgell              1992-1997,
- *  Copyright (C) Luke Kenneth Casson Leighton 1996-1997,
- *  Copyright (C) Paul Ashton                       1997.
+ *  Copyright (C) Andrew Tridgell              1992-1998,
+ *  Copyright (C) Luke Kenneth Casson Leighton 1996-1998,
+ *  Copyright (C) Paul Ashton                  1997-1998.
  *  
  *  This program is free software; you can redistribute it and/or modify
  *  it under the terms of the GNU General Public License as published by
 
 extern int DEBUGLEVEL;
 
+/****************************************************************************
+do a REG Open Policy
+****************************************************************************/
+BOOL do_reg_connect(struct cli_state *cli, char *full_keyname, char *key_name,
+                               POLICY_HND *reg_hnd)
+{
+       BOOL res = True;
+       uint32 reg_type = 0;
+
+       if (full_keyname == NULL)
+       {
+               return False;
+       }
+
+       ZERO_STRUCTP(reg_hnd);
+
+       /*
+        * open registry receive a policy handle
+        */
+
+       if (!reg_split_key(full_keyname, &reg_type, key_name))
+       {
+               DEBUG(0,("do_reg_connect: unrecognised key name %s\n", full_keyname));  
+               return False;
+       }
+
+       switch (reg_type)
+       {
+               case HKEY_LOCAL_MACHINE:
+               {
+                       res = res ? do_reg_open_hklm(cli,
+                                       0x84E0, 0x02000000,
+                                       reg_hnd) : False;
+                       break;
+               }
+       
+               case HKEY_USERS:
+               {
+                       res = res ? do_reg_open_hku(cli,
+                                       0x84E0, 0x02000000,
+                                       reg_hnd) : False;
+                       break;
+               }
+               default:
+               {
+                       DEBUG(0,("do_reg_connect: unrecognised hive key\n"));   
+                       return False;
+               }
+       }
+
+       return res;
+}
 
 /****************************************************************************
 do a REG Open Policy
 ****************************************************************************/
-BOOL do_reg_open_policy(struct cli_state *cli, uint16 unknown_0, uint32 level,
+BOOL do_reg_open_hklm(struct cli_state *cli, uint16 unknown_0, uint32 level,
                                POLICY_HND *hnd)
 {
        prs_struct rbuf;
        prs_struct buf; 
-       REG_Q_OPEN_POLICY q_o;
+       REG_Q_OPEN_HKLM q_o;
        BOOL valid_pol = False;
 
        if (hnd == NULL) return False;
@@ -48,30 +100,30 @@ BOOL do_reg_open_policy(struct cli_state *cli, uint16 unknown_0, uint32 level,
        prs_init(&buf , 1024, 4, SAFETY_MARGIN, False);
        prs_init(&rbuf, 0   , 4, SAFETY_MARGIN, True );
 
-       /* create and send a MSRPC command with api REG_OPEN_POLICY */
+       /* create and send a MSRPC command with api REG_OPEN_HKLM */
 
-       DEBUG(4,("REG Open Policy\n"));
+       DEBUG(4,("REG Open HKLM\n"));
 
-       make_reg_q_open_pol(&q_o, unknown_0, level);
+       make_reg_q_open_hklm(&q_o, unknown_0, level);
 
        /* turn parameters into data stream */
-       reg_io_q_open_policy("", &q_o, &buf, 0);
+       reg_io_q_open_hklm("", &q_o, &buf, 0);
 
        /* send the data on \PIPE\ */
-       if (rpc_api_pipe_req(cli, REG_OPEN_POLICY, &buf, &rbuf))
+       if (rpc_api_pipe_req(cli, REG_OPEN_HKLM, &buf, &rbuf))
        {
-               REG_R_OPEN_POLICY r_o;
+               REG_R_OPEN_HKLM r_o;
                BOOL p;
 
                ZERO_STRUCT(r_o);
 
-               reg_io_r_open_policy("", &r_o, &rbuf, 0);
+               reg_io_r_open_hklm("", &r_o, &rbuf, 0);
                p = rbuf.offset != 0;
 
                if (p && r_o.status != 0)
                {
                        /* report error code */
-                       DEBUG(0,("REG_OPEN_POLICY: %s\n", get_nt_error_msg(r_o.status)));
+                       DEBUG(0,("REG_OPEN_HKLM: %s\n", get_nt_error_msg(r_o.status)));
                        p = False;
                }
 
@@ -90,14 +142,14 @@ BOOL do_reg_open_policy(struct cli_state *cli, uint16 unknown_0, uint32 level,
 }
 
 /****************************************************************************
-do a REG Open Unknown 4
+do a REG Open HKU
 ****************************************************************************/
-BOOL do_reg_open_unk_4(struct cli_state *cli, uint16 unknown_0, uint32 level,
+BOOL do_reg_open_hku(struct cli_state *cli, uint16 unknown_0, uint32 level,
                                POLICY_HND *hnd)
 {
        prs_struct rbuf;
        prs_struct buf; 
-       REG_Q_OPEN_UNK_4 q_o;
+       REG_Q_OPEN_HKU q_o;
        BOOL valid_pol = False;
 
        if (hnd == NULL) return False;
@@ -105,30 +157,30 @@ BOOL do_reg_open_unk_4(struct cli_state *cli, uint16 unknown_0, uint32 level,
        prs_init(&buf , 1024, 4, SAFETY_MARGIN, False);
        prs_init(&rbuf, 0   , 4, SAFETY_MARGIN, True );
 
-       /* create and send a MSRPC command with api REG_OPEN_UNK_4 */
+       /* create and send a MSRPC command with api REG_OPEN_HKU */
 
-       DEBUG(4,("REG Open Unknown4\n"));
+       DEBUG(4,("REG Open HKU\n"));
 
-       make_reg_q_open_unk_4(&q_o, unknown_0, level);
+       make_reg_q_open_hku(&q_o, unknown_0, level);
 
        /* turn parameters into data stream */
-       reg_io_q_open_unk_4("", &q_o, &buf, 0);
+       reg_io_q_open_hku("", &q_o, &buf, 0);
 
        /* send the data on \PIPE\ */
-       if (rpc_api_pipe_req(cli, REG_OPEN_UNK_4, &buf, &rbuf))
+       if (rpc_api_pipe_req(cli, REG_OPEN_HKU, &buf, &rbuf))
        {
-               REG_R_OPEN_UNK_4 r_o;
+               REG_R_OPEN_HKU r_o;
                BOOL p;
 
                ZERO_STRUCT(r_o);
 
-               reg_io_r_open_unk_4("", &r_o, &rbuf, 0);
+               reg_io_r_open_hku("", &r_o, &rbuf, 0);
                p = rbuf.offset != 0;
 
                if (p && r_o.status != 0)
                {
                        /* report error code */
-                       DEBUG(0,("REG_OPEN_UNK_4: %s\n", get_nt_error_msg(r_o.status)));
+                       DEBUG(0,("REG_OPEN_HKU: %s\n", get_nt_error_msg(r_o.status)));
                        p = False;
                }
 
@@ -147,18 +199,74 @@ BOOL do_reg_open_unk_4(struct cli_state *cli, uint16 unknown_0, uint32 level,
 }
 
 /****************************************************************************
-do a REG Query Unknown 10
+do a REG Unknown 0xB command.  sent after a create key or create value.
+this might be some sort of "sync" or "refresh" command, sent after
+modification of the registry...
 ****************************************************************************/
-BOOL do_reg_query_unk_10(struct cli_state *cli, POLICY_HND *hnd,
-                               uint32 *unknown_0, uint32 *unknown_1,
+BOOL do_reg_flush_key(struct cli_state *cli, POLICY_HND *hnd)
+{
+       prs_struct rbuf;
+       prs_struct buf; 
+       REG_Q_FLUSH_KEY q_o;
+       BOOL valid_query = False;
+
+       if (hnd == NULL) return False;
+
+       prs_init(&buf , 1024, 4, SAFETY_MARGIN, False);
+       prs_init(&rbuf, 0   , 4, SAFETY_MARGIN, True );
+
+       /* create and send a MSRPC command with api REG_FLUSH_KEY */
+
+       DEBUG(4,("REG Unknown 0xB\n"));
+
+       make_reg_q_flush_key(&q_o, hnd);
+
+       /* turn parameters into data stream */
+       reg_io_q_flush_key("", &q_o, &buf, 0);
+
+       /* send the data on \PIPE\ */
+       if (rpc_api_pipe_req(cli, REG_FLUSH_KEY, &buf, &rbuf))
+       {
+               REG_R_FLUSH_KEY r_o;
+               BOOL p;
+
+               ZERO_STRUCT(r_o);
+
+               reg_io_r_flush_key("", &r_o, &rbuf, 0);
+               p = rbuf.offset != 0;
+
+               if (p && r_o.status != 0)
+               {
+                       /* report error code */
+                       DEBUG(0,("REG_FLUSH_KEY: %s\n", get_nt_error_msg(r_o.status)));
+                       p = False;
+               }
+
+               if (p)
+               {
+                       valid_query = True;
+               }
+       }
+
+       prs_mem_free(&rbuf);
+       prs_mem_free(&buf );
+
+       return valid_query;
+}
+
+/****************************************************************************
+do a REG Query Key
+****************************************************************************/
+BOOL do_reg_query_key(struct cli_state *cli, POLICY_HND *hnd,
+                               char *class, uint32 *class_len,
                                uint32 *num_subkeys, uint32 *max_subkeylen,
-                               uint32 *unknown_4, uint32 *num_values,
+                               uint32 *max_subkeysize, uint32 *num_values,
                                uint32 *max_valnamelen, uint32 *max_valbufsize,
-                               uint32 *unknown_8, NTTIME *mod_time)
+                               uint32 *sec_desc, NTTIME *mod_time)
 {
        prs_struct rbuf;
        prs_struct buf; 
-       REG_Q_QUERY_UNK_10 q_o;
+       REG_Q_QUERY_KEY q_o;
        BOOL valid_query = False;
 
        if (hnd == NULL) return False;
@@ -166,30 +274,30 @@ BOOL do_reg_query_unk_10(struct cli_state *cli, POLICY_HND *hnd,
        prs_init(&buf , 1024, 4, SAFETY_MARGIN, False);
        prs_init(&rbuf, 0   , 4, SAFETY_MARGIN, True );
 
-       /* create and send a MSRPC command with api REG_QUERY_UNK_10 */
+       /* create and send a MSRPC command with api REG_QUERY_KEY */
 
-       DEBUG(4,("REG Query Unknown 10\n"));
+       DEBUG(4,("REG Query Key\n"));
 
-       make_reg_q_query_unk_10(&q_o, hnd);
+       make_reg_q_query_key(&q_o, hnd, *class_len);
 
        /* turn parameters into data stream */
-       reg_io_q_query_unk_10("", &q_o, &buf, 0);
+       reg_io_q_query_key("", &q_o, &buf, 0);
 
        /* send the data on \PIPE\ */
-       if (rpc_api_pipe_req(cli, REG_QUERY_UNK_10, &buf, &rbuf))
+       if (rpc_api_pipe_req(cli, REG_QUERY_KEY, &buf, &rbuf))
        {
-               REG_R_QUERY_UNK_10 r_o;
+               REG_R_QUERY_KEY r_o;
                BOOL p;
 
                ZERO_STRUCT(r_o);
 
-               reg_io_r_query_unk_10("", &r_o, &rbuf, 0);
+               reg_io_r_query_key("", &r_o, &rbuf, 0);
                p = rbuf.offset != 0;
 
                if (p && r_o.status != 0)
                {
                        /* report error code */
-                       DEBUG(0,("REG_QUERY_UNK_10: %s\n", get_nt_error_msg(r_o.status)));
+                       DEBUG(0,("REG_QUERY_KEY: %s\n", get_nt_error_msg(r_o.status)));
                        p = False;
                }
 
@@ -197,16 +305,16 @@ BOOL do_reg_query_unk_10(struct cli_state *cli, POLICY_HND *hnd,
                {
                        valid_query = True;
                        
-                       *unknown_0      = r_o.unknown_0     ;
-                       *unknown_1      = r_o.unknown_1     ;
+                       *class_len      = r_o.hdr_class.uni_max_len;
+                       fstrcpy(class, unistr2_to_str(&r_o.uni_class));
                        *num_subkeys    = r_o.num_subkeys   ;
                        *max_subkeylen  = r_o.max_subkeylen ;
-                       *unknown_4      = r_o.unknown_4     ;
+                       *max_subkeysize = r_o.max_subkeysize;
                        *num_values     = r_o.num_values    ;
                        *max_valnamelen = r_o.max_valnamelen;
                        *max_valbufsize = r_o.max_valbufsize;
-                       *unknown_8      = r_o.unknown_8    ;
-                       *mod_time       = r_o.mod_time     ;
+                       *sec_desc       = r_o.sec_desc      ;
+                       *mod_time       = r_o.mod_time      ;
                }
        }
 
@@ -329,6 +437,54 @@ BOOL do_reg_query_info(struct cli_state *cli, POLICY_HND *hnd,
        return valid_query;
 }
 
+/****************************************************************************
+do a REG Set Key Security 
+****************************************************************************/
+BOOL do_reg_set_key_sec(struct cli_state *cli, POLICY_HND *hnd,
+                               uint32 sec_buf_size, SEC_DESC *sec_buf)
+{
+       prs_struct rbuf;
+       prs_struct buf; 
+       REG_Q_SET_KEY_SEC q_o;
+       BOOL valid_query = False;
+
+       if (hnd == NULL) return False;
+
+       prs_init(&buf , 1024, 4, SAFETY_MARGIN, False);
+       prs_init(&rbuf, 0   , 4, SAFETY_MARGIN, True );
+
+       /* create and send a MSRPC command with api REG_SET_KEY_SEC */
+
+       DEBUG(4,("REG Set Key security.\n"));
+
+       make_reg_q_set_key_sec(&q_o, hnd, sec_buf_size, sec_buf);
+
+       /* turn parameters into data stream */
+       reg_io_q_set_key_sec("", &q_o, &buf, 0);
+
+       /* send the data on \PIPE\ */
+       if (rpc_api_pipe_req(cli, REG_SET_KEY_SEC, &buf, &rbuf))
+       {
+               REG_R_SET_KEY_SEC r_o;
+               BOOL p;
+
+               ZERO_STRUCT(r_o);
+
+               reg_io_r_set_key_sec("", &r_o, &rbuf, 0);
+               p = rbuf.offset != 0;
+
+               if (p && r_o.status != 0)
+               {
+                       valid_query = True;
+               }
+       }
+
+       prs_mem_free(&rbuf);
+       prs_mem_free(&buf );
+
+       return valid_query;
+}
+
 /****************************************************************************
 do a REG Query Key Security 
 ****************************************************************************/
@@ -363,6 +519,10 @@ BOOL do_reg_get_key_sec(struct cli_state *cli, POLICY_HND *hnd,
                ZERO_STRUCT(r_o);
 
                r_o.data = sec_buf;
+               if (*sec_buf_size != 0)
+               {
+                       sec_buf->sec = (SEC_DESC*)malloc(*sec_buf_size);
+               }
                reg_io_r_get_key_sec("", &r_o, &rbuf, 0);
                p = rbuf.offset != 0;
 
@@ -394,6 +554,173 @@ BOOL do_reg_get_key_sec(struct cli_state *cli, POLICY_HND *hnd,
        return valid_query;
 }
 
+/****************************************************************************
+do a REG Delete Value
+****************************************************************************/
+BOOL do_reg_delete_val(struct cli_state *cli, POLICY_HND *hnd, char *val_name)
+{
+       prs_struct rbuf;
+       prs_struct buf; 
+       REG_Q_DELETE_VALUE q_o;
+       BOOL valid_delete = False;
+
+       if (hnd == NULL) return False;
+
+       prs_init(&buf , 1024, 4, SAFETY_MARGIN, False);
+       prs_init(&rbuf, 0   , 4, SAFETY_MARGIN, True );
+
+       /* create and send a MSRPC command with api REG_DELETE_VALUE */
+
+       DEBUG(4,("REG Delete Value: %s\n", val_name));
+
+       make_reg_q_delete_val(&q_o, hnd, val_name);
+
+       /* turn parameters into data stream */
+       reg_io_q_delete_val("", &q_o, &buf, 0);
+
+       /* send the data on \PIPE\ */
+       if (rpc_api_pipe_req(cli, REG_DELETE_VALUE, &buf, &rbuf))
+       {
+               REG_R_DELETE_VALUE r_o;
+               BOOL p;
+
+               ZERO_STRUCT(r_o);
+
+               reg_io_r_delete_val("", &r_o, &rbuf, 0);
+               p = rbuf.offset != 0;
+
+               if (p && r_o.status != 0)
+               {
+                       /* report error code */
+                       DEBUG(0,("REG_DELETE_VALUE: %s\n", get_nt_error_msg(r_o.status)));
+                       p = False;
+               }
+
+               if (p)
+               {
+                       valid_delete = True;
+               }
+       }
+
+       prs_mem_free(&rbuf);
+       prs_mem_free(&buf );
+
+       return valid_delete;
+}
+
+/****************************************************************************
+do a REG Delete Key
+****************************************************************************/
+BOOL do_reg_delete_key(struct cli_state *cli, POLICY_HND *hnd, char *key_name)
+{
+       prs_struct rbuf;
+       prs_struct buf; 
+       REG_Q_DELETE_KEY q_o;
+       BOOL valid_delete = False;
+
+       if (hnd == NULL) return False;
+
+       prs_init(&buf , 1024, 4, SAFETY_MARGIN, False);
+       prs_init(&rbuf, 0   , 4, SAFETY_MARGIN, True );
+
+       /* create and send a MSRPC command with api REG_DELETE_KEY */
+
+       DEBUG(4,("REG Delete Key: %s\n", key_name));
+
+       make_reg_q_delete_key(&q_o, hnd, key_name);
+
+       /* turn parameters into data stream */
+       reg_io_q_delete_key("", &q_o, &buf, 0);
+
+       /* send the data on \PIPE\ */
+       if (rpc_api_pipe_req(cli, REG_DELETE_KEY, &buf, &rbuf))
+       {
+               REG_R_DELETE_KEY r_o;
+               BOOL p;
+
+               ZERO_STRUCT(r_o);
+
+               reg_io_r_delete_key("", &r_o, &rbuf, 0);
+               p = rbuf.offset != 0;
+
+               if (p && r_o.status != 0)
+               {
+                       /* report error code */
+                       DEBUG(0,("REG_DELETE_KEY: %s\n", get_nt_error_msg(r_o.status)));
+                       p = False;
+               }
+
+               if (p)
+               {
+                       valid_delete = True;
+               }
+       }
+
+       prs_mem_free(&rbuf);
+       prs_mem_free(&buf );
+
+       return valid_delete;
+}
+
+/****************************************************************************
+do a REG Create Key
+****************************************************************************/
+BOOL do_reg_create_key(struct cli_state *cli, POLICY_HND *hnd,
+                               char *key_name, char *key_class,
+                               SEC_INFO *sam_access,
+                               POLICY_HND *key)
+{
+       prs_struct rbuf;
+       prs_struct buf; 
+       REG_Q_CREATE_KEY q_o;
+       BOOL valid_create = False;
+
+       if (hnd == NULL) return False;
+
+       prs_init(&buf , 1024, 4, SAFETY_MARGIN, False);
+       prs_init(&rbuf, 0   , 4, SAFETY_MARGIN, True );
+
+       /* create and send a MSRPC command with api REG_CREATE_KEY */
+
+       DEBUG(4,("REG Create Key: %s %s 0x%08x\n", key_name, key_class,
+               sam_access != NULL ? sam_access->perms : 0));
+
+       make_reg_q_create_key(&q_o, hnd, key_name, key_class, sam_access);
+
+       /* turn parameters into data stream */
+       reg_io_q_create_key("", &q_o, &buf, 0);
+
+       /* send the data on \PIPE\ */
+       if (rpc_api_pipe_req(cli, REG_CREATE_KEY, &buf, &rbuf))
+       {
+               REG_R_CREATE_KEY r_o;
+               BOOL p;
+
+               ZERO_STRUCT(r_o);
+
+               reg_io_r_create_key("", &r_o, &rbuf, 0);
+               p = rbuf.offset != 0;
+
+               if (p && r_o.status != 0)
+               {
+                       /* report error code */
+                       DEBUG(0,("REG_CREATE_KEY: %s\n", get_nt_error_msg(r_o.status)));
+                       p = False;
+               }
+
+               if (p)
+               {
+                       valid_create = True;
+                       memcpy(key, r_o.key_pol.data, sizeof(key->data));
+               }
+       }
+
+       prs_mem_free(&rbuf);
+       prs_mem_free(&buf );
+
+       return valid_create;
+}
+
 /****************************************************************************
 do a REG Enum Key
 ****************************************************************************/
@@ -455,6 +782,61 @@ BOOL do_reg_enum_key(struct cli_state *cli, POLICY_HND *hnd,
        return valid_query;
 }
 
+/****************************************************************************
+do a REG Create Value
+****************************************************************************/
+BOOL do_reg_create_val(struct cli_state *cli, POLICY_HND *hnd,
+                               char *val_name, uint32 type, BUFFER3 *data)
+{
+       prs_struct rbuf;
+       prs_struct buf; 
+       REG_Q_CREATE_VALUE q_o;
+       BOOL valid_create = False;
+
+       if (hnd == NULL) return False;
+
+       prs_init(&buf , 1024, 4, SAFETY_MARGIN, False);
+       prs_init(&rbuf, 0   , 4, SAFETY_MARGIN, True );
+
+       /* create and send a MSRPC command with api REG_CREATE_VALUE */
+
+       DEBUG(4,("REG Create Value: %s\n", val_name));
+
+       make_reg_q_create_val(&q_o, hnd, val_name, type, data);
+
+       /* turn parameters into data stream */
+       reg_io_q_create_val("", &q_o, &buf, 0);
+
+       /* send the data on \PIPE\ */
+       if (rpc_api_pipe_req(cli, REG_CREATE_VALUE, &buf, &rbuf))
+       {
+               REG_R_CREATE_VALUE r_o;
+               BOOL p;
+
+               ZERO_STRUCT(r_o);
+
+               reg_io_r_create_val("", &r_o, &rbuf, 0);
+               p = rbuf.offset != 0;
+
+               if (p && r_o.status != 0)
+               {
+                       /* report error code */
+                       DEBUG(0,("REG_CREATE_VALUE: %s\n", get_nt_error_msg(r_o.status)));
+                       p = False;
+               }
+
+               if (p)
+               {
+                       valid_create = True;
+               }
+       }
+
+       prs_mem_free(&rbuf);
+       prs_mem_free(&buf );
+
+       return valid_create;
+}
+
 /****************************************************************************
 do a REG Enum Value
 ****************************************************************************/