1 /***************************************************************************
2 * _ _ ____ _
3 * Project ___| | | | _ \| |
4 * / __| | | | |_) | |
5 * | (__| |_| | _ <| |___
6 * \___|\___/|_| \_\_____|
7 *
8 * Copyright (C) Mandy Wu, <mandy.wu@intel.com>
9 * Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al.
10 *
11 * This software is licensed as described in the file COPYING, which
12 * you should have received as part of this distribution. The terms
13 * are also available at https://curl.se/docs/copyright.html.
14 *
15 * You may opt to use, copy, modify, merge, publish, distribute and/or sell
16 * copies of the Software, and permit persons to whom the Software is
17 * furnished to do so, under the terms of the COPYING file.
18 *
19 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
20 * KIND, either express or implied.
21 *
22 * SPDX-License-Identifier: curl
23 *
24 ***************************************************************************/
25 #include "server_setup.h"
26
27 /*
28 * This is a fake ntlm_auth, which is used for testing NTLM single-sign-on.
29 * When DEBUGBUILD is defined, libcurl invoke this tool instead of real winbind
30 * daemon helper /usr/bin/ntlm_auth. This tool will accept commands and
31 * responses with a pre-written string saved in test case test2005.
32 */
33
34 #define ENABLE_CURLX_PRINTF
35 #include "curlx.h" /* from the private lib dir */
36 #include "getpart.h"
37 #include "util.h"
38
39 /* include memdebug.h last */
40 #include "memdebug.h"
41
42 #define LOGFILE "%s/fake_ntlm%ld.log"
43 static const char *logdir = "log";
44
45 const char *serverlogfile;
46
47 /*
48 * Returns an allocated buffer with printable representation of input
49 * buffer contents or returns NULL on out of memory condition.
50 */
printable(char * inbuf,size_t inlength)51 static char *printable(char *inbuf, size_t inlength)
52 {
53 char *outbuf;
54 char *newbuf;
55 size_t newsize;
56 size_t outsize;
57 size_t outincr = 0;
58 size_t i, o = 0;
59
60 #define HEX_FMT_STR "[0x%02X]"
61 #define HEX_STR_LEN 6
62 #define NOTHING_STR "[NOTHING]"
63 #define NOTHING_LEN 9
64
65 if(!inlength)
66 inlength = strlen(inbuf);
67
68 if(inlength) {
69 outincr = ((inlength/2) < (HEX_STR_LEN + 1)) ?
70 HEX_STR_LEN + 1 : inlength/2;
71 outsize = inlength + outincr;
72 }
73 else
74 outsize = NOTHING_LEN + 1;
75
76 outbuf = malloc(outsize);
77 if(!outbuf)
78 return NULL;
79
80 if(!inlength) {
81 msnprintf(&outbuf[0], outsize, "%s", NOTHING_STR);
82 return outbuf;
83 }
84
85 for(i = 0; i<inlength; i++) {
86
87 if(o > outsize - (HEX_STR_LEN + 1)) {
88 newsize = outsize + outincr;
89 newbuf = realloc(outbuf, newsize);
90 if(!newbuf) {
91 free(outbuf);
92 return NULL;
93 }
94 outbuf = newbuf;
95 outsize = newsize;
96 }
97
98 if((inbuf[i] > 0x20) && (inbuf[i] < 0x7F)) {
99 outbuf[o] = inbuf[i];
100 o++;
101 }
102 else {
103 msnprintf(&outbuf[o], outsize - o, HEX_FMT_STR, inbuf[i]);
104 o += HEX_STR_LEN;
105 }
106
107 }
108 outbuf[o] = '\0';
109
110 return outbuf;
111 }
112
main(int argc,char * argv[])113 int main(int argc, char *argv[])
114 {
115 char buf[1024];
116 char logfilename[256];
117 FILE *stream;
118 int error;
119 char *type1_input = NULL, *type3_input = NULL;
120 char *type1_output = NULL, *type3_output = NULL;
121 size_t size = 0;
122 long testnum;
123 const char *env;
124 int arg = 1;
125 const char *helper_user = "unknown";
126 const char *helper_proto = "unknown";
127 const char *helper_domain = "unknown";
128 bool use_cached_creds = FALSE;
129 char *msgbuf;
130
131 buf[0] = '\0';
132
133 while(argc > arg) {
134 if(!strcmp("--use-cached-creds", argv[arg])) {
135 use_cached_creds = TRUE;
136 arg++;
137 }
138 else if(!strcmp("--helper-protocol", argv[arg])) {
139 arg++;
140 if(argc > arg)
141 helper_proto = argv[arg++];
142 }
143 else if(!strcmp("--username", argv[arg])) {
144 arg++;
145 if(argc > arg)
146 helper_user = argv[arg++];
147 }
148 else if(!strcmp("--domain", argv[arg])) {
149 arg++;
150 if(argc > arg)
151 helper_domain = argv[arg++];
152 }
153 else {
154 puts("Usage: fake_ntlm [option]\n"
155 " --use-cached-creds\n"
156 " --helper-protocol [protocol]\n"
157 " --username [username]\n"
158 " --domain [domain]");
159 exit(1);
160 }
161 }
162
163 env = getenv("CURL_NTLM_LOGDIR");
164 if(env) {
165 logdir = env;
166 }
167
168 env = getenv("CURL_NTLM_AUTH_TESTNUM");
169 if(env) {
170 char *endptr;
171 long lnum = strtol(env, &endptr, 10);
172 if((endptr != env + strlen(env)) || (lnum < 1L)) {
173 fprintf(stderr, "Test number not valid in CURL_NTLM_AUTH_TESTNUM");
174 exit(1);
175 }
176 testnum = lnum;
177 }
178 else {
179 fprintf(stderr, "Test number not specified in CURL_NTLM_AUTH_TESTNUM");
180 exit(1);
181 }
182
183 /* logmsg cannot be used until this file name is set */
184 msnprintf(logfilename, sizeof(logfilename), LOGFILE, logdir, testnum);
185 serverlogfile = logfilename;
186
187 logmsg("fake_ntlm (user: %s) (proto: %s) (domain: %s) (cached creds: %s)",
188 helper_user, helper_proto, helper_domain,
189 (use_cached_creds) ? "yes" : "no");
190
191 env = getenv("CURL_NTLM_AUTH_SRCDIR");
192 if(env) {
193 path = env;
194 }
195
196 stream = test2fopen(testnum, logdir);
197 if(!stream) {
198 error = errno;
199 logmsg("fopen() failed with error: %d %s", error, strerror(error));
200 logmsg("Couldn't open test file %ld", testnum);
201 exit(1);
202 }
203 else {
204 /* get the ntlm_auth input/output */
205 error = getpart(&type1_input, &size, "ntlm_auth_type1", "input", stream);
206 fclose(stream);
207 if(error || size == 0) {
208 logmsg("getpart() type 1 input failed with error: %d", error);
209 exit(1);
210 }
211 }
212
213 stream = test2fopen(testnum, logdir);
214 if(!stream) {
215 error = errno;
216 logmsg("fopen() failed with error: %d %s", error, strerror(error));
217 logmsg("Couldn't open test file %ld", testnum);
218 }
219 else {
220 size = 0;
221 error = getpart(&type3_input, &size, "ntlm_auth_type3", "input", stream);
222 fclose(stream);
223 if(error || size == 0) {
224 logmsg("getpart() type 3 input failed with error: %d", error);
225 exit(1);
226 }
227 }
228
229 while(fgets(buf, sizeof(buf), stdin)) {
230 if(strcmp(buf, type1_input) == 0) {
231 stream = test2fopen(testnum, logdir);
232 if(!stream) {
233 error = errno;
234 logmsg("fopen() failed with error: %d %s", error, strerror(error));
235 logmsg("Couldn't open test file %ld", testnum);
236 exit(1);
237 }
238 else {
239 size = 0;
240 error = getpart(&type1_output, &size, "ntlm_auth_type1", "output",
241 stream);
242 fclose(stream);
243 if(error || size == 0) {
244 logmsg("getpart() type 1 output failed with error: %d", error);
245 exit(1);
246 }
247 }
248 printf("%s", type1_output);
249 fflush(stdout);
250 }
251 else if(strncmp(buf, type3_input, strlen(type3_input)) == 0) {
252 stream = test2fopen(testnum, logdir);
253 if(!stream) {
254 error = errno;
255 logmsg("fopen() failed with error: %d %s", error, strerror(error));
256 logmsg("Couldn't open test file %ld", testnum);
257 exit(1);
258 }
259 else {
260 size = 0;
261 error = getpart(&type3_output, &size, "ntlm_auth_type3", "output",
262 stream);
263 fclose(stream);
264 if(error || size == 0) {
265 logmsg("getpart() type 3 output failed with error: %d", error);
266 exit(1);
267 }
268 }
269 printf("%s", type3_output);
270 fflush(stdout);
271 }
272 else {
273 printf("Unknown request\n");
274 msgbuf = printable(buf, 0);
275 if(msgbuf) {
276 logmsg("invalid input: '%s'\n", msgbuf);
277 free(msgbuf);
278 }
279 else
280 logmsg("OOM formatting invalid input: '%s'\n", buf);
281 exit(1);
282 }
283 }
284 logmsg("Exit");
285 return 1;
286 }
287
