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