Line data Source code
1 : /*
2 : * Unix SMB/CIFS implementation.
3 : * Registry helper routines
4 : * Copyright (C) Volker Lendecke 2006
5 : * Copyright (C) Guenther Deschner 2009
6 : * Copyright (C) Jelmer Vernooij 2003-2007
7 : *
8 : * This program is free software; you can redistribute it and/or modify it
9 : * under the terms of the GNU General Public License as published by the Free
10 : * Software Foundation; either version 3 of the License, or (at your option)
11 : * any later version.
12 : *
13 : * This program is distributed in the hope that it will be useful, but WITHOUT
14 : * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
15 : * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
16 : * more details.
17 : *
18 : * You should have received a copy of the GNU General Public License along with
19 : * this program; if not, see <http://www.gnu.org/licenses/>.
20 : */
21 :
22 : #include "includes.h"
23 : #include "../librpc/gen_ndr/ndr_misc.h"
24 : #include "../libcli/registry/util_reg.h"
25 :
26 : /**
27 : * @file
28 : * @brief Registry utility functions
29 : */
30 :
31 : static const struct {
32 : uint32_t id;
33 : const char *name;
34 : } reg_value_types[] = {
35 : { REG_NONE, "REG_NONE" },
36 : { REG_SZ, "REG_SZ" },
37 : { REG_EXPAND_SZ, "REG_EXPAND_SZ" },
38 : { REG_BINARY, "REG_BINARY" },
39 : { REG_DWORD, "REG_DWORD" },
40 : { REG_DWORD_BIG_ENDIAN, "REG_DWORD_BIG_ENDIAN" },
41 : { REG_LINK, "REG_LINK" },
42 : { REG_MULTI_SZ, "REG_MULTI_SZ" },
43 : { REG_RESOURCE_LIST, "REG_RESOURCE_LIST" },
44 : { REG_FULL_RESOURCE_DESCRIPTOR, "REG_FULL_RESOURCE_DESCRIPTOR" },
45 : { REG_RESOURCE_REQUIREMENTS_LIST, "REG_RESOURCE_REQUIREMENTS_LIST" },
46 : { REG_QWORD, "REG_QWORD" },
47 :
48 : { 0, NULL }
49 : };
50 :
51 : /** Return string description of registry value type */
52 4005 : _PUBLIC_ const char *str_regtype(int type)
53 : {
54 : unsigned int i;
55 23822 : for (i = 0; reg_value_types[i].name; i++) {
56 23374 : if (reg_value_types[i].id == type)
57 3542 : return reg_value_types[i].name;
58 : }
59 :
60 448 : return "Unknown";
61 : }
62 :
63 : /** Return registry value type for string description */
64 580 : _PUBLIC_ int regtype_by_string(const char *str)
65 : {
66 : unsigned int i;
67 3294 : for (i = 0; reg_value_types[i].name; i++) {
68 3100 : if (strequal(reg_value_types[i].name, str))
69 386 : return reg_value_types[i].id;
70 : }
71 :
72 171 : return -1;
73 : }
74 :
75 : /*******************************************************************
76 : push a string in unix charset into a REG_SZ UCS2 null terminated blob
77 : ********************************************************************/
78 :
79 9827 : bool push_reg_sz(TALLOC_CTX *mem_ctx, DATA_BLOB *blob, const char *s)
80 : {
81 : union winreg_Data data;
82 : enum ndr_err_code ndr_err;
83 9827 : data.string = s;
84 9827 : ndr_err = ndr_push_union_blob(blob, mem_ctx, &data, REG_SZ,
85 : (ndr_push_flags_fn_t)ndr_push_winreg_Data);
86 9827 : return NDR_ERR_CODE_IS_SUCCESS(ndr_err);
87 : }
88 :
89 : /*******************************************************************
90 : push a string_array in unix charset into a REG_MULTI_SZ UCS2 double-null
91 : terminated blob
92 : ********************************************************************/
93 :
94 3863 : bool push_reg_multi_sz(TALLOC_CTX *mem_ctx, DATA_BLOB *blob, const char **a)
95 : {
96 : union winreg_Data data;
97 : enum ndr_err_code ndr_err;
98 3863 : data.string_array = a;
99 3863 : ndr_err = ndr_push_union_blob(blob, mem_ctx, &data, REG_MULTI_SZ,
100 : (ndr_push_flags_fn_t)ndr_push_winreg_Data);
101 3863 : return NDR_ERR_CODE_IS_SUCCESS(ndr_err);
102 : }
103 :
104 : /*******************************************************************
105 : pull a string in unix charset out of a REG_SZ UCS2 null terminated blob
106 : ********************************************************************/
107 :
108 110553 : bool pull_reg_sz(TALLOC_CTX *mem_ctx, const DATA_BLOB *blob, const char **s)
109 : {
110 : union winreg_Data data;
111 : enum ndr_err_code ndr_err;
112 110553 : ndr_err = ndr_pull_union_blob(blob, mem_ctx, &data, REG_SZ,
113 : (ndr_pull_flags_fn_t)ndr_pull_winreg_Data);
114 110553 : if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
115 0 : return false;
116 : }
117 110553 : *s = data.string;
118 110553 : return true;
119 : }
120 :
121 : /*******************************************************************
122 : pull a string_array in unix charset out of a REG_MULTI_SZ UCS2 double-null
123 : terminated blob
124 : ********************************************************************/
125 :
126 10 : bool pull_reg_multi_sz(TALLOC_CTX *mem_ctx,
127 : const DATA_BLOB *blob, const char ***a)
128 : {
129 : union winreg_Data data;
130 : enum ndr_err_code ndr_err;
131 10 : ndr_err = ndr_pull_union_blob(blob, mem_ctx, &data, REG_MULTI_SZ,
132 : (ndr_pull_flags_fn_t)ndr_pull_winreg_Data);
133 10 : if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
134 0 : return false;
135 : }
136 10 : *a = data.string_array;
137 10 : return true;
138 : }
|
