r6942: * merging the registry changes back to the 3.0 tree
[samba.git] / source3 / rpc_parse / parse_reg.c
index 23262279a488b615b89fce64fd354bb9ae3d0acb..ce23d15a11ca224d72d8f2e98f57c4980ad1c95a 100644 (file)
@@ -1,10 +1,13 @@
-/* 
- *  Unix SMB/Netbios implementation.
- *  Version 1.9.
+/*
+ *  Unix SMB/CIFS implementation.
  *  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) Marc Jacobsen                     1999.
+ *  Copyright (C) Simo Sorce                        2000.
+ *  Copyright (C) Jeremy Cooper                     2004
+ *  Copyright (C) Gerald Carter                     2002-2005.
  *  
  *  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
  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  */
 
-
 #include "includes.h"
 
-extern int DEBUGLEVEL;
+#undef DBGC_CLASS
+#define DBGC_CLASS DBGC_RPC_PARSE
+
+/*******************************************************************
+ Fill in a REGVAL_BUFFER for the data given a REGISTRY_VALUE
+ *******************************************************************/
 
+static uint32 reg_init_regval_buffer( REGVAL_BUFFER *buf2, REGISTRY_VALUE *val )
+{
+       uint32          real_size = 0;
+       
+       if ( !buf2 || !val )
+               return 0;
+               
+       real_size = regval_size(val);
+       init_regval_buffer( buf2, (unsigned char*)regval_data_p(val), real_size );
+
+       return real_size;
+}
 
 /*******************************************************************
- Inits a structure.
+ Inits a hive connect request structure
 ********************************************************************/
 
-void init_reg_q_open_hklm(REG_Q_OPEN_HKLM *q_o,
-                               uint16 unknown_0, uint32 level)
+void init_reg_q_open_hive( REG_Q_OPEN_HIVE *q_o, uint32 access_desired )
 {
-       q_o->ptr = 1;
-       q_o->unknown_0 = unknown_0;
-       q_o->unknown_1 = 0x0; /* random - changes */
-       q_o->level = level;
+       
+       q_o->server = TALLOC_P( get_talloc_ctx(), uint16);
+       *q_o->server = 0x1;
+       
+       q_o->access = access_desired;
 }
 
 /*******************************************************************
-reads or writes a structure.
+Marshalls a hive connect request
 ********************************************************************/
 
-BOOL reg_io_q_open_hklm(char *desc,  REG_Q_OPEN_HKLM *r_q, prs_struct *ps, int depth)
+BOOL reg_io_q_open_hive(const char *desc, REG_Q_OPEN_HIVE *q_u,
+                        prs_struct *ps, int depth)
 {
-       if (r_q == NULL)
-               return False;
-
-       prs_debug(ps, depth, desc, "reg_io_q_open_hklm");
+       prs_debug(ps, depth, desc, "reg_io_q_open_hive");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
-       
-       if(!prs_uint32("ptr      ", ps, depth, &r_q->ptr))
+
+       if(!prs_pointer("server", ps, depth, (void**)&q_u->server, sizeof(uint16), (PRS_POINTER_CAST)prs_uint16))
                return False;
 
-       if (r_q->ptr != 0) {
-               if(!prs_uint16("unknown_0", ps, depth, &r_q->unknown_0))
-                       return False;
-               if(!prs_uint16("unknown_1", ps, depth, &r_q->unknown_1))
-                       return False;
-               if(!prs_uint32("level    ", ps, depth, &r_q->level))
-                       return False;
-       }
+       if(!prs_align(ps))
+               return False;
+       if(!prs_uint32("access", ps, depth, &q_u->access))
+               return False;
 
        return True;
 }
 
 
 /*******************************************************************
-reads or writes a structure.
+Unmarshalls a hive connect response
 ********************************************************************/
 
-BOOL reg_io_r_open_hklm(char *desc,  REG_R_OPEN_HKLM *r_r, prs_struct *ps, int depth)
+BOOL reg_io_r_open_hive(const char *desc,  REG_R_OPEN_HIVE *r_u,
+                        prs_struct *ps, int depth)
 {
-       if (r_r == NULL)
+       if ( !r_u )
                return False;
 
-       prs_debug(ps, depth, desc, "reg_io_r_open_hklm");
+       prs_debug(ps, depth, desc, "reg_io_r_open_hive");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!smb_io_pol_hnd("", &r_r->pol, ps, depth))
+       if(!smb_io_pol_hnd("", &r_u->pol, ps, depth))
                return False;
 
-       if(!prs_uint32("status", ps, depth, &r_r->status))
+       if(!prs_werror("status", ps, depth, &r_u->status))
                return False;
 
        return True;
 }
 
-
 /*******************************************************************
  Inits a structure.
 ********************************************************************/
