diff options
-rw-r--r-- | lib/util/base64.c | 141 | ||||
-rw-r--r-- | lib/util/util.h | 13 | ||||
-rw-r--r-- | lib/util/util_str.c | 2 | ||||
-rwxr-xr-x | lib/util/wscript_build | 2 | ||||
-rw-r--r-- | source3/Makefile.in | 2 | ||||
-rw-r--r-- | source3/include/proto.h | 3 | ||||
-rw-r--r-- | source3/lib/util_str.c | 115 | ||||
-rw-r--r-- | source4/utils/ntlm_auth.c | 31 |
8 files changed, 158 insertions, 151 deletions
diff --git a/lib/util/base64.c b/lib/util/base64.c new file mode 100644 index 0000000000..19ce2d1b85 --- /dev/null +++ b/lib/util/base64.c @@ -0,0 +1,141 @@ +/* + Unix SMB/CIFS implementation. + Samba utility functions + + Copyright (C) Andrew Tridgell 1992-2001 + Copyright (C) Simo Sorce 2001-2002 + Copyright (C) Martin Pool 2003 + Copyright (C) James Peach 2006 + Copyright (C) Jeremy Allison 1992-2007 + + 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 + the Free Software Foundation; either version 3 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program. If not, see <http://www.gnu.org/licenses/>. +*/ + +#include "includes.h" + +static const char b64[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"; + +/** + * Decode a base64 string into a DATA_BLOB - simple and slow algorithm + **/ +_PUBLIC_ DATA_BLOB base64_decode_data_blob(const char *s) +{ + int bit_offset, byte_offset, idx, i, n; + DATA_BLOB decoded = data_blob(s, strlen(s)+1); + unsigned char *d = decoded.data; + char *p; + + n=i=0; + + while (*s && (p=strchr_m(b64,*s))) { + idx = (int)(p - b64); + byte_offset = (i*6)/8; + bit_offset = (i*6)%8; + d[byte_offset] &= ~((1<<(8-bit_offset))-1); + if (bit_offset < 3) { + d[byte_offset] |= (idx << (2-bit_offset)); + n = byte_offset+1; + } else { + d[byte_offset] |= (idx >> (bit_offset-2)); + d[byte_offset+1] = 0; + d[byte_offset+1] |= (idx << (8-(bit_offset-2))) & 0xFF; + n = byte_offset+2; + } + s++; i++; + } + + if ((n > 0) && (*s == '=')) { + n -= 1; + } + + /* fix up length */ + decoded.length = n; + return decoded; +} + +/** + * Decode a base64 string in-place - wrapper for the above + **/ +_PUBLIC_ void base64_decode_inplace(char *s) +{ + DATA_BLOB decoded = base64_decode_data_blob(s); + + if ( decoded.length != 0 ) { + memcpy(s, decoded.data, decoded.length); + + /* null terminate */ + s[decoded.length] = '\0'; + } else { + *s = '\0'; + } + + data_blob_free(&decoded); +} + +/** + * Encode a base64 string into a talloc()ed string caller to free. + * + * From SQUID: adopted from http://ftp.sunet.se/pub2/gnu/vm/base64-encode.c + * with adjustments + **/ + +_PUBLIC_ char *base64_encode_data_blob(TALLOC_CTX *mem_ctx, DATA_BLOB data) +{ + int bits = 0; + int char_count = 0; + size_t out_cnt, len, output_len; + char *result; + + if (!data.length || !data.data) + return NULL; + + out_cnt = 0; + len = data.length; + output_len = data.length * 2 + 4; /* Account for closing bytes. 4 is + * random but should be enough for + * the = and \0 */ + result = talloc_array(mem_ctx, char, output_len); /* get us plenty of space */ + SMB_ASSERT(result != NULL); + + while (len--) { + int c = (unsigned char) *(data.data++); + bits += c; + char_count++; + if (char_count == 3) { + result[out_cnt++] = b64[bits >> 18]; + result[out_cnt++] = b64[(bits >> 12) & 0x3f]; + result[out_cnt++] = b64[(bits >> 6) & 0x3f]; + result[out_cnt++] = b64[bits & 0x3f]; + bits = 0; + char_count = 0; + } else { + bits <<= 8; + } + } + if (char_count != 0) { + bits <<= 16 - (8 * char_count); + result[out_cnt++] = b64[bits >> 18]; + result[out_cnt++] = b64[(bits >> 12) & 0x3f]; + if (char_count == 1) { + result[out_cnt++] = '='; + result[out_cnt++] = '='; + } else { + result[out_cnt++] = b64[(bits >> 6) & 0x3f]; + result[out_cnt++] = '='; + } + } + result[out_cnt] = '\0'; /* terminate */ + return result; +} + diff --git a/lib/util/util.h b/lib/util/util.h index 48d6566dd0..45779912f3 100644 --- a/lib/util/util.h +++ b/lib/util/util.h @@ -410,6 +410,19 @@ _PUBLIC_ int strwicmp(const char *psz1, const char *psz2); _PUBLIC_ void string_replace(char *s, char oldc, char newc); /** + Base64 decode a string, place into a data blob. Caller to data_blob_free() the result. +**/ +_PUBLIC_ DATA_BLOB base64_decode_data_blob(const char *s); +/** + Base64 decode a string, inplace +**/ +_PUBLIC_ void base64_decode_inplace(char *s); +/** + Base64 encode a binary data blob into a string +**/ +_PUBLIC_ char *base64_encode_data_blob(TALLOC_CTX *mem_ctx, DATA_BLOB data); + +/** * Compare 2 strings. * * @note The comparison is case-insensitive. diff --git a/lib/util/util_str.c b/lib/util/util_str.c index 8695266655..41183ff570 100644 --- a/lib/util/util_str.c +++ b/lib/util/util_str.c @@ -320,5 +320,3 @@ _PUBLIC_ void string_replace(char *s, char oldc, char newc) s++; } } - - diff --git a/lib/util/wscript_build b/lib/util/wscript_build index 6aaf04c96e..e39af54c0f 100755 --- a/lib/util/wscript_build +++ b/lib/util/wscript_build @@ -7,7 +7,7 @@ bld.SAMBA_LIBRARY('samba-util-common', util_file.c time.c rbtree.c rfc1738.c select.c genrand.c fsusage.c blocking.c become_daemon.c signal.c system.c params.c util.c util_id.c util_net.c - util_strlist.c idtree.c debug.c fault.c''', + util_strlist.c idtree.c debug.c fault.c base64.c''', public_deps='talloc pthread LIBCRYPTO', # until we get all the dependencies in this library in common # we need to allow this library to be built with unresolved symbols diff --git a/source3/Makefile.in b/source3/Makefile.in index 15279ee039..05843efd63 100644 --- a/source3/Makefile.in +++ b/source3/Makefile.in @@ -450,7 +450,7 @@ LIB_OBJ = $(LIBSAMBAUTIL_OBJ) $(UTIL_OBJ) $(CRYPTO_OBJ) \ lib/access.o lib/smbrun.o \ lib/bitmap.o lib/dprintf.o $(UTIL_REG_OBJ) \ lib/wins_srv.o \ - lib/util_str.o lib/util_sid.o \ + lib/util_str.o ../lib/util/base64.o lib/util_sid.o \ lib/util_unistr.o ../lib/util/charset/codepoints.o ../lib/util/charset/util_str.o lib/util_file.o \ lib/util.o lib/util_names.o \ lib/util_sock.o lib/sock_exec.o lib/util_sec.o \ diff --git a/source3/include/proto.h b/source3/include/proto.h index a565f93462..91ff45d7b9 100644 --- a/source3/include/proto.h +++ b/source3/include/proto.h @@ -1026,9 +1026,6 @@ char *ipstr_list_make(char **ipstr_list, int ip_count); int ipstr_list_parse(const char *ipstr_list, struct ip_service **ip_list); void ipstr_list_free(char* ipstr_list); -DATA_BLOB base64_decode_data_blob(const char *s); -void base64_decode_inplace(char *s); -char *base64_encode_data_blob(TALLOC_CTX *mem_ctx, DATA_BLOB data); uint64_t STR_TO_SMB_BIG_UINT(const char *nptr, const char **entptr); SMB_OFF_T conv_str_size(const char * str); void string_append(char **left, const char *right); diff --git a/source3/lib/util_str.c b/source3/lib/util_str.c index 7b50717463..0f75f45bf1 100644 --- a/source3/lib/util_str.c +++ b/source3/lib/util_str.c @@ -1515,121 +1515,6 @@ void ipstr_list_free(char* ipstr_list) SAFE_FREE(ipstr_list); } -static const char b64[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"; - -/** - * Decode a base64 string into a DATA_BLOB - simple and slow algorithm - **/ -DATA_BLOB base64_decode_data_blob(const char *s) -{ - int bit_offset, byte_offset, idx, i, n; - DATA_BLOB decoded = data_blob(s, strlen(s)+1); - unsigned char *d = decoded.data; - char *p; - - n=i=0; - - while (*s && (p=strchr_m(b64,*s))) { - idx = (int)(p - b64); - byte_offset = (i*6)/8; - bit_offset = (i*6)%8; - d[byte_offset] &= ~((1<<(8-bit_offset))-1); - if (bit_offset < 3) { - d[byte_offset] |= (idx << (2-bit_offset)); - n = byte_offset+1; - } else { - d[byte_offset] |= (idx >> (bit_offset-2)); - d[byte_offset+1] = 0; - d[byte_offset+1] |= (idx << (8-(bit_offset-2))) & 0xFF; - n = byte_offset+2; - } - s++; i++; - } - - if ((n > 0) && (*s == '=')) { - n -= 1; - } - - /* fix up length */ - decoded.length = n; - return decoded; -} - -/** - * Decode a base64 string in-place - wrapper for the above - **/ -void base64_decode_inplace(char *s) -{ - DATA_BLOB decoded = base64_decode_data_blob(s); - - if ( decoded.length != 0 ) { - memcpy(s, decoded.data, decoded.length); - - /* null terminate */ - s[decoded.length] = '\0'; - } else { - *s = '\0'; - } - - data_blob_free(&decoded); -} - -/** - * Encode a base64 string into a talloc()ed string caller to free. - * - * From SQUID: adopted from http://ftp.sunet.se/pub2/gnu/vm/base64-encode.c - * with adjustments - **/ - -char *base64_encode_data_blob(TALLOC_CTX *mem_ctx, DATA_BLOB data) -{ - int bits = 0; - int char_count = 0; - size_t out_cnt, len, output_len; - char *result; - - if (!data.length || !data.data) - return NULL; - - out_cnt = 0; - len = data.length; - output_len = data.length * 2 + 4; /* Account for closing bytes. 4 is - * random but should be enough for - * the = and \0 */ - result = TALLOC_ARRAY(mem_ctx, char, output_len); /* get us plenty of space */ - SMB_ASSERT(result != NULL); - - while (len--) { - int c = (unsigned char) *(data.data++); - bits += c; - char_count++; - if (char_count == 3) { - result[out_cnt++] = b64[bits >> 18]; - result[out_cnt++] = b64[(bits >> 12) & 0x3f]; - result[out_cnt++] = b64[(bits >> 6) & 0x3f]; - result[out_cnt++] = b64[bits & 0x3f]; - bits = 0; - char_count = 0; - } else { - bits <<= 8; - } - } - if (char_count != 0) { - bits <<= 16 - (8 * char_count); - result[out_cnt++] = b64[bits >> 18]; - result[out_cnt++] = b64[(bits >> 12) & 0x3f]; - if (char_count == 1) { - result[out_cnt++] = '='; - result[out_cnt++] = '='; - } else { - result[out_cnt++] = b64[(bits >> 6) & 0x3f]; - result[out_cnt++] = '='; - } - } - result[out_cnt] = '\0'; /* terminate */ - return result; -} - /* read a SMB_BIG_UINT from a string */ uint64_t STR_TO_SMB_BIG_UINT(const char *nptr, const char **entptr) { diff --git a/source4/utils/ntlm_auth.c b/source4/utils/ntlm_auth.c index 34f79715ff..a3701f8822 100644 --- a/source4/utils/ntlm_auth.c +++ b/source4/utils/ntlm_auth.c @@ -142,33 +142,6 @@ static bool parse_ntlm_auth_domain_user(const char *domuser, char **domain, return true; } -/** - * Decode a base64 string into a DATA_BLOB - simple and slow algorithm - **/ -static DATA_BLOB base64_decode_data_blob(TALLOC_CTX *mem_ctx, const char *s) -{ - DATA_BLOB ret = data_blob_talloc(mem_ctx, s, strlen(s)+1); - ret.length = ldb_base64_decode((char *)ret.data); - return ret; -} - -/** - * Encode a base64 string into a talloc()ed string caller to free. - **/ -static char *base64_encode_data_blob(TALLOC_CTX *mem_ctx, DATA_BLOB data) -{ - return ldb_base64_encode(mem_ctx, (const char *)data.data, data.length); -} - -/** - * Decode a base64 string in-place - wrapper for the above - **/ -static void base64_decode_inplace(char *s) -{ - ldb_base64_decode(s); -} - - /* Authenticate a user with a plaintext password */ @@ -291,7 +264,7 @@ static void manage_gensec_get_pw_request(enum stdio_helper_mode stdio_helper_mod } if (strlen(buf) > 3) { - in = base64_decode_data_blob(NULL, buf + 3); + in = base64_decode_data_blob(buf + 3); } else { in = data_blob(NULL, 0); } @@ -433,7 +406,7 @@ static void manage_gensec_request(enum stdio_helper_mode stdio_helper_mode, mux_printf(mux_id, "OK\n"); return; } - in = base64_decode_data_blob(NULL, buf + 3); + in = base64_decode_data_blob(buf + 3); } else { in = data_blob(NULL, 0); } |