2 * Unix SMB/CIFS implementation.
3 * Registry helper routines
4 * Copyright (C) Volker Lendecke 2006
6 * This program is free software; you can redistribute it and/or modify it
7 * under the terms of the GNU General Public License as published by the Free
8 * Software Foundation; either version 3 of the License, or (at your option)
11 * This program is distributed in the hope that it will be useful, but WITHOUT
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
16 * You should have received a copy of the GNU General Public License along with
17 * this program; if not, see <http://www.gnu.org/licenses/>.
22 #include "reg_util_marshalling.h"
25 #define DBGC_CLASS DBGC_REGISTRY
27 WERROR
registry_pull_value(TALLOC_CTX
*mem_ctx
,
28 struct registry_value
**pvalue
,
29 enum winreg_Type type
, uint8
*data
,
30 uint32 size
, uint32 length
)
32 struct registry_value
*value
;
35 if (!(value
= TALLOC_ZERO_P(mem_ctx
, struct registry_value
))) {
43 case REG_DWORD_BIG_ENDIAN
:
44 if ((size
!= 4) || (length
!= 4)) {
45 err
= WERR_INVALID_PARAM
;
48 value
->v
.dword
= IVAL(data
, 0);
51 if ((size
!= 8) || (length
!= 8)) {
52 err
= WERR_INVALID_PARAM
;
55 value
->v
.qword
= BVAL(data
, 0);
61 * Make sure we get a NULL terminated string for
62 * convert_string_talloc().
68 /* win2k regedit gives us a string of 1 byte when
69 * creating a new value of type REG_SZ. this workaround
70 * replaces the input by using the same string as
73 if (!(tmp
= SMB_MALLOC_ARRAY(smb_ucs2_t
, 2))) {
79 DEBUG(10, ("got REG_SZ value of length 1 - workaround "
82 else if ((length
% 2) != 0) {
83 err
= WERR_INVALID_PARAM
;
87 uint32 num_ucs2
= length
/ 2;
88 if (!(tmp
= SMB_MALLOC_ARRAY(smb_ucs2_t
, num_ucs2
+1))) {
93 memcpy((void *)tmp
, (const void *)data
, length
);
96 if (length
+ 2 < length
) {
99 err
= WERR_INVALID_PARAM
;
103 if (!convert_string_talloc(value
, CH_UTF16LE
, CH_UNIX
, tmp
,
104 length
+2, (void *)&value
->v
.sz
.str
,
105 &value
->v
.sz
.len
, False
)) {
107 err
= WERR_INVALID_PARAM
;
119 blob
= data_blob_const(data
, length
);
121 if (!pull_reg_multi_sz(mem_ctx
, &blob
, &vals
)) {
126 for (i
=0; vals
[i
]; i
++) {
130 value
->v
.multi_sz
.num_strings
= i
;
131 value
->v
.multi_sz
.strings
= (char **)vals
;
136 value
->v
.binary
= data_blob_talloc(mem_ctx
, data
, length
);
139 err
= WERR_INVALID_PARAM
;
151 WERROR
registry_push_value(TALLOC_CTX
*mem_ctx
,
152 const struct registry_value
*value
,
155 switch (value
->type
) {
157 case REG_DWORD_BIG_ENDIAN
: {
159 SIVAL(buf
, 0, value
->v
.dword
);
160 *presult
= data_blob_talloc(mem_ctx
, (void *)buf
, 4);
161 if (presult
->data
== NULL
) {
168 SBVAL(buf
, 0, value
->v
.qword
);
169 *presult
= data_blob_talloc(mem_ctx
, (void *)buf
, 8);
170 if (presult
->data
== NULL
) {
176 case REG_EXPAND_SZ
: {
177 if (!push_reg_sz(mem_ctx
, presult
, value
->v
.sz
.str
))
184 /* handle the case where we don't get a NULL terminated array */
188 array
= talloc_array(mem_ctx
, const char *,
189 value
->v
.multi_sz
.num_strings
+ 1);
194 for (i
=0; i
< value
->v
.multi_sz
.num_strings
; i
++) {
195 array
[i
] = value
->v
.multi_sz
.strings
[i
];
199 if (!push_reg_multi_sz(mem_ctx
, presult
, array
)) {
207 *presult
= data_blob_talloc(mem_ctx
,
208 value
->v
.binary
.data
,
209 value
->v
.binary
.length
);
212 return WERR_INVALID_PARAM
;