@@ -109,39 +122,40 @@ void init_reg_q_flush_key(REG_Q_FLUSH_KEY *q_u, POLICY_HND *pol)
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_q_flush_key(char *desc,  REG_Q_FLUSH_KEY *r_q, prs_struct *ps, int depth)
+BOOL reg_io_q_flush_key(const char *desc,  REG_Q_FLUSH_KEY *q_u, prs_struct *ps, int depth)
 {
-       if (r_q == NULL)
+       if ( !q_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_q_flush_key");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!smb_io_pol_hnd("", &r_q->pol, ps, depth))
+       if(!smb_io_pol_hnd("", &q_u->pol, ps, depth))
                return False;
 
        return True;
 }
 
 /*******************************************************************
-reads or writes a structure.
+Unmarshalls a registry key flush response
 ********************************************************************/
 
-BOOL reg_io_r_flush_key(char *desc,  REG_R_FLUSH_KEY *r_r, prs_struct *ps, int depth)
+BOOL reg_io_r_flush_key(const char *desc,  REG_R_FLUSH_KEY *r_u,
+                        prs_struct *ps, int depth)
 {
-       if (r_r == NULL)
+       if ( !r_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_r_flush_key");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!prs_uint32("status", ps, depth, &r_r->status))
+       if(!prs_werror("status", ps, depth, &r_u->status))
                return False;
 
        return True;
@@ -151,12 +165,14 @@ BOOL reg_io_r_flush_key(char *desc,  REG_R_FLUSH_KEY *r_r, prs_struct *ps, int d
 reads or writes SEC_DESC_BUF and SEC_DATA structures.
 ********************************************************************/
 
-static BOOL reg_io_hdrbuf_sec(uint32 ptr, uint32 *ptr3, BUFHDR *hdr_sec, SEC_DESC_BUF *data, prs_struct *ps, int depth)
+static BOOL reg_io_hdrbuf_sec(uint32 ptr, uint32 *ptr3, BUFHDR *hdr_sec,
+                              SEC_DESC_BUF *data, prs_struct *ps, int depth)
 {
        if (ptr != 0) {
                uint32 hdr_offset;
                uint32 old_offset;
-               if(!smb_io_hdrbuf_pre("hdr_sec", hdr_sec, ps, depth, &hdr_offset))
+               if(!smb_io_hdrbuf_pre("hdr_sec", hdr_sec, ps, depth,
+                                     &hdr_offset))
                        return False;
 
                old_offset = prs_offset(ps);
@@ -167,17 +183,19 @@ static BOOL reg_io_hdrbuf_sec(uint32 ptr, uint32 *ptr3, BUFHDR *hdr_sec, SEC_DES
                }
 
                if (ptr3 == NULL || *ptr3 != 0) {
-                       if(!sec_io_desc_buf("data   ", &data, ps, depth)) /* JRA - this line is probably wrong... */
+                       /* JRA - this next line is probably wrong... */
+                       if(!sec_io_desc_buf("data   ", &data, ps, depth))
                                return False;
                }
 
-               if(!smb_io_hdrbuf_post("hdr_sec", hdr_sec, ps, depth, hdr_offset,
-                                  data->max_len, data->len))
+               if(!smb_io_hdrbuf_post("hdr_sec", hdr_sec, ps, depth,
+                                      hdr_offset, data->max_len, data->len))
                                return False;
-               if(!prs_set_offset(ps, old_offset + data->len + sizeof(uint32) * ((ptr3 != NULL) ? 5 : 3)))
+               if(!prs_set_offset(ps, old_offset + data->len +
+                                      sizeof(uint32) * ((ptr3 != NULL) ? 5 : 3)))
                        return False;
 
-               if(prs_align(ps))
+               if(!prs_align(ps))
                        return False;
        }
 
@@ -185,31 +203,25 @@ static BOOL reg_io_hdrbuf_sec(uint32 ptr, uint32 *ptr3, BUFHDR *hdr_sec, SEC_DES
 }
 
 /*******************************************************************
- Inits a structure.
+ Inits a registry key create request
 ********************************************************************/
 
 void init_reg_q_create_key(REG_Q_CREATE_KEY *q_c, POLICY_HND *hnd,
-                               char *name, char *class, SEC_ACCESS *sam_access,
-                               SEC_DESC_BUF *sec_buf)
+                           char *name, char *class, uint32 access_desired,
+                           SEC_DESC_BUF *sec_buf)
 {
-       int len_name  = name  != NULL ? strlen(name ) + 1: 0;
-       int len_class = class != NULL ? strlen(class) + 1: 0;
-
        ZERO_STRUCTP(q_c);
 
        memcpy(&q_c->pnt_pol, hnd, sizeof(q_c->pnt_pol));
 
-       init_uni_hdr(&q_c->hdr_name, len_name);
-       init_unistr2(&q_c->uni_name, name, len_name);
 
-       init_uni_hdr(&q_c->hdr_class, len_class);
-       init_unistr2(&q_c->uni_class, class, len_class);
+       init_unistr4( &q_c->name, name, UNI_STR_TERMINATE );
+       init_unistr4( &q_c->class, class, UNI_STR_TERMINATE );
 
-       q_c->reserved = 0x00000000;
-       memcpy(&q_c->sam_access, sam_access, sizeof(q_c->sam_access));
+       q_c->access = access_desired;
 
-       q_c->ptr1 = 1;
-       q_c->sec_info = DACL_SECURITY_INFORMATION | SACL_SECURITY_INFORMATION;
+       q_c->sec_info = TALLOC_P( get_talloc_ctx(), uint32 );
+       *q_c->sec_info = DACL_SECURITY_INFORMATION | SACL_SECURITY_INFORMATION;
 
        q_c->data = sec_buf;
        q_c->ptr2 = 1;
@@ -219,82 +231,78 @@ void init_reg_q_create_key(REG_Q_CREATE_KEY *q_c, POLICY_HND *hnd,
 }
 
 /*******************************************************************
-reads or writes a structure.
+Marshalls a registry key create request
 ********************************************************************/
 
-BOOL reg_io_q_create_key(char *desc,  REG_Q_CREATE_KEY *r_q, prs_struct *ps, int depth)
+BOOL reg_io_q_create_key(const char *desc,  REG_Q_CREATE_KEY *q_u,
+                         prs_struct *ps, int depth)
 {
-       if (r_q == NULL)
+       if ( !q_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_q_create_key");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!smb_io_pol_hnd("", &r_q->pnt_pol, ps, depth))
+       if(!smb_io_pol_hnd("", &q_u->pnt_pol, ps, depth))
                return False;
 
-       if(!smb_io_unihdr ("", &r_q->hdr_name, ps, depth))
-               return False;
-       if(!smb_io_unistr2("", &r_q->uni_name, r_q->hdr_name.buffer, ps, depth))
+       if(!prs_unistr4 ("name", ps, depth, &q_u->name))
                return False;
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
 
-       if(!smb_io_unihdr ("", &r_q->hdr_class, ps, depth))
-               return False;
-       if(!smb_io_unistr2("", &r_q->uni_class, r_q->hdr_class.buffer, ps, depth))
+       if(!prs_unistr4 ("class", ps, depth, &q_u->class))
                return False;
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
 
-       if(!prs_uint32("reserved", ps, depth, &r_q->reserved))
+       if(!prs_uint32("reserved", ps, depth, &q_u->reserved))
                return False;
-       if(!sec_io_access("sam_access", &r_q->sam_access, ps, depth))
+       if(!prs_uint32("access", ps, depth, &q_u->access))
                return False;
 
-       if(!prs_uint32("ptr1", ps, depth, &r_q->ptr1))
+       if(!prs_pointer("sec_info", ps, depth, (void**)&q_u->sec_info, sizeof(uint32), (PRS_POINTER_CAST)prs_uint32))
                return False;
 
-       if (r_q->ptr1 != 0) {
-               if(!prs_uint32("sec_info", ps, depth, &r_q->sec_info))
-                       return False;
-       }
-
-       if(!prs_uint32("ptr2", ps, depth, &r_q->ptr2))
+       if(!prs_uint32("ptr2", ps, depth, &q_u->ptr2))
                return False;
-       if(!reg_io_hdrbuf_sec(r_q->ptr2, &r_q->ptr3, &r_q->hdr_sec, r_q->data, ps, depth))
+       if(!reg_io_hdrbuf_sec(q_u->ptr2, &q_u->ptr3, &q_u->hdr_sec, q_u->data,
+                             ps, depth))
                return False;
 
-       if(!prs_uint32("unknown_2", ps, depth, &r_q->unknown_2))
+#if 0
+       if(!prs_uint32("unknown_2", ps, depth, &q_u->unknown_2))
                return False;
+#endif
 
        return True;
 }
 
 /*******************************************************************
-reads or writes a structure.
+Unmarshalls a registry key create response
 ********************************************************************/
 
-BOOL reg_io_r_create_key(char *desc,  REG_R_CREATE_KEY *r_r, prs_struct *ps, int depth)
+BOOL reg_io_r_create_key(const char *desc,  REG_R_CREATE_KEY *r_u,
+                         prs_struct *ps, int depth)
 {
-       if (r_r == NULL)
+       if ( !r_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_r_create_key");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!smb_io_pol_hnd("", &r_r->key_pol, ps, depth))
+       if(!smb_io_pol_hnd("", &r_u->key_pol, ps, depth))
                return False;
-       if(!prs_uint32("unknown", ps, depth, &r_r->unknown))
+       if(!prs_uint32("unknown", ps, depth, &r_u->unknown))
                return False;
 
-       if(!prs_uint32("status", ps, depth, &r_r->status))
+       if(!prs_werror("status", ps, depth, &r_u->status))
                return False;
 
        return True;
@@ -306,40 +314,36 @@ BOOL reg_io_r_create_key(char *desc,  REG_R_CREATE_KEY *r_r, prs_struct *ps, int
 ********************************************************************/
 
 void init_reg_q_delete_val(REG_Q_DELETE_VALUE *q_c, POLICY_HND *hnd,
-                               char *name)
+                           char *name)
 {
-       int len_name  = name  != NULL ? strlen(name ) + 1: 0;
        ZERO_STRUCTP(q_c);
 
        memcpy(&q_c->pnt_pol, hnd, sizeof(q_c->pnt_pol));
-
-       init_uni_hdr(&q_c->hdr_name, len_name);
-       init_unistr2(&q_c->uni_name, name, len_name);
+       init_unistr4(&q_c->name, name, UNI_STR_TERMINATE);
 }
 
 /*******************************************************************
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_q_delete_val(char *desc,  REG_Q_DELETE_VALUE *r_q, prs_struct *ps, int depth)
+BOOL reg_io_q_delete_val(const char *desc, REG_Q_DELETE_VALUE *q_u,
+                         prs_struct *ps, int depth)
 {
-       if (r_q == NULL)
+       if ( !q_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_q_delete_val");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!smb_io_pol_hnd("", &r_q->pnt_pol, ps, depth))
+       if(!smb_io_pol_hnd("", &q_u->pnt_pol, ps, depth))
                return False;
 
-       if(!smb_io_unihdr ("", &r_q->hdr_name, ps, depth))
-               return False;
-       if(!smb_io_unistr2("", &r_q->uni_name, r_q->hdr_name.buffer, ps, depth))
+       if(!prs_unistr4("name", ps, depth, &q_u->name))
                return False;
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
 
        return True;
@@ -350,18 +354,19 @@ BOOL reg_io_q_delete_val(char *desc,  REG_Q_DELETE_VALUE *r_q, prs_struct *ps, i
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_r_delete_val(char *desc,  REG_R_DELETE_VALUE *r_r, prs_struct *ps, int depth)
+BOOL reg_io_r_delete_val(const char *desc,  REG_R_DELETE_VALUE *r_u,
+                         prs_struct *ps, int depth)
 {
-       if (r_r == NULL)
+       if ( !r_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_r_delete_val");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!prs_uint32("status", ps, depth, &r_r->status))
+       if(!prs_werror("status", ps, depth, &r_u->status))
                return False;
 
        return True;
@@ -372,40 +377,37 @@ BOOL reg_io_r_delete_val(char *desc,  REG_R_DELETE_VALUE *r_r, prs_struct *ps, i
 ********************************************************************/
 
 void init_reg_q_delete_key(REG_Q_DELETE_KEY *q_c, POLICY_HND *hnd,
-                               char *name)
+                           char *name)
 {
-       int len_name  = name  != NULL ? strlen(name ) + 1: 0;
        ZERO_STRUCTP(q_c);
 
        memcpy(&q_c->pnt_pol, hnd, sizeof(q_c->pnt_pol));
 
-       init_uni_hdr(&q_c->hdr_name, len_name);
-       init_unistr2(&q_c->uni_name, name, len_name);
+       init_unistr4(&q_c->name, name, UNI_STR_TERMINATE);
 }
 
 /*******************************************************************
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_q_delete_key(char *desc,  REG_Q_DELETE_KEY *r_q, prs_struct *ps, int depth)
+BOOL reg_io_q_delete_key(const char *desc,  REG_Q_DELETE_KEY *q_u,
+                         prs_struct *ps, int depth)
 {
-       if (r_q == NULL)
+       if ( !q_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_q_delete_key");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!smb_io_pol_hnd("", &r_q->pnt_pol, ps, depth))
+       if(!smb_io_pol_hnd("", &q_u->pnt_pol, ps, depth))
                return False;
 
-       if(!smb_io_unihdr ("", &r_q->hdr_name, ps, depth))
-               return False;
-       if(!smb_io_unistr2("", &r_q->uni_name, r_q->hdr_name.buffer, ps, depth))
+       if(!prs_unistr4("", ps, depth, &q_u->name))
                return False;
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
 
        return True;
@@ -415,18 +417,18 @@ BOOL reg_io_q_delete_key(char *desc,  REG_Q_DELETE_KEY *r_q, prs_struct *ps, int
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_r_delete_key(char *desc,  REG_R_DELETE_KEY *r_r, prs_struct *ps, int depth)
+BOOL reg_io_r_delete_key(const char *desc,  REG_R_DELETE_KEY *r_u, prs_struct *ps, int depth)
 {
-       if (r_r == NULL)
+       if ( !r_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_r_delete_key");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!prs_uint32("status", ps, depth, &r_r->status))
+       if(!prs_werror("status", ps, depth, &r_u->status))
                return False;
 
        return True;
@@ -436,39 +438,35 @@ BOOL reg_io_r_delete_key(char *desc,  REG_R_DELETE_KEY *r_r, prs_struct *ps, int
  Inits a structure.
 ********************************************************************/
 
-void init_reg_q_query_key(REG_Q_QUERY_KEY *q_o, POLICY_HND *hnd,
-                               uint32 max_class_len)
+void init_reg_q_query_key(REG_Q_QUERY_KEY *q_o, POLICY_HND *hnd, const char *class)
 {
        ZERO_STRUCTP(q_o);
 
        memcpy(&q_o->pol, hnd, sizeof(q_o->pol));
-       init_uni_hdr(&q_o->hdr_class, max_class_len);
-       q_o->uni_class.uni_max_len = max_class_len;
+       init_unistr4(&q_o->class, class, UNI_STR_TERMINATE);
 }
 
 /*******************************************************************
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_q_query_key(char *desc,  REG_Q_QUERY_KEY *r_q, prs_struct *ps, int depth)
+BOOL reg_io_q_query_key(const char *desc,  REG_Q_QUERY_KEY *q_u, prs_struct *ps, int depth)
 {
-       if (r_q == NULL)
+       if ( !q_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_q_query_key");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!smb_io_pol_hnd("", &r_q->pol, ps, depth))
-               return False;
-       if(!smb_io_unihdr ("", &r_q->hdr_class, ps, depth))
+       if(!smb_io_pol_hnd("", &q_u->pol, ps, depth))
                return False;
-       if(!smb_io_unistr2("", &r_q->uni_class, r_q->hdr_class.buffer, ps, depth))
+       if(!prs_unistr4("class", ps, depth, &q_u->class))
                return False;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
 
        return True;
@@ -479,43 +477,41 @@ BOOL reg_io_q_query_key(char *desc,  REG_Q_QUERY_KEY *r_q, prs_struct *ps, int d
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_r_query_key(char *desc,  REG_R_QUERY_KEY *r_r, prs_struct *ps, int depth)
+BOOL reg_io_r_query_key(const char *desc,  REG_R_QUERY_KEY *r_u, prs_struct *ps, int depth)
 {
-       if (r_r == NULL)
+       if ( !r_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_r_query_key");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!smb_io_unihdr ("", &r_r->hdr_class, ps, depth))
-               return False;
-       if(!smb_io_unistr2("", &r_r->uni_class, r_r->hdr_class.buffer, ps, depth))
+       if(!prs_unistr4("class", ps, depth, &r_u->class))
                return False;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
 
-       if(!prs_uint32("num_subkeys   ", ps, depth, &r_r->num_subkeys))
+       if(!prs_uint32("num_subkeys   ", ps, depth, &r_u->num_subkeys))
                return False;
-       if(!prs_uint32("max_subkeylen ", ps, depth, &r_r->max_subkeylen))
+       if(!prs_uint32("max_subkeylen ", ps, depth, &r_u->max_subkeylen))
                return False;
-       if(!prs_uint32("mak_subkeysize", ps, depth, &r_r->max_subkeysize))
+       if(!prs_uint32("reserved      ", ps, depth, &r_u->reserved))
                return False;
-       if(!prs_uint32("num_values    ", ps, depth, &r_r->num_values))
+       if(!prs_uint32("num_values    ", ps, depth, &r_u->num_values))
                return False;
-       if(!prs_uint32("max_valnamelen", ps, depth, &r_r->max_valnamelen))
+       if(!prs_uint32("max_valnamelen", ps, depth, &r_u->max_valnamelen))
                return False;
-       if(!prs_uint32("max_valbufsize", ps, depth, &r_r->max_valbufsize))
+       if(!prs_uint32("max_valbufsize", ps, depth, &r_u->max_valbufsize))
                return False;
-       if(!prs_uint32("sec_desc      ", ps, depth, &r_r->sec_desc))
+       if(!prs_uint32("sec_desc      ", ps, depth, &r_u->sec_desc))
                return False;
-       if(!smb_io_time("mod_time     ", &r_r->mod_time, ps, depth))
+       if(!smb_io_time("mod_time     ", &r_u->mod_time, ps, depth))
                return False;
-       
-       if(!prs_uint32("status", ps, depth, &r_r->status))
+
+       if(!prs_werror("status", ps, depth, &r_u->status))
                return False;
 
        return True;
@@ -525,27 +521,28 @@ BOOL reg_io_r_query_key(char *desc,  REG_R_QUERY_KEY *r_r, prs_struct *ps, int d
  Inits a structure.
 ********************************************************************/
 
-void init_reg_q_unk_1a(REG_Q_UNK_1A *q_o, POLICY_HND *hnd)
+void init_reg_q_getversion(REG_Q_GETVERSION *q_o, POLICY_HND *hnd)
 {
        memcpy(&q_o->pol, hnd, sizeof(q_o->pol));
 }
 
+
 /*******************************************************************
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_q_unk_1a(char *desc,  REG_Q_UNK_1A *r_q, prs_struct *ps, int depth)
+BOOL reg_io_q_getversion(const char *desc,  REG_Q_GETVERSION *q_u, prs_struct *ps, int depth)
 {
-       if (r_q == NULL)
+       if ( !q_u )
                return False;
 
-       prs_debug(ps, depth, desc, "reg_io_q_unk_1a");
+       prs_debug(ps, depth, desc, "reg_io_q_getversion");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
-       
-       if(!smb_io_pol_hnd("", &r_q->pol, ps, depth))
+
+       if(!smb_io_pol_hnd("", &q_u->pol, ps, depth))
                return False;
 
        return True;
@@ -555,63 +552,70 @@ BOOL reg_io_q_unk_1a(char *desc,  REG_Q_UNK_1A *r_q, prs_struct *ps, int depth)
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_r_unk_1a(char *desc,  REG_R_UNK_1A *r_r, prs_struct *ps, int depth)
+BOOL reg_io_r_getversion(const char *desc,  REG_R_GETVERSION *r_u, prs_struct *ps, int depth)
 {
-       if (r_r == NULL)
+       if ( !r_u )
                return False;
 
-       prs_debug(ps, depth, desc, "reg_io_r_unk_1a");
+       prs_debug(ps, depth, desc, "reg_io_r_getversion");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
-       
-       if(!prs_uint32("unknown", ps, depth, &r_r->unknown))
+
+       if(!prs_uint32("unknown", ps, depth, &r_u->unknown))
                return False;
-       if(!prs_uint32("status" , ps, depth, &r_r->status))
+       if(!prs_werror("status" , ps, depth, &r_u->status))
                return False;
 
        return True;
 }
 
+
 /*******************************************************************
- Inits a structure.
+reads or writes a structure.
 ********************************************************************/
 
-void init_reg_q_open_hku(REG_Q_OPEN_HKU *q_o,
-                               uint16 unknown_0, uint32 level)
+BOOL reg_io_q_restore_key(const char *desc,  REG_Q_RESTORE_KEY *q_u, prs_struct *ps, int depth)
 {
-       q_o->ptr = 1;
-       q_o->unknown_0 = unknown_0;
-       q_o->unknown_1 = 0x0; /* random - changes */
-       q_o->level = level;
+       if ( !q_u )
+               return False;
+
+       prs_debug(ps, depth, desc, "reg_io_q_restore_key");
+       depth++;
+
+       if(!prs_align(ps))
+               return False;
+
+       if(!smb_io_pol_hnd("", &q_u->pol, ps, depth))
+               return False;
+
+       if(!prs_unistr4("filename", ps, depth, &q_u->filename))
+               return False;
+
+       if(!prs_uint32("flags", ps, depth, &q_u->flags))
+               return False;
+
+       return True;
 }
 
 /*******************************************************************
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_q_open_hku(char *desc,  REG_Q_OPEN_HKU *r_q, prs_struct *ps, int depth)
+BOOL reg_io_r_restore_key(const char *desc,  REG_R_RESTORE_KEY *r_u, prs_struct *ps, int depth)
 {
-       if (r_q == NULL)
+       if ( !r_u )
                return False;
 
-       prs_debug(ps, depth, desc, "reg_io_q_open_hku");
+       prs_debug(ps, depth, desc, "reg_io_r_restore_key");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!prs_uint32("ptr      ", ps, depth, &r_q->ptr))
+       if(!prs_werror("status" , ps, depth, &r_u->status))
                return False;
-       if (r_q->ptr != 0) {
-               if(!prs_uint16("unknown_0", ps, depth, &r_q->unknown_0))
-                       return False;
-               if(!prs_uint16("unknown_1", ps, depth, &r_q->unknown_1))
-                       return False;
-               if(!prs_uint32("level    ", ps, depth, &r_q->level))
-                       return False;
-       }
 
        return True;
 }
@@ -620,21 +624,47 @@ BOOL reg_io_q_open_hku(char *desc,  REG_Q_OPEN_HKU *r_q, prs_struct *ps, int dep
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_r_open_hku(char *desc,  REG_R_OPEN_HKU *r_r, prs_struct *ps, int depth)
+BOOL reg_io_q_save_key(const char *desc,  REG_Q_SAVE_KEY *q_u, prs_struct *ps, int depth)
 {
-       if (r_r == NULL)
+       if ( !q_u )
                return False;
 
-       prs_debug(ps, depth, desc, "reg_io_r_open_hku");
+       prs_debug(ps, depth, desc, "reg_io_q_save_key");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
-       
-       if(!smb_io_pol_hnd("", &r_r->pol, ps, depth))
+
+       if(!smb_io_pol_hnd("", &q_u->pol, ps, depth))
+               return False;
+
+       if(!prs_unistr4("filename", ps, depth, &q_u->filename))
+               return False;
+
+#if 0  /* reg_io_sec_attr() */
+       if(!prs_uint32("unknown", ps, depth, &q_u->unknown))
+               return False;
+#endif
+
+       return True;
+}
+
+/*******************************************************************
+reads or writes a structure.
+********************************************************************/
+
+BOOL reg_io_r_save_key(const char *desc,  REG_R_SAVE_KEY *r_u, prs_struct *ps, int depth)
+{
+       if ( !r_u )
                return False;
 
-       if(!prs_uint32("status", ps, depth, &r_r->status))
+       prs_debug(ps, depth, desc, "reg_io_r_save_key");
+       depth++;
+
+       if(!prs_align(ps))
+               return False;
+       
+       if(!prs_werror("status" , ps, depth, &r_u->status))
                return False;
 
        return True;
@@ -655,20 +685,20 @@ void init_reg_q_close(REG_Q_CLOSE *q_c, POLICY_HND *hnd)
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_q_close(char *desc,  REG_Q_CLOSE *q_u, prs_struct *ps, int depth)
+BOOL reg_io_q_close(const char *desc,  REG_Q_CLOSE *q_u, prs_struct *ps, int depth)
 {
        if (q_u == NULL)
                return False;
 
-       prs_debug(ps, depth, desc, "reg_io_q_unknown_1");
+       prs_debug(ps, depth, desc, "reg_io_q_close");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
 
        if(!smb_io_pol_hnd("", &q_u->pol, ps, depth))
                return False;
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
 
        return True;
@@ -678,23 +708,23 @@ BOOL reg_io_q_close(char *desc,  REG_Q_CLOSE *q_u, prs_struct *ps, int depth)
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_r_close(char *desc,  REG_R_CLOSE *r_u, prs_struct *ps, int depth)
+BOOL reg_io_r_close(const char *desc,  REG_R_CLOSE *r_u, prs_struct *ps, int depth)
 {
-       if (r_u == NULL)
+       if ( !r_u )
                return False;
 
-       prs_debug(ps, depth, desc, "reg_io_r_unknown_1");
+       prs_debug(ps, depth, desc, "reg_io_r_close");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
 
        if(!smb_io_pol_hnd("", &r_u->pol, ps, depth))
                return False;
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
 
-       if(!prs_uint32("status", ps, depth, &r_u->status))
+       if(!prs_werror("status", ps, depth, &r_u->status))
                return False;
 
        return True;
@@ -704,41 +734,42 @@ BOOL reg_io_r_close(char *desc,  REG_R_CLOSE *r_u, prs_struct *ps, int depth)
 makes a structure.
 ********************************************************************/
 
-void init_reg_q_set_key_sec(REG_Q_SET_KEY_SEC *q_i, POLICY_HND *pol, SEC_DESC_BUF *sec_desc_buf)
+void init_reg_q_set_key_sec(REG_Q_SET_KEY_SEC *q_u, POLICY_HND *pol,
+                            uint32 sec_info, SEC_DESC_BUF *sec_desc_buf)
 {
-       memcpy(&q_i->pol, pol, sizeof(q_i->pol));
+       memcpy(&q_u->pol, pol, sizeof(q_u->pol));
 
-       q_i->sec_info = DACL_SECURITY_INFORMATION;
+       q_u->sec_info = sec_info;
 
-       q_i->ptr = 1;
-       init_buf_hdr(&q_i->hdr_sec, sec_desc_buf->len, sec_desc_buf->len);
-       q_i->data = sec_desc_buf;
+       q_u->ptr = 1;
+       init_buf_hdr(&q_u->hdr_sec, sec_desc_buf->len, sec_desc_buf->len);
+       q_u->data = sec_desc_buf;
 }
 
 /*******************************************************************
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_q_set_key_sec(char *desc,  REG_Q_SET_KEY_SEC *r_q, prs_struct *ps, int depth)
+BOOL reg_io_q_set_key_sec(const char *desc,  REG_Q_SET_KEY_SEC *q_u, prs_struct *ps, int depth)
 {
-       if (r_q == NULL)
+       if ( !q_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_q_set_key_sec");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!smb_io_pol_hnd("", &r_q->pol, ps, depth))
+       if(!smb_io_pol_hnd("", &q_u->pol, ps, depth))
                return False;
 
-       if(!prs_uint32("sec_info", ps, depth, &r_q->sec_info))
+       if(!prs_uint32("sec_info", ps, depth, &q_u->sec_info))
                return False;
-       if(!prs_uint32("ptr    ", ps, depth, &r_q->ptr))
+       if(!prs_uint32("ptr    ", ps, depth, &q_u->ptr))
                return False;
 
-       if(!reg_io_hdrbuf_sec(r_q->ptr, NULL, &r_q->hdr_sec, r_q->data, ps, depth))
+       if(!reg_io_hdrbuf_sec(q_u->ptr, NULL, &q_u->hdr_sec, q_u->data, ps, depth))
                return False;
 
        return True;
@@ -748,18 +779,18 @@ BOOL reg_io_q_set_key_sec(char *desc,  REG_Q_SET_KEY_SEC *r_q, prs_struct *ps, i
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_r_set_key_sec(char *desc, REG_R_SET_KEY_SEC *r_q, prs_struct *ps, int depth)
+BOOL reg_io_r_set_key_sec(const char *desc, REG_R_SET_KEY_SEC *q_u, prs_struct *ps, int depth)
 {
-       if (r_q == NULL)
+       if ( !q_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_r_set_key_sec");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!prs_uint32("status", ps, depth, &r_q->status))
+       if(!prs_werror("status", ps, depth, &q_u->status))
                return False;
 
        return True;
@@ -770,45 +801,44 @@ BOOL reg_io_r_set_key_sec(char *desc, REG_R_SET_KEY_SEC *r_q, prs_struct *ps, in
 makes a structure.
 ********************************************************************/
 
-void init_reg_q_get_key_sec(REG_Q_GET_KEY_SEC *q_i, POLICY_HND *pol, 
-                               uint32 sec_buf_size, SEC_DESC_BUF *psdb)
+void init_reg_q_get_key_sec(REG_Q_GET_KEY_SEC *q_u, POLICY_HND *pol, 
+                            uint32 sec_info, uint32 sec_buf_size,
+                            SEC_DESC_BUF *psdb)
 {
-       memcpy(&q_i->pol, pol, sizeof(q_i->pol));
+       memcpy(&q_u->pol, pol, sizeof(q_u->pol));
 
-       q_i->sec_info = OWNER_SECURITY_INFORMATION |
-                       GROUP_SECURITY_INFORMATION |
-                       DACL_SECURITY_INFORMATION;
+       q_u->sec_info = sec_info;
 
-       q_i->ptr = psdb != NULL ? 1 : 0;
-       q_i->data = psdb;
+       q_u->ptr = psdb != NULL ? 1 : 0;
+       q_u->data = psdb;
 
-       init_buf_hdr(&q_i->hdr_sec, sec_buf_size, 0);
+       init_buf_hdr(&q_u->hdr_sec, sec_buf_size, 0);
 }
 
 /*******************************************************************
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_q_get_key_sec(char *desc,  REG_Q_GET_KEY_SEC *r_q, prs_struct *ps, int depth)
+BOOL reg_io_q_get_key_sec(const char *desc,  REG_Q_GET_KEY_SEC *q_u, prs_struct *ps, int depth)
 {
-       if (r_q == NULL)
+       if ( !q_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_q_get_key_sec");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!smb_io_pol_hnd("", &r_q->pol, ps, depth))
+       if(!smb_io_pol_hnd("", &q_u->pol, ps, depth))
                return False;
 
-       if(!prs_uint32("sec_info", ps, depth, &r_q->sec_info))
+       if(!prs_uint32("sec_info", ps, depth, &q_u->sec_info))
                return False;
-       if(!prs_uint32("ptr     ", ps, depth, &r_q->ptr))
+       if(!prs_uint32("ptr     ", ps, depth, &q_u->ptr))
                return False;
 
-       if(!reg_io_hdrbuf_sec(r_q->ptr, NULL, &r_q->hdr_sec, r_q->data, ps, depth))
+       if(!reg_io_hdrbuf_sec(q_u->ptr, NULL, &q_u->hdr_sec, q_u->data, ps, depth))
                return False;
 
        return True;
@@ -820,7 +850,7 @@ makes a structure.
 ********************************************************************/
  void init_reg_r_get_key_sec(REG_R_GET_KEY_SEC *r_i, POLICY_HND *pol, 
                                uint32 buf_len, uint8 *buf,
-                               uint32 status)
+                               NTSTATUS status)
 {
        r_i->ptr = 1;
        init_buf_hdr(&r_i->hdr_sec, buf_len, buf_len);
@@ -834,30 +864,30 @@ makes a structure.
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_r_get_key_sec(char *desc,  REG_R_GET_KEY_SEC *r_q, prs_struct *ps, int depth)
+BOOL reg_io_r_get_key_sec(const char *desc,  REG_R_GET_KEY_SEC *q_u, prs_struct *ps, int depth)
 {
-       if (r_q == NULL)
+       if ( !q_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_r_get_key_sec");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!prs_uint32("ptr      ", ps, depth, &r_q->ptr))
+       if(!prs_uint32("ptr      ", ps, depth, &q_u->ptr))
                return False;
 
-       if (r_q->ptr != 0) {
-               if(!smb_io_hdrbuf("", &r_q->hdr_sec, ps, depth))
+       if (q_u->ptr != 0) {
+               if(!smb_io_hdrbuf("", &q_u->hdr_sec, ps, depth))
                        return False;
-               if(!sec_io_desc_buf("", &r_q->data, ps, depth))
+               if(!sec_io_desc_buf("", &q_u->data, ps, depth))
                        return False;
-               if(prs_align(ps))
+               if(!prs_align(ps))
                        return False;
        }
 
-       if(!prs_uint32("status", ps, depth, &r_q->status))
+       if(!prs_werror("status", ps, depth, &q_u->status))
                return False;
 
        return True;
@@ -867,255 +897,265 @@ BOOL reg_io_r_get_key_sec(char *desc,  REG_R_GET_KEY_SEC *r_q, prs_struct *ps, i
 makes a structure.
 ********************************************************************/
 
-void init_reg_q_info(REG_Q_INFO *q_i, POLICY_HND *pol, char *product_type,
-                               time_t unix_time, uint8 major, uint8 minor)
+BOOL init_reg_q_info(REG_Q_INFO *q_u, POLICY_HND *pol, const char *val_name,
+                     REGVAL_BUFFER *value_output)
 {
-       int len_type  = strlen(product_type);
+        if (q_u == NULL)
+                return False;
+
+        q_u->pol = *pol;
+
+        init_unistr4(&q_u->name, val_name, UNI_STR_TERMINATE);
 
-       memcpy(&q_i->pol, pol, sizeof(q_i->pol));
+        q_u->ptr_reserved = 1;
+        q_u->ptr_buf = 1;
 
-       init_uni_hdr(&q_i->hdr_type, len_type);
-       init_unistr2(&q_i->uni_type, product_type, len_type);
+        q_u->ptr_bufsize = 1;
+        q_u->bufsize = value_output->buf_max_len;
+        q_u->buf_unk = 0;
 
-       q_i->ptr1 = 1;
-       unix_to_nt_time(&q_i->time, unix_time);
-       q_i->major_version1 = major;
-       q_i->minor_version1 = minor;
-       memset(q_i->pad1, 0, sizeof(q_i->pad1));
+        q_u->unk1 = 0;
+        q_u->ptr_buflen = 1;
+        q_u->buflen = value_output->buf_max_len; 
 
-       q_i->ptr2 = 1;
-       q_i->major_version2 = major;
-       q_i->minor_version2 = minor;
-       memset(q_i->pad2, 0, sizeof(q_i->pad2));
+        q_u->ptr_buflen2 = 1;
+        q_u->buflen2 = 0;
 
-       q_i->ptr3 = 1;
-       q_i->unknown = 0x00000000;
+        return True;
 }
 
 /*******************************************************************
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_q_info(char *desc,  REG_Q_INFO *r_q, prs_struct *ps, int depth)
+BOOL reg_io_q_info(const char *desc,  REG_Q_INFO *q_u, prs_struct *ps, int depth)
 {
-       if (r_q == NULL)
+       if ( !q_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_q_info");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!smb_io_pol_hnd("", &r_q->pol, ps, depth))
-               return False;
-       if(!smb_io_unihdr ("", &r_q->hdr_type, ps, depth))
+       if(!smb_io_pol_hnd("", &q_u->pol, ps, depth))
                return False;
-       if(!smb_io_unistr2("", &r_q->uni_type, r_q->hdr_type.buffer, ps, depth))
+       if(!prs_unistr4("name", ps, depth, &q_u->name))
                return False;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!prs_uint32("ptr1", ps, depth, &r_q->ptr1))
+       if(!prs_uint32("ptr_reserved", ps, depth, &(q_u->ptr_reserved)))
                return False;
 
-       if (r_q->ptr1 != 0) {
-               if(!smb_io_time("", &r_q->time, ps, depth))
-                       return False;
-               if(!prs_uint8 ("major_version1", ps, depth, &r_q->major_version1))
-                       return False;
-               if(!prs_uint8 ("minor_version1", ps, depth, &r_q->minor_version1))
-                       return False;
-               if(!prs_uint8s(False, "pad1", ps, depth, r_q->pad1, sizeof(r_q->pad1)))
-                       return False;
-       }
-
-       if(!prs_uint32("ptr2", ps, depth, &r_q->ptr2))
+       if(!prs_uint32("ptr_buf", ps, depth, &(q_u->ptr_buf)))
                return False;
 
-       if (r_q->ptr2 != 0) {
-               if(!prs_uint8 ("major_version2", ps, depth, &r_q->major_version2))
+       if(q_u->ptr_buf) {
+               if(!prs_uint32("ptr_bufsize", ps, depth, &(q_u->ptr_bufsize)))
                        return False;
-               if(!prs_uint8 ("minor_version2", ps, depth, &r_q->minor_version2))
+               if(!prs_uint32("bufsize", ps, depth, &(q_u->bufsize)))
                        return False;
-               if(!prs_uint8s(False, "pad2", ps, depth, r_q->pad2, sizeof(r_q->pad2)))
+               if(!prs_uint32("buf_unk", ps, depth, &(q_u->buf_unk)))
                        return False;
        }
 
-       if(!prs_uint32("ptr3", ps, depth, &r_q->ptr3))
+       if(!prs_uint32("unk1", ps, depth, &(q_u->unk1)))
                return False;
 
-       if (r_q->ptr3 != 0) {
-               if(!prs_uint32("unknown", ps, depth, &r_q->unknown))
+       if(!prs_uint32("ptr_buflen", ps, depth, &(q_u->ptr_buflen)))
+               return False;
+
+       if (q_u->ptr_buflen) {
+               if(!prs_uint32("buflen", ps, depth, &(q_u->buflen)))
+                       return False;
+               if(!prs_uint32("ptr_buflen2", ps, depth, &(q_u->ptr_buflen2)))
+                       return False;
+               if(!prs_uint32("buflen2", ps, depth, &(q_u->buflen2)))
                        return False;
        }
 
-       return True;
+       return True;
 }
 
 /*******************************************************************
  Inits a structure.
+ New version to replace older init_reg_r_info()
 ********************************************************************/
 
-void init_reg_r_info(REG_R_INFO *r_r,
-                               uint32 level, char *os_type,
-                               uint32 unknown_0, uint32 unknown_1,
-                               uint32 status)
+BOOL init_reg_r_info(uint32 include_keyval, REG_R_INFO *r_u,
+                    REGISTRY_VALUE *val, WERROR status)
 {
-       uint8 buf[512];
-       int len = dos_struni2((char *)buf, os_type, sizeof(buf));
+       uint32                  buf_len = 0;
+       REGVAL_BUFFER           buf2;
+               
+       if( !r_u || !val )
+               return False;
+       
+       r_u->type = TALLOC_P( get_talloc_ctx(), uint32 );
+       *r_u->type = val->type;
 
-       r_r->ptr1 = 1;
-       r_r->level = level;
+       buf_len = reg_init_regval_buffer( &buf2, val );
+       
+       r_u->buf_max_len = TALLOC_P( get_talloc_ctx(), uint32 );
+       *r_u->buf_max_len = buf_len;
 
-       r_r->ptr_type = 1;
-       init_buffer2(&r_r->uni_type, buf, len*2);
+       r_u->buf_len = TALLOC_P( get_talloc_ctx(), uint32 );
+       *r_u->buf_len = buf_len;
+       
+       /* if include_keyval is not set, don't send the key value, just
+          the buflen data. probably used by NT5 to allocate buffer space - SK */
 
-       r_r->ptr2 = 1;
-       r_r->unknown_0 = unknown_0;
+       if ( include_keyval ) {
+               r_u->value = TALLOC_P( get_talloc_ctx(), REGVAL_BUFFER );
+               /* steal the memory */
+               *r_u->value = buf2;
+       }
 
-       r_r->ptr3 = 1;
-       r_r->unknown_1 = unknown_1;
+       r_u->status = status;
 
-       r_r->status = status;
+       return True;
 }
 
 /*******************************************************************
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_r_info(char *desc, REG_R_INFO *r_r, prs_struct *ps, int depth)
+BOOL reg_io_r_info(const char *desc, REG_R_INFO *r_u, prs_struct *ps, int depth)
 {
-       if (r_r == NULL)
+       if ( !r_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_r_info");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!prs_uint32("ptr1", ps, depth, &r_r->ptr1))
+       if ( !prs_pointer("type", ps, depth, (void**)&r_u->type, sizeof(uint32), (PRS_POINTER_CAST)prs_uint32))
                return False;
 
-       if (r_r->ptr1 != 0) {
-               if(!prs_uint32("level", ps, depth, &r_r->level))
-                       return False;
-               if(!prs_uint32("ptr_type", ps, depth, &r_r->ptr_type))
-                       return False;
+       if ( !prs_pointer("value", ps, depth, (void**)&r_u->value, sizeof(REGVAL_BUFFER), (PRS_POINTER_CAST)smb_io_regval_buffer))
+               return False;
+       if(!prs_align(ps))
+               return False;
 
-               if(!smb_io_buffer2("uni_type", &r_r->uni_type, r_r->ptr_type, ps, depth))
-                       return False;
-               if(prs_align(ps))
-                       return False;
+       if ( !prs_pointer("buf_max_len", ps, depth, (void**)&r_u->buf_max_len, sizeof(uint32), (PRS_POINTER_CAST)prs_uint32))
+               return False;
+       if ( !prs_pointer("buf_len", ps, depth, (void**)&r_u->buf_len, sizeof(uint32), (PRS_POINTER_CAST)prs_uint32))
+               return False;
 
-               if(!prs_uint32("ptr2", ps, depth, &r_r->ptr2))
-                       return False;
+       if(!prs_werror("status", ps, depth, &r_u->status))
+               return False;
 
-               if (r_r->ptr2 != 0) {
-                       if(!prs_uint32("unknown_0", ps, depth, &r_r->unknown_0))
-                               return False;
-               }
+       return True;
+}
 
-               if(!prs_uint32("ptr3", ps, depth, &r_r->ptr3))
-                       return False;
+/*******************************************************************
+makes a structure.
+********************************************************************/
 
-               if (r_r->ptr3 != 0) {
-                       if(!prs_uint32("unknown_1", ps, depth, &r_r->unknown_1))
-                               return False;
-               }
+void init_reg_q_enum_val(REG_Q_ENUM_VALUE *q_u, POLICY_HND *pol,
+                               uint32 val_idx,
+                               uint32 max_name_len, uint32 max_buf_len)
+{
+       ZERO_STRUCTP(q_u);
 
-       }
-       if(!prs_uint32("status", ps, depth, &r_r->status))
-               return False;
+       memcpy(&q_u->pol, pol, sizeof(q_u->pol));
 
-       return True;
+       q_u->val_index = val_idx;
+
+       q_u->name.size = max_name_len*2;
+       q_u->name.string = TALLOC_ZERO_P( get_talloc_ctx(), UNISTR2 );
+       q_u->name.string->uni_max_len = max_name_len;
+       
+       q_u->type = TALLOC_P( get_talloc_ctx(), uint32 );
+       *q_u->type = 0x0;
+
+       q_u->value = TALLOC_ZERO_P( get_talloc_ctx(), REGVAL_BUFFER );
+       q_u->value->buf_max_len = max_buf_len;
+
+       q_u->buffer_len  = TALLOC_P( get_talloc_ctx(), uint32 );
+       *q_u->buffer_len = max_buf_len;
+
+       q_u->name_len  = TALLOC_P( get_talloc_ctx(), uint32 );
+       *q_u->name_len = 0x0;
 }
 
 /*******************************************************************
 makes a structure.
 ********************************************************************/
 
-void init_reg_q_enum_val(REG_Q_ENUM_VALUE *q_i, POLICY_HND *pol,
-                               uint32 val_idx, uint32 max_val_len,
-                               uint32 max_buf_len)
+void init_reg_r_enum_val(REG_R_ENUM_VALUE *r_u, REGISTRY_VALUE *val )
 {
-       ZERO_STRUCTP(q_i);
+       uint32 real_size;
+       
+       ZERO_STRUCTP(r_u);
 
-       memcpy(&q_i->pol, pol, sizeof(q_i->pol));
+       /* value name */
 
-       q_i->val_index = val_idx;
-       init_uni_hdr(&q_i->hdr_name, max_val_len);
-       q_i->uni_name.uni_max_len = max_val_len;
+       DEBUG(10,("init_reg_r_enum_val: Valuename => [%s]\n", val->valuename));
        
-       q_i->ptr_type = 1;
-       q_i->type = 0x0;
-
-       q_i->ptr_value = 1;
-       q_i->buf_value.buf_max_len = max_buf_len;
+       init_unistr4( &r_u->name, val->valuename, UNI_STR_TERMINATE);
+               
+       /* type */
+       
+       r_u->type = TALLOC_P( get_talloc_ctx(), uint32 );
+       *r_u->type = val->type;
 
-       q_i->ptr1 = 1;
-       q_i->len_value1 = max_buf_len;
+       /* REG_SZ & REG_MULTI_SZ must be converted to UNICODE */
+       
+       r_u->value = TALLOC_P( get_talloc_ctx(), REGVAL_BUFFER );
+       real_size = reg_init_regval_buffer( r_u->value, val );
+       
+       /* lengths */
 
-       q_i->ptr2 = 1;
-       q_i->len_value2 = 0;
+       r_u->buffer_len1  = TALLOC_P( get_talloc_ctx(), uint32 );
+       *r_u->buffer_len1 = real_size;
+       r_u->buffer_len2  = TALLOC_P( get_talloc_ctx(), uint32 );
+       *r_u->buffer_len2 = real_size;
 }
 
 /*******************************************************************
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_q_enum_val(char *desc,  REG_Q_ENUM_VALUE *q_q, prs_struct *ps, int depth)
+BOOL reg_io_q_enum_val(const char *desc,  REG_Q_ENUM_VALUE *q_u, prs_struct *ps, int depth)
 {
-       if (q_q == NULL)
+       if (q_u == NULL)
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_q_enum_val");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!smb_io_pol_hnd("", &q_q->pol, ps, depth))
+       if(!smb_io_pol_hnd("", &q_u->pol, ps, depth))
                return False;
        
-       if(!prs_uint32("val_index", ps, depth, &q_q->val_index))
+       if(!prs_uint32("val_index", ps, depth, &q_u->val_index))
                return False;
-       if(!smb_io_unihdr ("hdr_name", &q_q->hdr_name, ps, depth))
+               
+       if(!prs_unistr4("name", ps, depth, &q_u->name ))
                return False;
-       if(!smb_io_unistr2("uni_name", &q_q->uni_name, q_q->hdr_name.buffer, ps, depth))
-               return False;
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
 
-       if(!prs_uint32("ptr_type", ps, depth, &q_q->ptr_type))
+       if(!prs_pointer("type", ps, depth, (void**)&q_u->type, sizeof(uint32), (PRS_POINTER_CAST)prs_uint32))
                return False;
 
-       if (q_q->ptr_type != 0) {
-               if(!prs_uint32("type", ps, depth, &q_q->type))
-                       return False;
-       }
-
-       if(!prs_uint32("ptr_value", ps, depth, &q_q->ptr_value))
-               return False;
-       if(!smb_io_buffer2("buf_value", &q_q->buf_value, q_q->ptr_value, ps, depth))
+       if ( !prs_pointer("value", ps, depth, (void**)&q_u->value, sizeof(REGVAL_BUFFER), (PRS_POINTER_CAST)smb_io_regval_buffer))
                return False;
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
 
-       if(!prs_uint32("ptr1", ps, depth, &q_q->ptr1))
+       if(!prs_pointer("buffer_len", ps, depth, (void**)&q_u->buffer_len, sizeof(uint32), (PRS_POINTER_CAST)prs_uint32))
                return False;
-       if (q_q->ptr1 != 0) {
-               if(!prs_uint32("len_value1", ps, depth, &q_q->len_value1))
-                       return False;
-       }
-       if(!prs_uint32("ptr2", ps, depth, &q_q->ptr2))
+       if(!prs_pointer("name_len", ps, depth, (void**)&q_u->name_len, sizeof(uint32), (PRS_POINTER_CAST)prs_uint32))
                return False;
-       if (q_q->ptr2 != 0) {
-               if(!prs_uint32("len_value2", ps, depth, &q_q->len_value2))
-                       return False;
-       }
 
        return True;
 }
@@ -1124,54 +1164,36 @@ BOOL reg_io_q_enum_val(char *desc,  REG_Q_ENUM_VALUE *q_q, prs_struct *ps, int d
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_r_enum_val(char *desc,  REG_R_ENUM_VALUE *r_q, prs_struct *ps, int depth)
+BOOL reg_io_r_enum_val(const char *desc,  REG_R_ENUM_VALUE *r_u, prs_struct *ps, int depth)
 {
-       if (r_q == NULL)
+       if ( !r_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_r_enum_val");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!smb_io_unihdr ("hdr_name", &r_q->hdr_name, ps, depth))
+       if(!prs_unistr4("name", ps, depth, &r_u->name ))
                return False;
-       if(!smb_io_unistr2("uni_name", &r_q->uni_name, r_q->hdr_name.buffer, ps, depth))
-               return False;
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
 
-       if(!prs_uint32("ptr_type", ps, depth, &r_q->ptr_type))
+       if(!prs_pointer("type", ps, depth, (void**)&r_u->type, sizeof(uint32), (PRS_POINTER_CAST)prs_uint32))
                return False;
 
-       if (r_q->ptr_type != 0) {
-               if(!prs_uint32("type", ps, depth, &r_q->type))
-                       return False;
-       }
-
-       if(!prs_uint32("ptr_value", ps, depth, &r_q->ptr_value))
-               return False;
-       if(!smb_io_buffer2("buf_value", r_q->buf_value, r_q->ptr_value, ps, depth))
+       if ( !prs_pointer("value", ps, depth, (void**)&r_u->value, sizeof(REGVAL_BUFFER), (PRS_POINTER_CAST)smb_io_regval_buffer))
                return False;
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
 
-       if(!prs_uint32("ptr1", ps, depth, &r_q->ptr1))
+       if(!prs_pointer("buffer_len1", ps, depth, (void**)&r_u->buffer_len1, sizeof(uint32), (PRS_POINTER_CAST)prs_uint32))
                return False;
-       if (r_q->ptr1 != 0) {
-               if(!prs_uint32("len_value1", ps, depth, &r_q->len_value1))
-                       return False;
-       }
-
-       if(!prs_uint32("ptr2", ps, depth, &r_q->ptr2))
+       if(!prs_pointer("buffer_len2", ps, depth, (void**)&r_u->buffer_len2, sizeof(uint32), (PRS_POINTER_CAST)prs_uint32))
                return False;
-       if (r_q->ptr2 != 0) {
-               if(!prs_uint32("len_value2", ps, depth, &r_q->len_value2))
-                       return False;
-       }
 
-       if(!prs_uint32("status", ps, depth, &r_q->status))
+       if(!prs_werror("status", ps, depth, &r_u->status))
                return False;
 
        return True;
@@ -1181,53 +1203,53 @@ BOOL reg_io_r_enum_val(char *desc,  REG_R_ENUM_VALUE *r_q, prs_struct *ps, int d
 makes a structure.
 ********************************************************************/
 
-void init_reg_q_create_val(REG_Q_CREATE_VALUE *q_i, POLICY_HND *pol,
+void init_reg_q_set_val(REG_Q_SET_VALUE *q_u, POLICY_HND *pol,
                                char *val_name, uint32 type,
-                               BUFFER3 *val)
+                               RPC_DATA_BLOB *val)
 {
-       int val_len = strlen(val_name) + 1;
+       ZERO_STRUCTP(q_u);
 
-       ZERO_STRUCTP(q_i);
-
-       memcpy(&q_i->pol, pol, sizeof(q_i->pol));
+       memcpy(&q_u->pol, pol, sizeof(q_u->pol));
 
-       init_uni_hdr(&q_i->hdr_name, val_len);
-       init_unistr2(&q_i->uni_name, val_name, val_len);
+       init_unistr4(&q_u->name, val_name, UNI_STR_TERMINATE);
        
-       q_i->type      = type;
-       q_i->buf_value = val;
+       q_u->type      = type;
+       q_u->value     = *val;
+       q_u->size      = val->buf_len;
 }
 
 /*******************************************************************
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_q_create_val(char *desc,  REG_Q_CREATE_VALUE *q_q, prs_struct *ps, int depth)
+BOOL reg_io_q_set_val(const char *desc,  REG_Q_SET_VALUE *q_u, prs_struct *ps, int depth)
 {
-       if (q_q == NULL)
+       if (q_u == NULL)
                return False;
 
-       prs_debug(ps, depth, desc, "reg_io_q_create_val");
+       prs_debug(ps, depth, desc, "reg_io_q_set_val");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!smb_io_pol_hnd("", &q_q->pol, ps, depth))
+       if(!smb_io_pol_hnd("", &q_u->pol, ps, depth))
                return False;
        
-       if(!smb_io_unihdr ("hdr_name", &q_q->hdr_name, ps, depth))
+       if(!prs_unistr4("name", ps, depth, &q_u->name ))
                return False;
-       if(!smb_io_unistr2("uni_name", &q_q->uni_name, q_q->hdr_name.buffer, ps, depth))
+       if(!prs_align(ps))
                return False;
-       if(prs_align(ps))
+
+       if(!prs_uint32("type", ps, depth, &q_u->type))
                return False;
 
-       if(!prs_uint32("type", ps, depth, &q_q->type))
+       if(!smb_io_rpc_blob("value", &q_u->value, ps, depth ))
                return False;
-       if(!smb_io_buffer3("buf_value", q_q->buf_value, ps, depth))
+       if(!prs_align(ps))
                return False;
-       if(prs_align(ps))
+
+       if(!prs_uint32("size", ps, depth, &q_u->size))
                return False;
 
        return True;
@@ -1237,18 +1259,18 @@ BOOL reg_io_q_create_val(char *desc,  REG_Q_CREATE_VALUE *q_q, prs_struct *ps, i
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_r_create_val(char *desc,  REG_R_CREATE_VALUE *r_q, prs_struct *ps, int depth)
+BOOL reg_io_r_set_val(const char *desc,  REG_R_SET_VALUE *q_u, prs_struct *ps, int depth)
 {
-       if (r_q == NULL)
+       if ( !q_u )
                return False;
 
-       prs_debug(ps, depth, desc, "reg_io_r_create_val");
+       prs_debug(ps, depth, desc, "reg_io_r_set_val");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!prs_uint32("status", ps, depth, &r_q->status))
+       if(!prs_werror("status", ps, depth, &q_u->status))
                return False;
 
        return True;
@@ -1258,73 +1280,87 @@ BOOL reg_io_r_create_val(char *desc,  REG_R_CREATE_VALUE *r_q, prs_struct *ps, i
 makes a structure.
 ********************************************************************/
 
-void init_reg_q_enum_key(REG_Q_ENUM_KEY *q_i, POLICY_HND *pol, uint32 key_idx)
+void init_reg_q_enum_key(REG_Q_ENUM_KEY *q_u, POLICY_HND *pol, uint32 key_idx)
 {
-       memcpy(&q_i->pol, pol, sizeof(q_i->pol));
+       memcpy(&q_u->pol, pol, sizeof(q_u->pol));
 
-       q_i->key_index = key_idx;
-       q_i->key_name_len = 0;
-       q_i->unknown_1 = 0x0414;
+       q_u->key_index = key_idx;
+       q_u->key_name_len = 0;
+       q_u->unknown_1 = 0x0414;
 
-       q_i->ptr1 = 1;
-       q_i->unknown_2 = 0x0000020A;
-       memset(q_i->pad1, 0, sizeof(q_i->pad1));
+       q_u->ptr1 = 1;
+       q_u->unknown_2 = 0x0000020A;
+       memset(q_u->pad1, 0, sizeof(q_u->pad1));
 
-       q_i->ptr2 = 1;
-       memset(q_i->pad2, 0, sizeof(q_i->pad2));
+       q_u->ptr2 = 1;
+       memset(q_u->pad2, 0, sizeof(q_u->pad2));
 
-       q_i->ptr3 = 1;
-       unix_to_nt_time(&q_i->time, 0);            /* current time? */
+       q_u->ptr3 = 1;
+       unix_to_nt_time(&q_u->time, 0);            /* current time? */
+}
+
+/*******************************************************************
+makes a reply structure.
+********************************************************************/
+
+void init_reg_r_enum_key(REG_R_ENUM_KEY *r_u, char *subkey )
+{
+       if ( !r_u )
+               return;
+               
+       init_unistr4( &r_u->keyname, subkey, STR_TERMINATE );
+       r_u->classname = TALLOC_ZERO_P( get_talloc_ctx(), UNISTR4 );
+       r_u->time       = TALLOC_ZERO_P( get_talloc_ctx(), NTTIME );
 }
 
 /*******************************************************************
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_q_enum_key(char *desc,  REG_Q_ENUM_KEY *q_q, prs_struct *ps, int depth)
+BOOL reg_io_q_enum_key(const char *desc,  REG_Q_ENUM_KEY *q_u, prs_struct *ps, int depth)
 {
-       if (q_q == NULL)
+       if (q_u == NULL)
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_q_enum_key");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
                return False;
        
-       if(!smb_io_pol_hnd("", &q_q->pol, ps, depth))
+       if(!smb_io_pol_hnd("", &q_u->pol, ps, depth))
                return False;
        
-       if(!prs_uint32("key_index", ps, depth, &q_q->key_index))
+       if(!prs_uint32("key_index", ps, depth, &q_u->key_index))
                return False;
-       if(!prs_uint16("key_name_len", ps, depth, &q_q->key_name_len))
+       if(!prs_uint16("key_name_len", ps, depth, &q_u->key_name_len))
                return False;
-       if(!prs_uint16("unknown_1", ps, depth, &q_q->unknown_1))
+       if(!prs_uint16("unknown_1", ps, depth, &q_u->unknown_1))
                return False;
 
-       if(!prs_uint32("ptr1", ps, depth, &q_q->ptr1))
+       if(!prs_uint32("ptr1", ps, depth, &q_u->ptr1))
                return False;
 
-       if (q_q->ptr1 != 0) {
-               if(!prs_uint32("unknown_2", ps, depth, &q_q->unknown_2))
+       if (q_u->ptr1 != 0) {
+               if(!prs_uint32("unknown_2", ps, depth, &q_u->unknown_2))
                        return False;
-               if(!prs_uint8s(False, "pad1", ps, depth, q_q->pad1, sizeof(q_q->pad1)))
+               if(!prs_uint8s(False, "pad1", ps, depth, q_u->pad1, sizeof(q_u->pad1)))
                        return False;
        }
 
-       if(!prs_uint32("ptr2", ps, depth, &q_q->ptr2))
+       if(!prs_uint32("ptr2", ps, depth, &q_u->ptr2))
                return False;
 
-       if (q_q->ptr2 != 0) {
-               if(!prs_uint8s(False, "pad2", ps, depth, q_q->pad2, sizeof(q_q->pad2)))
+       if (q_u->ptr2 != 0) {
+               if(!prs_uint8s(False, "pad2", ps, depth, q_u->pad2, sizeof(q_u->pad2)))
                        return False;
        }
 
-       if(!prs_uint32("ptr3", ps, depth, &q_q->ptr3))
+       if(!prs_uint32("ptr3", ps, depth, &q_u->ptr3))
                return False;
 
-       if (q_q->ptr3 != 0) {
-               if(!smb_io_time("", &q_q->time, ps, depth))
+       if (q_u->ptr3 != 0) {
+               if(!smb_io_time("", &q_u->time, ps, depth))
                        return False;
        }
 
@@ -1335,139 +1371,334 @@ BOOL reg_io_q_enum_key(char *desc,  REG_Q_ENUM_KEY *q_q, prs_struct *ps, int dep
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_r_enum_key(char *desc,  REG_R_ENUM_KEY *r_q, prs_struct *ps, int depth)
+BOOL reg_io_r_enum_key(const char *desc,  REG_R_ENUM_KEY *q_u, prs_struct *ps, int depth)
 {
-       if (r_q == NULL)
+       if ( !q_u )
                return False;
 
        prs_debug(ps, depth, desc, "reg_io_r_enum_key");
        depth++;
 
-       if(prs_align(ps))
+       if(!prs_align(ps))
+               return False;
+       if ( !prs_unistr4( "keyname", ps, depth, &q_u->keyname ) )
                return False;
        
-       if(!prs_uint16("key_name_len", ps, depth, &r_q->key_name_len))
+       if(!prs_align(ps))
                return False;
-       if(!prs_uint16("unknown_1", ps, depth, &r_q->unknown_1))
+       if (!prs_pointer("class", ps, depth, (void**)&q_u->classname, sizeof(UNISTR4), (PRS_POINTER_CAST)prs_unistr4))
                return False;
 
-       if(!prs_uint32("ptr1", ps, depth, &r_q->ptr1))
+       if(!prs_align(ps))
+               return False;
+       if (!prs_pointer("time", ps, depth, (void**)&q_u->time, sizeof(NTTIME), (PRS_POINTER_CAST)smb_io_nttime))
                return False;
 
-       if (r_q->ptr1 != 0) {
-               if(!prs_uint32("unknown_2", ps, depth, &r_q->unknown_2))
-                       return False;
-               if(!prs_uint32("unknown_3", ps, depth, &r_q->unknown_3))
-                       return False;
-               if(!smb_io_unistr3("key_name", &r_q->key_name, ps, depth))
-                       return False;
-               if(prs_align(ps))
-                       return False;
-       }
+       if(!prs_align(ps))
+               return False;
+       if(!prs_werror("status", ps, depth, &q_u->status))
+               return False;
 
-       if(!prs_uint32("ptr2", ps, depth, &r_q->ptr2))
+       return True;
+}
+
+/*******************************************************************
+makes a structure.
+********************************************************************/
+
+void init_reg_q_open_entry(REG_Q_OPEN_ENTRY *q_u, POLICY_HND *pol,
+                               char *key_name, uint32 access_desired)
+{
+       memcpy(&q_u->pol, pol, sizeof(q_u->pol));
+
+       init_unistr4(&q_u->name, key_name, UNI_STR_TERMINATE);
+
+       q_u->unknown_0 = 0x00000000;
+       q_u->access = access_desired;
+}
+
+/*******************************************************************
+reads or writes a structure.
+********************************************************************/
+
+BOOL reg_io_q_open_entry(const char *desc,  REG_Q_OPEN_ENTRY *q_u, prs_struct *ps, int depth)
+{
+       if ( !q_u )
                return False;
 
-       if (r_q->ptr2 != 0) {
-               if(!prs_uint8s(False, "pad2", ps, depth, r_q->pad2, sizeof(r_q->pad2)))
-                       return False;
-       }
+       prs_debug(ps, depth, desc, "reg_io_q_open_entry");
+       depth++;
 
-       if(!prs_uint32("ptr3", ps, depth, &r_q->ptr3))
+       if(!prs_align(ps))
+               return False;
+       
+       if(!smb_io_pol_hnd("", &q_u->pol, ps, depth))
+               return False;
+       if(!prs_unistr4("name", ps, depth, &q_u->name))
                return False;
 
-       if (r_q->ptr3 != 0) {
-               if(!smb_io_time("", &r_q->time, ps, depth))
-                       return False;
+       if(!prs_align(ps))
+               return False;
+       
+       if(!prs_uint32("unknown_0        ", ps, depth, &q_u->unknown_0))
+               return False;
+       if(!prs_uint32("access", ps, depth, &q_u->access))
+               return False;
+
+       return True;
+}
+
+/*******************************************************************
+ Inits a structure.
+********************************************************************/
+
+void init_reg_r_open_entry(REG_R_OPEN_ENTRY *r_u,
+                          POLICY_HND *pol, WERROR werr)
+{
+       if (W_ERROR_IS_OK(werr)) {
+               memcpy(&r_u->pol, pol, sizeof(r_u->pol));
+       } else {
+               ZERO_STRUCT(r_u->pol);
        }
+       r_u->status = werr;
+}
+
+/*******************************************************************
+reads or writes a structure.
+********************************************************************/
 
-       if(!prs_uint32("status", ps, depth, &r_q->status))
+BOOL reg_io_r_open_entry(const char *desc,  REG_R_OPEN_ENTRY *r_u, prs_struct *ps, int depth)
+{
+       if ( !r_u )
+               return False;
+
+       prs_debug(ps, depth, desc, "reg_io_r_open_entry");
+       depth++;
+
+       if(!prs_align(ps))
+               return False;
+       
+       if(!smb_io_pol_hnd("", &r_u->pol, ps, depth))
+               return False;
+
+       if(!prs_werror("status", ps, depth, &r_u->status))
                return False;
 
        return True;
 }
 
 /*******************************************************************
-makes a structure.
+Inits a structure.
 ********************************************************************/
 
-void init_reg_q_open_entry(REG_Q_OPEN_ENTRY *r_q, POLICY_HND *pol,
-                               char *key_name, uint32 unk)
+void init_reg_q_shutdown(REG_Q_SHUTDOWN *q_u, const char *msg,
+                       uint32 timeout, BOOL do_reboot, BOOL force)
 {
-       int len_name = strlen(key_name)+1;
+       q_u->server = TALLOC_P( get_talloc_ctx(), uint16 );
+       *q_u->server = 0x1;
 
-       memcpy(&r_q->pol, pol, sizeof(r_q->pol));
+       q_u->message = TALLOC_P( get_talloc_ctx(), UNISTR4 );
+       init_unistr4( q_u->message, msg, UNI_FLAGS_NONE );
 
-       init_uni_hdr(&r_q->hdr_name, len_name);
-       init_unistr2(&r_q->uni_name, key_name, len_name);
+       q_u->timeout = timeout;
 
-       r_q->unknown_0 = 0x00000000;
-       r_q->unknown_1 = unk;
+       q_u->reboot = do_reboot ? 1 : 0;
+       q_u->force = force ? 1 : 0;
+}
+
+/*******************************************************************
+Inits a REG_Q_SHUTDOWN_EX structure.
+********************************************************************/
+
+void init_reg_q_shutdown_ex(REG_Q_SHUTDOWN_EX * q_u_ex, const char *msg,
+                       uint32 timeout, BOOL do_reboot, BOOL force, uint32 reason)
+{
+       REG_Q_SHUTDOWN q_u;
+       
+       ZERO_STRUCT( q_u );
+       
+       init_reg_q_shutdown( &q_u, msg, timeout, do_reboot, force );
+       
+       /* steal memory */
+       
+       q_u_ex->server  = q_u.server;
+       q_u_ex->message = q_u.message;
+       
+       q_u_ex->reboot  = q_u.reboot;
+       q_u_ex->force   = q_u.force;
+       
+       q_u_ex->reason = reason;
 }
 
 /*******************************************************************
 reads or writes a structure.
 ********************************************************************/
 
-BOOL reg_io_q_open_entry(char *desc,  REG_Q_OPEN_ENTRY *r_q, prs_struct *ps, int depth)
+BOOL reg_io_q_shutdown(const char *desc, REG_Q_SHUTDOWN *q_u, prs_struct *ps,
+                      int depth)
 {
-       if (r_q == NULL)
+       if ( !q_u )
                return False;
 
-       prs_debug(ps, depth, desc, "reg_io_q_entry");
+       prs_debug(ps, depth, desc, "reg_io_q_shutdown");
        depth++;
 
-       if(prs_align(ps))
+       if (!prs_align(ps))
                return False;
-       
-       if(!smb_io_pol_hnd("", &r_q->pol, ps, depth))
+
+       if (!prs_pointer("server", ps, depth, (void**)&q_u->server, sizeof(uint16), (PRS_POINTER_CAST)prs_uint16))
                return False;
-       if(!smb_io_unihdr ("", &r_q->hdr_name, ps, depth))
+
+       if (!prs_pointer("message", ps, depth, (void**)&q_u->message, sizeof(UNISTR4), (PRS_POINTER_CAST)prs_unistr4))
                return False;
-       if(!smb_io_unistr2("", &r_q->uni_name, r_q->hdr_name.buffer, ps, depth))
+
+       if (!prs_align(ps))
                return False;
 
-       if(prs_align(ps))
+       if (!prs_uint32("timeout", ps, depth, &(q_u->timeout)))
                return False;
-       
-       if(!prs_uint32("unknown_0", ps, depth, &r_q->unknown_0))
+
+       if (!prs_uint8("force  ", ps, depth, &(q_u->force)))
                return False;
-       if(!prs_uint32("unknown_1", ps, depth, &r_q->unknown_1))
+       if (!prs_uint8("reboot ", ps, depth, &(q_u->reboot)))
                return False;
 
+
        return True;
 }
 
 /*******************************************************************
- Inits a structure.
+reads or writes a structure.
+********************************************************************/
+BOOL reg_io_r_shutdown(const char *desc, REG_R_SHUTDOWN *r_u, prs_struct *ps,
+                      int depth)
+{
+       if ( !r_u )
+               return False;
+
+       prs_debug(ps, depth, desc, "reg_io_r_shutdown");
+       depth++;
+
+       if(!prs_align(ps))
+               return False;
+
+       if(!prs_werror("status", ps, depth, &r_u->status))
+               return False;
+
+       return True;
+}
+
+/*******************************************************************
+reads or writes a REG_Q_SHUTDOWN_EX structure.
+********************************************************************/
+
+BOOL reg_io_q_shutdown_ex(const char *desc, REG_Q_SHUTDOWN_EX *q_u, prs_struct *ps,
+                      int depth)
+{
+       if ( !q_u )
+               return False;
+
+       prs_debug(ps, depth, desc, "reg_io_q_shutdown_ex");
+       depth++;
+
+       if (!prs_align(ps))
+               return False;
+
+       if (!prs_pointer("server", ps, depth, (void**)&q_u->server, sizeof(uint16), (PRS_POINTER_CAST)prs_uint16))
+               return False;
+
+       if (!prs_pointer("message", ps, depth, (void**)&q_u->message, sizeof(UNISTR4), (PRS_POINTER_CAST)prs_unistr4))
+               return False;
+
+       if (!prs_align(ps))
+               return False;
+
+       if (!prs_uint32("timeout", ps, depth, &(q_u->timeout)))
+               return False;
+
+       if (!prs_uint8("force  ", ps, depth, &(q_u->force)))
+               return False;
+       if (!prs_uint8("reboot ", ps, depth, &(q_u->reboot)))
+               return False;
+
+       if (!prs_align(ps))
+               return False;
+       if (!prs_uint32("reason", ps, depth, &(q_u->reason)))
+               return False;
+
+
+       return True;
+}
+
+/*******************************************************************
+reads or writes a REG_R_SHUTDOWN_EX structure.
 ********************************************************************/
+BOOL reg_io_r_shutdown_ex(const char *desc, REG_R_SHUTDOWN_EX *r_u, prs_struct *ps,
+                      int depth)
+{
+       if ( !r_u )
+               return False;
+
+       prs_debug(ps, depth, desc, "reg_io_r_shutdown_ex");
+       depth++;
+
+       if(!prs_align(ps))
+               return False;
+
+       if(!prs_werror("status", ps, depth, &r_u->status))
+               return False;
+
+       return True;
+}
 
-void init_reg_r_open_entry(REG_R_OPEN_ENTRY *r_r,
-                               POLICY_HND *pol, uint32 status)
+
+
+/*******************************************************************
+Inits a structure.
+********************************************************************/
+void init_reg_q_abort_shutdown(REG_Q_ABORT_SHUTDOWN *q_u)
 {
-       memcpy(&r_r->pol, pol, sizeof(r_r->pol));
-       r_r->status = status;
+       q_u->server = TALLOC_P( get_talloc_ctx(), uint16 );
+       *q_u->server = 0x1;
 }
 
 /*******************************************************************
 reads or writes a structure.
 ********************************************************************/
-
-BOOL reg_io_r_open_entry(char *desc,  REG_R_OPEN_ENTRY *r_r, prs_struct *ps, int depth)
+BOOL reg_io_q_abort_shutdown(const char *desc, REG_Q_ABORT_SHUTDOWN *q_u,
+                            prs_struct *ps, int depth)
 {
-       if (r_r == NULL)
+       if ( !q_u )
                return False;
 
-       prs_debug(ps, depth, desc, "reg_io_r_open_entry");
+       prs_debug(ps, depth, desc, "reg_io_q_abort_shutdown");
        depth++;
 
-       if(prs_align(ps))
+       if (!prs_align(ps))
                return False;
-       
-       if(!smb_io_pol_hnd("", &r_r->pol, ps, depth))
+
+       if (!prs_pointer("server", ps, depth, (void**)&q_u->server, sizeof(uint16), (PRS_POINTER_CAST)prs_uint16))
+               return False;
+
+       return True;
+}
+
+/*******************************************************************
+reads or writes a structure.
+********************************************************************/
+BOOL reg_io_r_abort_shutdown(const char *desc, REG_R_ABORT_SHUTDOWN *r_u,
+                            prs_struct *ps, int depth)
+{
+       if ( !r_u )
+               return False;
+
+       prs_debug(ps, depth, desc, "reg_io_r_abort_shutdown");
+       depth++;
+
+       if (!prs_align(ps))
                return False;
 
-       if(!prs_uint32("status", ps, depth, &r_r->status))
+       if (!prs_werror("status", ps, depth, &r_u->status))
                return False;
 
        return True;