Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 1 | /*************************************************************************** |
| 2 | * _ _ ____ _ |
| 3 | * Project ___| | | | _ \| | |
| 4 | * / __| | | | |_) | | |
| 5 | * | (__| |_| | _ <| |___ |
| 6 | * \___|\___/|_| \_\_____| |
| 7 | * |
| 8 | * Copyright (C) 2014, Bill Nagel <wnagel@tycoint.com>, Exacq Technologies |
Elliott Hughes | 0128fe4 | 2018-02-27 14:57:55 -0800 | [diff] [blame] | 9 | * Copyright (C) 2016-2018, Daniel Stenberg, <daniel@haxx.se>, et al. |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 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 |
Alex Deymo | d15eaac | 2016-06-28 14:49:26 -0700 | [diff] [blame] | 13 | * are also available at https://curl.haxx.se/docs/copyright.html. |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 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 | |
| 24 | #include "curl_setup.h" |
| 25 | |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 26 | #if !defined(CURL_DISABLE_SMB) && defined(USE_NTLM) && \ |
| 27 | (CURL_SIZEOF_CURL_OFF_T > 4) |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 28 | |
| 29 | #if !defined(USE_WINDOWS_SSPI) || defined(USE_WIN32_CRYPTO) |
| 30 | |
| 31 | #define BUILDING_CURL_SMB_C |
| 32 | |
| 33 | #ifdef HAVE_PROCESS_H |
| 34 | #include <process.h> |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 35 | #ifdef CURL_WINDOWS_APP |
| 36 | #define getpid GetCurrentProcessId |
| 37 | #elif !defined(MSDOS) |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 38 | #define getpid _getpid |
| 39 | #endif |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 40 | #endif |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 41 | |
| 42 | #include "smb.h" |
| 43 | #include "urldata.h" |
| 44 | #include "sendf.h" |
| 45 | #include "multiif.h" |
| 46 | #include "connect.h" |
| 47 | #include "progress.h" |
| 48 | #include "transfer.h" |
| 49 | #include "vtls/vtls.h" |
| 50 | #include "curl_ntlm_core.h" |
| 51 | #include "escape.h" |
| 52 | #include "curl_endian.h" |
| 53 | |
| 54 | /* The last #include files should be: */ |
| 55 | #include "curl_memory.h" |
| 56 | #include "memdebug.h" |
| 57 | |
| 58 | /* Local API functions */ |
| 59 | static CURLcode smb_setup_connection(struct connectdata *conn); |
| 60 | static CURLcode smb_connect(struct connectdata *conn, bool *done); |
| 61 | static CURLcode smb_connection_state(struct connectdata *conn, bool *done); |
| 62 | static CURLcode smb_request_state(struct connectdata *conn, bool *done); |
| 63 | static CURLcode smb_done(struct connectdata *conn, CURLcode status, |
| 64 | bool premature); |
| 65 | static CURLcode smb_disconnect(struct connectdata *conn, bool dead); |
| 66 | static int smb_getsock(struct connectdata *conn, curl_socket_t *socks, |
| 67 | int numsocks); |
| 68 | static CURLcode smb_parse_url_path(struct connectdata *conn); |
| 69 | |
| 70 | /* |
| 71 | * SMB handler interface |
| 72 | */ |
| 73 | const struct Curl_handler Curl_handler_smb = { |
| 74 | "SMB", /* scheme */ |
| 75 | smb_setup_connection, /* setup_connection */ |
| 76 | ZERO_NULL, /* do_it */ |
| 77 | smb_done, /* done */ |
| 78 | ZERO_NULL, /* do_more */ |
| 79 | smb_connect, /* connect_it */ |
| 80 | smb_connection_state, /* connecting */ |
| 81 | smb_request_state, /* doing */ |
| 82 | smb_getsock, /* proto_getsock */ |
| 83 | smb_getsock, /* doing_getsock */ |
| 84 | ZERO_NULL, /* domore_getsock */ |
| 85 | ZERO_NULL, /* perform_getsock */ |
| 86 | smb_disconnect, /* disconnect */ |
| 87 | ZERO_NULL, /* readwrite */ |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 88 | ZERO_NULL, /* connection_check */ |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 89 | PORT_SMB, /* defport */ |
| 90 | CURLPROTO_SMB, /* protocol */ |
| 91 | PROTOPT_NONE /* flags */ |
| 92 | }; |
| 93 | |
| 94 | #ifdef USE_SSL |
| 95 | /* |
| 96 | * SMBS handler interface |
| 97 | */ |
| 98 | const struct Curl_handler Curl_handler_smbs = { |
| 99 | "SMBS", /* scheme */ |
| 100 | smb_setup_connection, /* setup_connection */ |
| 101 | ZERO_NULL, /* do_it */ |
| 102 | smb_done, /* done */ |
| 103 | ZERO_NULL, /* do_more */ |
| 104 | smb_connect, /* connect_it */ |
| 105 | smb_connection_state, /* connecting */ |
| 106 | smb_request_state, /* doing */ |
| 107 | smb_getsock, /* proto_getsock */ |
| 108 | smb_getsock, /* doing_getsock */ |
| 109 | ZERO_NULL, /* domore_getsock */ |
| 110 | ZERO_NULL, /* perform_getsock */ |
| 111 | smb_disconnect, /* disconnect */ |
| 112 | ZERO_NULL, /* readwrite */ |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 113 | ZERO_NULL, /* connection_check */ |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 114 | PORT_SMBS, /* defport */ |
| 115 | CURLPROTO_SMBS, /* protocol */ |
| 116 | PROTOPT_SSL /* flags */ |
| 117 | }; |
| 118 | #endif |
| 119 | |
| 120 | #define MAX_PAYLOAD_SIZE 0x8000 |
| 121 | #define MAX_MESSAGE_SIZE (MAX_PAYLOAD_SIZE + 0x1000) |
| 122 | #define CLIENTNAME "curl" |
| 123 | #define SERVICENAME "?????" |
| 124 | |
| 125 | /* Append a string to an SMB message */ |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 126 | #define MSGCAT(str) \ |
| 127 | strcpy(p, (str)); \ |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 128 | p += strlen(str); |
| 129 | |
| 130 | /* Append a null-terminated string to an SMB message */ |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 131 | #define MSGCATNULL(str) \ |
| 132 | strcpy(p, (str)); \ |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 133 | p += strlen(str) + 1; |
| 134 | |
| 135 | /* SMB is mostly little endian */ |
| 136 | #if (defined(__BYTE_ORDER__) && __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__) || \ |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 137 | defined(__OS400__) |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 138 | static unsigned short smb_swap16(unsigned short x) |
| 139 | { |
Alex Deymo | d15eaac | 2016-06-28 14:49:26 -0700 | [diff] [blame] | 140 | return (unsigned short) ((x << 8) | ((x >> 8) & 0xff)); |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 141 | } |
| 142 | |
| 143 | static unsigned int smb_swap32(unsigned int x) |
| 144 | { |
| 145 | return (x << 24) | ((x << 8) & 0xff0000) | ((x >> 8) & 0xff00) | |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 146 | ((x >> 24) & 0xff); |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 147 | } |
| 148 | |
Elliott Hughes | 0128fe4 | 2018-02-27 14:57:55 -0800 | [diff] [blame] | 149 | static curl_off_t smb_swap64(curl_off_t x) |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 150 | { |
Elliott Hughes | 0128fe4 | 2018-02-27 14:57:55 -0800 | [diff] [blame] | 151 | return ((curl_off_t) smb_swap32((unsigned int) x) << 32) | |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 152 | smb_swap32((unsigned int) (x >> 32)); |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 153 | } |
Elliott Hughes | 0128fe4 | 2018-02-27 14:57:55 -0800 | [diff] [blame] | 154 | |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 155 | #else |
| 156 | # define smb_swap16(x) (x) |
| 157 | # define smb_swap32(x) (x) |
| 158 | # define smb_swap64(x) (x) |
| 159 | #endif |
| 160 | |
| 161 | /* SMB request state */ |
| 162 | enum smb_req_state { |
| 163 | SMB_REQUESTING, |
| 164 | SMB_TREE_CONNECT, |
| 165 | SMB_OPEN, |
| 166 | SMB_DOWNLOAD, |
| 167 | SMB_UPLOAD, |
| 168 | SMB_CLOSE, |
| 169 | SMB_TREE_DISCONNECT, |
| 170 | SMB_DONE |
| 171 | }; |
| 172 | |
| 173 | /* SMB request data */ |
| 174 | struct smb_request { |
| 175 | enum smb_req_state state; |
| 176 | char *share; |
| 177 | char *path; |
| 178 | unsigned short tid; /* Even if we connect to the same tree as another */ |
| 179 | unsigned short fid; /* request, the tid will be different */ |
| 180 | CURLcode result; |
| 181 | }; |
| 182 | |
| 183 | static void conn_state(struct connectdata *conn, enum smb_conn_state newstate) |
| 184 | { |
| 185 | struct smb_conn *smb = &conn->proto.smbc; |
| 186 | #if defined(DEBUGBUILD) && !defined(CURL_DISABLE_VERBOSE_STRINGS) |
| 187 | /* For debug purposes */ |
| 188 | static const char * const names[] = { |
| 189 | "SMB_NOT_CONNECTED", |
| 190 | "SMB_CONNECTING", |
| 191 | "SMB_NEGOTIATE", |
| 192 | "SMB_SETUP", |
| 193 | "SMB_CONNECTED", |
| 194 | /* LAST */ |
| 195 | }; |
| 196 | |
| 197 | if(smb->state != newstate) |
| 198 | infof(conn->data, "SMB conn %p state change from %s to %s\n", |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 199 | (void *)smb, names[smb->state], names[newstate]); |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 200 | #endif |
| 201 | |
| 202 | smb->state = newstate; |
| 203 | } |
| 204 | |
| 205 | static void request_state(struct connectdata *conn, |
| 206 | enum smb_req_state newstate) |
| 207 | { |
| 208 | struct smb_request *req = conn->data->req.protop; |
| 209 | #if defined(DEBUGBUILD) && !defined(CURL_DISABLE_VERBOSE_STRINGS) |
| 210 | /* For debug purposes */ |
| 211 | static const char * const names[] = { |
| 212 | "SMB_REQUESTING", |
| 213 | "SMB_TREE_CONNECT", |
| 214 | "SMB_OPEN", |
| 215 | "SMB_DOWNLOAD", |
| 216 | "SMB_UPLOAD", |
| 217 | "SMB_CLOSE", |
| 218 | "SMB_TREE_DISCONNECT", |
| 219 | "SMB_DONE", |
| 220 | /* LAST */ |
| 221 | }; |
| 222 | |
| 223 | if(req->state != newstate) |
| 224 | infof(conn->data, "SMB request %p state change from %s to %s\n", |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 225 | (void *)req, names[req->state], names[newstate]); |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 226 | #endif |
| 227 | |
| 228 | req->state = newstate; |
| 229 | } |
| 230 | |
| 231 | static CURLcode smb_setup_connection(struct connectdata *conn) |
| 232 | { |
| 233 | struct smb_request *req; |
| 234 | |
| 235 | /* Initialize the request state */ |
| 236 | conn->data->req.protop = req = calloc(1, sizeof(struct smb_request)); |
| 237 | if(!req) |
| 238 | return CURLE_OUT_OF_MEMORY; |
| 239 | |
| 240 | /* Parse the URL path */ |
| 241 | return smb_parse_url_path(conn); |
| 242 | } |
| 243 | |
| 244 | static CURLcode smb_connect(struct connectdata *conn, bool *done) |
| 245 | { |
| 246 | struct smb_conn *smbc = &conn->proto.smbc; |
| 247 | char *slash; |
| 248 | |
| 249 | (void) done; |
| 250 | |
| 251 | /* Check we have a username and password to authenticate with */ |
| 252 | if(!conn->bits.user_passwd) |
| 253 | return CURLE_LOGIN_DENIED; |
| 254 | |
| 255 | /* Initialize the connection state */ |
| 256 | memset(smbc, 0, sizeof(*smbc)); |
| 257 | smbc->state = SMB_CONNECTING; |
| 258 | smbc->recv_buf = malloc(MAX_MESSAGE_SIZE); |
| 259 | if(!smbc->recv_buf) |
| 260 | return CURLE_OUT_OF_MEMORY; |
| 261 | |
| 262 | /* Multiple requests are allowed with this connection */ |
| 263 | connkeep(conn, "SMB default"); |
| 264 | |
| 265 | /* Parse the username, domain, and password */ |
| 266 | slash = strchr(conn->user, '/'); |
| 267 | if(!slash) |
| 268 | slash = strchr(conn->user, '\\'); |
| 269 | |
| 270 | if(slash) { |
| 271 | smbc->user = slash + 1; |
| 272 | smbc->domain = strdup(conn->user); |
| 273 | if(!smbc->domain) |
| 274 | return CURLE_OUT_OF_MEMORY; |
| 275 | smbc->domain[slash - conn->user] = 0; |
| 276 | } |
| 277 | else { |
| 278 | smbc->user = conn->user; |
| 279 | smbc->domain = strdup(conn->host.name); |
| 280 | if(!smbc->domain) |
| 281 | return CURLE_OUT_OF_MEMORY; |
| 282 | } |
| 283 | |
| 284 | return CURLE_OK; |
| 285 | } |
| 286 | |
| 287 | static CURLcode smb_recv_message(struct connectdata *conn, void **msg) |
| 288 | { |
| 289 | struct smb_conn *smbc = &conn->proto.smbc; |
| 290 | char *buf = smbc->recv_buf; |
| 291 | ssize_t bytes_read; |
| 292 | size_t nbt_size; |
| 293 | size_t msg_size; |
| 294 | size_t len = MAX_MESSAGE_SIZE - smbc->got; |
| 295 | CURLcode result; |
| 296 | |
| 297 | result = Curl_read(conn, FIRSTSOCKET, buf + smbc->got, len, &bytes_read); |
| 298 | if(result) |
| 299 | return result; |
| 300 | |
| 301 | if(!bytes_read) |
| 302 | return CURLE_OK; |
| 303 | |
| 304 | smbc->got += bytes_read; |
| 305 | |
| 306 | /* Check for a 32-bit nbt header */ |
| 307 | if(smbc->got < sizeof(unsigned int)) |
| 308 | return CURLE_OK; |
| 309 | |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 310 | nbt_size = Curl_read16_be((const unsigned char *) |
| 311 | (buf + sizeof(unsigned short))) + |
| 312 | sizeof(unsigned int); |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 313 | if(smbc->got < nbt_size) |
| 314 | return CURLE_OK; |
| 315 | |
| 316 | msg_size = sizeof(struct smb_header); |
| 317 | if(nbt_size >= msg_size + 1) { |
| 318 | /* Add the word count */ |
| 319 | msg_size += 1 + ((unsigned char) buf[msg_size]) * sizeof(unsigned short); |
| 320 | if(nbt_size >= msg_size + sizeof(unsigned short)) { |
| 321 | /* Add the byte count */ |
| 322 | msg_size += sizeof(unsigned short) + |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 323 | Curl_read16_le((const unsigned char *)&buf[msg_size]); |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 324 | if(nbt_size < msg_size) |
| 325 | return CURLE_READ_ERROR; |
| 326 | } |
| 327 | } |
| 328 | |
| 329 | *msg = buf; |
| 330 | |
| 331 | return CURLE_OK; |
| 332 | } |
| 333 | |
| 334 | static void smb_pop_message(struct connectdata *conn) |
| 335 | { |
| 336 | struct smb_conn *smbc = &conn->proto.smbc; |
| 337 | |
| 338 | smbc->got = 0; |
| 339 | } |
| 340 | |
| 341 | static void smb_format_message(struct connectdata *conn, struct smb_header *h, |
| 342 | unsigned char cmd, size_t len) |
| 343 | { |
| 344 | struct smb_conn *smbc = &conn->proto.smbc; |
| 345 | struct smb_request *req = conn->data->req.protop; |
| 346 | unsigned int pid; |
| 347 | |
| 348 | memset(h, 0, sizeof(*h)); |
| 349 | h->nbt_length = htons((unsigned short) (sizeof(*h) - sizeof(unsigned int) + |
| 350 | len)); |
| 351 | memcpy((char *)h->magic, "\xffSMB", 4); |
| 352 | h->command = cmd; |
| 353 | h->flags = SMB_FLAGS_CANONICAL_PATHNAMES | SMB_FLAGS_CASELESS_PATHNAMES; |
| 354 | h->flags2 = smb_swap16(SMB_FLAGS2_IS_LONG_NAME | SMB_FLAGS2_KNOWS_LONG_NAME); |
| 355 | h->uid = smb_swap16(smbc->uid); |
| 356 | h->tid = smb_swap16(req->tid); |
| 357 | pid = getpid(); |
| 358 | h->pid_high = smb_swap16((unsigned short)(pid >> 16)); |
| 359 | h->pid = smb_swap16((unsigned short) pid); |
| 360 | } |
| 361 | |
| 362 | static CURLcode smb_send(struct connectdata *conn, ssize_t len, |
| 363 | size_t upload_size) |
| 364 | { |
| 365 | struct smb_conn *smbc = &conn->proto.smbc; |
| 366 | ssize_t bytes_written; |
| 367 | CURLcode result; |
| 368 | |
| 369 | result = Curl_write(conn, FIRSTSOCKET, conn->data->state.uploadbuffer, |
| 370 | len, &bytes_written); |
| 371 | if(result) |
| 372 | return result; |
| 373 | |
| 374 | if(bytes_written != len) { |
| 375 | smbc->send_size = len; |
| 376 | smbc->sent = bytes_written; |
| 377 | } |
| 378 | |
| 379 | smbc->upload_size = upload_size; |
| 380 | |
| 381 | return CURLE_OK; |
| 382 | } |
| 383 | |
| 384 | static CURLcode smb_flush(struct connectdata *conn) |
| 385 | { |
| 386 | struct smb_conn *smbc = &conn->proto.smbc; |
| 387 | ssize_t bytes_written; |
| 388 | ssize_t len = smbc->send_size - smbc->sent; |
| 389 | CURLcode result; |
| 390 | |
| 391 | if(!smbc->send_size) |
| 392 | return CURLE_OK; |
| 393 | |
| 394 | result = Curl_write(conn, FIRSTSOCKET, |
| 395 | conn->data->state.uploadbuffer + smbc->sent, |
| 396 | len, &bytes_written); |
| 397 | if(result) |
| 398 | return result; |
| 399 | |
| 400 | if(bytes_written != len) |
| 401 | smbc->sent += bytes_written; |
| 402 | else |
| 403 | smbc->send_size = 0; |
| 404 | |
| 405 | return CURLE_OK; |
| 406 | } |
| 407 | |
| 408 | static CURLcode smb_send_message(struct connectdata *conn, unsigned char cmd, |
| 409 | const void *msg, size_t msg_len) |
| 410 | { |
| 411 | smb_format_message(conn, (struct smb_header *)conn->data->state.uploadbuffer, |
| 412 | cmd, msg_len); |
| 413 | memcpy(conn->data->state.uploadbuffer + sizeof(struct smb_header), |
| 414 | msg, msg_len); |
| 415 | |
| 416 | return smb_send(conn, sizeof(struct smb_header) + msg_len, 0); |
| 417 | } |
| 418 | |
| 419 | static CURLcode smb_send_negotiate(struct connectdata *conn) |
| 420 | { |
| 421 | const char *msg = "\x00\x0c\x00\x02NT LM 0.12"; |
| 422 | |
| 423 | return smb_send_message(conn, SMB_COM_NEGOTIATE, msg, 15); |
| 424 | } |
| 425 | |
| 426 | static CURLcode smb_send_setup(struct connectdata *conn) |
| 427 | { |
| 428 | struct smb_conn *smbc = &conn->proto.smbc; |
| 429 | struct smb_setup msg; |
| 430 | char *p = msg.bytes; |
| 431 | unsigned char lm_hash[21]; |
| 432 | unsigned char lm[24]; |
| 433 | unsigned char nt_hash[21]; |
| 434 | unsigned char nt[24]; |
| 435 | |
| 436 | size_t byte_count = sizeof(lm) + sizeof(nt); |
| 437 | byte_count += strlen(smbc->user) + strlen(smbc->domain); |
| 438 | byte_count += strlen(OS) + strlen(CLIENTNAME) + 4; /* 4 null chars */ |
| 439 | if(byte_count > sizeof(msg.bytes)) |
| 440 | return CURLE_FILESIZE_EXCEEDED; |
| 441 | |
| 442 | Curl_ntlm_core_mk_lm_hash(conn->data, conn->passwd, lm_hash); |
| 443 | Curl_ntlm_core_lm_resp(lm_hash, smbc->challenge, lm); |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 444 | #ifdef USE_NTRESPONSES |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 445 | Curl_ntlm_core_mk_nt_hash(conn->data, conn->passwd, nt_hash); |
| 446 | Curl_ntlm_core_lm_resp(nt_hash, smbc->challenge, nt); |
| 447 | #else |
| 448 | memset(nt, 0, sizeof(nt)); |
| 449 | #endif |
| 450 | |
| 451 | memset(&msg, 0, sizeof(msg)); |
| 452 | msg.word_count = SMB_WC_SETUP_ANDX; |
| 453 | msg.andx.command = SMB_COM_NO_ANDX_COMMAND; |
| 454 | msg.max_buffer_size = smb_swap16(MAX_MESSAGE_SIZE); |
| 455 | msg.max_mpx_count = smb_swap16(1); |
| 456 | msg.vc_number = smb_swap16(1); |
| 457 | msg.session_key = smb_swap32(smbc->session_key); |
| 458 | msg.capabilities = smb_swap32(SMB_CAP_LARGE_FILES); |
| 459 | msg.lengths[0] = smb_swap16(sizeof(lm)); |
| 460 | msg.lengths[1] = smb_swap16(sizeof(nt)); |
| 461 | memcpy(p, lm, sizeof(lm)); |
| 462 | p += sizeof(lm); |
| 463 | memcpy(p, nt, sizeof(nt)); |
| 464 | p += sizeof(nt); |
| 465 | MSGCATNULL(smbc->user); |
| 466 | MSGCATNULL(smbc->domain); |
| 467 | MSGCATNULL(OS); |
| 468 | MSGCATNULL(CLIENTNAME); |
| 469 | byte_count = p - msg.bytes; |
| 470 | msg.byte_count = smb_swap16((unsigned short)byte_count); |
| 471 | |
| 472 | return smb_send_message(conn, SMB_COM_SETUP_ANDX, &msg, |
| 473 | sizeof(msg) - sizeof(msg.bytes) + byte_count); |
| 474 | } |
| 475 | |
| 476 | static CURLcode smb_send_tree_connect(struct connectdata *conn) |
| 477 | { |
| 478 | struct smb_request *req = conn->data->req.protop; |
| 479 | struct smb_tree_connect msg; |
| 480 | char *p = msg.bytes; |
| 481 | |
| 482 | size_t byte_count = strlen(conn->host.name) + strlen(req->share); |
| 483 | byte_count += strlen(SERVICENAME) + 5; /* 2 nulls and 3 backslashes */ |
| 484 | if(byte_count > sizeof(msg.bytes)) |
| 485 | return CURLE_FILESIZE_EXCEEDED; |
| 486 | |
| 487 | memset(&msg, 0, sizeof(msg)); |
| 488 | msg.word_count = SMB_WC_TREE_CONNECT_ANDX; |
| 489 | msg.andx.command = SMB_COM_NO_ANDX_COMMAND; |
| 490 | msg.pw_len = 0; |
| 491 | MSGCAT("\\\\"); |
| 492 | MSGCAT(conn->host.name); |
| 493 | MSGCAT("\\"); |
| 494 | MSGCATNULL(req->share); |
| 495 | MSGCATNULL(SERVICENAME); /* Match any type of service */ |
| 496 | byte_count = p - msg.bytes; |
| 497 | msg.byte_count = smb_swap16((unsigned short)byte_count); |
| 498 | |
| 499 | return smb_send_message(conn, SMB_COM_TREE_CONNECT_ANDX, &msg, |
| 500 | sizeof(msg) - sizeof(msg.bytes) + byte_count); |
| 501 | } |
| 502 | |
| 503 | static CURLcode smb_send_open(struct connectdata *conn) |
| 504 | { |
| 505 | struct smb_request *req = conn->data->req.protop; |
| 506 | struct smb_nt_create msg; |
| 507 | size_t byte_count; |
| 508 | |
| 509 | if((strlen(req->path) + 1) > sizeof(msg.bytes)) |
| 510 | return CURLE_FILESIZE_EXCEEDED; |
| 511 | |
| 512 | memset(&msg, 0, sizeof(msg)); |
| 513 | msg.word_count = SMB_WC_NT_CREATE_ANDX; |
| 514 | msg.andx.command = SMB_COM_NO_ANDX_COMMAND; |
| 515 | byte_count = strlen(req->path); |
| 516 | msg.name_length = smb_swap16((unsigned short)byte_count); |
| 517 | msg.share_access = smb_swap32(SMB_FILE_SHARE_ALL); |
| 518 | if(conn->data->set.upload) { |
| 519 | msg.access = smb_swap32(SMB_GENERIC_READ | SMB_GENERIC_WRITE); |
| 520 | msg.create_disposition = smb_swap32(SMB_FILE_OVERWRITE_IF); |
| 521 | } |
| 522 | else { |
| 523 | msg.access = smb_swap32(SMB_GENERIC_READ); |
| 524 | msg.create_disposition = smb_swap32(SMB_FILE_OPEN); |
| 525 | } |
| 526 | msg.byte_count = smb_swap16((unsigned short) ++byte_count); |
| 527 | strcpy(msg.bytes, req->path); |
| 528 | |
| 529 | return smb_send_message(conn, SMB_COM_NT_CREATE_ANDX, &msg, |
| 530 | sizeof(msg) - sizeof(msg.bytes) + byte_count); |
| 531 | } |
| 532 | |
| 533 | static CURLcode smb_send_close(struct connectdata *conn) |
| 534 | { |
| 535 | struct smb_request *req = conn->data->req.protop; |
| 536 | struct smb_close msg; |
| 537 | |
| 538 | memset(&msg, 0, sizeof(msg)); |
| 539 | msg.word_count = SMB_WC_CLOSE; |
| 540 | msg.fid = smb_swap16(req->fid); |
| 541 | |
| 542 | return smb_send_message(conn, SMB_COM_CLOSE, &msg, sizeof(msg)); |
| 543 | } |
| 544 | |
| 545 | static CURLcode smb_send_tree_disconnect(struct connectdata *conn) |
| 546 | { |
| 547 | struct smb_tree_disconnect msg; |
| 548 | |
| 549 | memset(&msg, 0, sizeof(msg)); |
| 550 | |
| 551 | return smb_send_message(conn, SMB_COM_TREE_DISCONNECT, &msg, sizeof(msg)); |
| 552 | } |
| 553 | |
| 554 | static CURLcode smb_send_read(struct connectdata *conn) |
| 555 | { |
| 556 | struct smb_request *req = conn->data->req.protop; |
| 557 | curl_off_t offset = conn->data->req.offset; |
| 558 | struct smb_read msg; |
| 559 | |
| 560 | memset(&msg, 0, sizeof(msg)); |
| 561 | msg.word_count = SMB_WC_READ_ANDX; |
| 562 | msg.andx.command = SMB_COM_NO_ANDX_COMMAND; |
| 563 | msg.fid = smb_swap16(req->fid); |
| 564 | msg.offset = smb_swap32((unsigned int) offset); |
| 565 | msg.offset_high = smb_swap32((unsigned int) (offset >> 32)); |
| 566 | msg.min_bytes = smb_swap16(MAX_PAYLOAD_SIZE); |
| 567 | msg.max_bytes = smb_swap16(MAX_PAYLOAD_SIZE); |
| 568 | |
| 569 | return smb_send_message(conn, SMB_COM_READ_ANDX, &msg, sizeof(msg)); |
| 570 | } |
| 571 | |
| 572 | static CURLcode smb_send_write(struct connectdata *conn) |
| 573 | { |
| 574 | struct smb_write *msg = (struct smb_write *)conn->data->state.uploadbuffer; |
| 575 | struct smb_request *req = conn->data->req.protop; |
| 576 | curl_off_t offset = conn->data->req.offset; |
| 577 | |
| 578 | curl_off_t upload_size = conn->data->req.size - conn->data->req.bytecount; |
| 579 | if(upload_size >= MAX_PAYLOAD_SIZE - 1) /* There is one byte of padding */ |
| 580 | upload_size = MAX_PAYLOAD_SIZE - 1; |
| 581 | |
| 582 | memset(msg, 0, sizeof(*msg)); |
| 583 | msg->word_count = SMB_WC_WRITE_ANDX; |
| 584 | msg->andx.command = SMB_COM_NO_ANDX_COMMAND; |
| 585 | msg->fid = smb_swap16(req->fid); |
| 586 | msg->offset = smb_swap32((unsigned int) offset); |
| 587 | msg->offset_high = smb_swap32((unsigned int) (offset >> 32)); |
| 588 | msg->data_length = smb_swap16((unsigned short) upload_size); |
| 589 | msg->data_offset = smb_swap16(sizeof(*msg) - sizeof(unsigned int)); |
| 590 | msg->byte_count = smb_swap16((unsigned short) (upload_size + 1)); |
| 591 | |
| 592 | smb_format_message(conn, &msg->h, SMB_COM_WRITE_ANDX, |
| 593 | sizeof(*msg) - sizeof(msg->h) + (size_t) upload_size); |
| 594 | |
| 595 | return smb_send(conn, sizeof(*msg), (size_t) upload_size); |
| 596 | } |
| 597 | |
| 598 | static CURLcode smb_send_and_recv(struct connectdata *conn, void **msg) |
| 599 | { |
| 600 | struct smb_conn *smbc = &conn->proto.smbc; |
| 601 | CURLcode result; |
| 602 | |
| 603 | /* Check if there is data in the transfer buffer */ |
| 604 | if(!smbc->send_size && smbc->upload_size) { |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 605 | int nread = smbc->upload_size > UPLOAD_BUFSIZE ? UPLOAD_BUFSIZE : |
| 606 | (int) smbc->upload_size; |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 607 | conn->data->req.upload_fromhere = conn->data->state.uploadbuffer; |
| 608 | result = Curl_fillreadbuffer(conn, nread, &nread); |
| 609 | if(result && result != CURLE_AGAIN) |
| 610 | return result; |
| 611 | if(!nread) |
| 612 | return CURLE_OK; |
| 613 | |
| 614 | smbc->upload_size -= nread; |
| 615 | smbc->send_size = nread; |
| 616 | smbc->sent = 0; |
| 617 | } |
| 618 | |
| 619 | /* Check if there is data to send */ |
| 620 | if(smbc->send_size) { |
| 621 | result = smb_flush(conn); |
| 622 | if(result) |
| 623 | return result; |
| 624 | } |
| 625 | |
| 626 | /* Check if there is still data to be sent */ |
| 627 | if(smbc->send_size || smbc->upload_size) |
| 628 | return CURLE_AGAIN; |
| 629 | |
| 630 | return smb_recv_message(conn, msg); |
| 631 | } |
| 632 | |
| 633 | static CURLcode smb_connection_state(struct connectdata *conn, bool *done) |
| 634 | { |
| 635 | struct smb_conn *smbc = &conn->proto.smbc; |
| 636 | struct smb_negotiate_response *nrsp; |
| 637 | struct smb_header *h; |
| 638 | CURLcode result; |
| 639 | void *msg = NULL; |
| 640 | |
| 641 | if(smbc->state == SMB_CONNECTING) { |
| 642 | #ifdef USE_SSL |
| 643 | if((conn->handler->flags & PROTOPT_SSL)) { |
Alex Deymo | 486467e | 2017-12-19 19:04:07 +0100 | [diff] [blame] | 644 | bool ssl_done = FALSE; |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 645 | result = Curl_ssl_connect_nonblocking(conn, FIRSTSOCKET, &ssl_done); |
| 646 | if(result && result != CURLE_AGAIN) |
| 647 | return result; |
| 648 | if(!ssl_done) |
| 649 | return CURLE_OK; |
| 650 | } |
| 651 | #endif |
| 652 | |
| 653 | result = smb_send_negotiate(conn); |
| 654 | if(result) { |
| 655 | connclose(conn, "SMB: failed to send negotiate message"); |
| 656 | return result; |
| 657 | } |
| 658 | |
| 659 | conn_state(conn, SMB_NEGOTIATE); |
| 660 | } |
| 661 | |
| 662 | /* Send the previous message and check for a response */ |
| 663 | result = smb_send_and_recv(conn, &msg); |
| 664 | if(result && result != CURLE_AGAIN) { |
| 665 | connclose(conn, "SMB: failed to communicate"); |
| 666 | return result; |
| 667 | } |
| 668 | |
| 669 | if(!msg) |
| 670 | return CURLE_OK; |
| 671 | |
| 672 | h = msg; |
| 673 | |
| 674 | switch(smbc->state) { |
| 675 | case SMB_NEGOTIATE: |
Elliott Hughes | cee0338 | 2017-06-23 12:17:18 -0700 | [diff] [blame] | 676 | if(h->status || smbc->got < sizeof(*nrsp) + sizeof(smbc->challenge) - 1) { |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 677 | connclose(conn, "SMB: negotiation failed"); |
| 678 | return CURLE_COULDNT_CONNECT; |
| 679 | } |
| 680 | nrsp = msg; |
| 681 | memcpy(smbc->challenge, nrsp->bytes, sizeof(smbc->challenge)); |
| 682 | smbc->session_key = smb_swap32(nrsp->session_key); |
| 683 | result = smb_send_setup(conn); |
| 684 | if(result) { |
| 685 | connclose(conn, "SMB: failed to send setup message"); |
| 686 | return result; |
| 687 | } |
| 688 | conn_state(conn, SMB_SETUP); |
| 689 | break; |
| 690 | |
| 691 | case SMB_SETUP: |
| 692 | if(h->status) { |
| 693 | connclose(conn, "SMB: authentication failed"); |
| 694 | return CURLE_LOGIN_DENIED; |
| 695 | } |
| 696 | smbc->uid = smb_swap16(h->uid); |
| 697 | conn_state(conn, SMB_CONNECTED); |
| 698 | *done = true; |
| 699 | break; |
| 700 | |
| 701 | default: |
| 702 | smb_pop_message(conn); |
| 703 | return CURLE_OK; /* ignore */ |
| 704 | } |
| 705 | |
| 706 | smb_pop_message(conn); |
| 707 | |
| 708 | return CURLE_OK; |
| 709 | } |
| 710 | |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 711 | /* |
Elliott Hughes | cac3980 | 2018-04-27 16:19:43 -0700 | [diff] [blame] | 712 | * Convert a timestamp from the Windows world (100 nsec units from 1 Jan 1601) |
| 713 | * to Posix time. Cap the output to fit within a time_t. |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 714 | */ |
Elliott Hughes | cac3980 | 2018-04-27 16:19:43 -0700 | [diff] [blame] | 715 | static void get_posix_time(time_t *out, curl_off_t timestamp) |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 716 | { |
Elliott Hughes | 0128fe4 | 2018-02-27 14:57:55 -0800 | [diff] [blame] | 717 | timestamp -= 116444736000000000; |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 718 | timestamp /= 10000000; |
Elliott Hughes | cac3980 | 2018-04-27 16:19:43 -0700 | [diff] [blame] | 719 | #if SIZEOF_TIME_T < SIZEOF_CURL_OFF_T |
| 720 | if(timestamp > TIME_T_MAX) |
| 721 | *out = TIME_T_MAX; |
| 722 | else if(timestamp < TIME_T_MIN) |
| 723 | *out = TIME_T_MIN; |
| 724 | else |
| 725 | #endif |
| 726 | *out = (time_t) timestamp; |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 727 | } |
| 728 | |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 729 | static CURLcode smb_request_state(struct connectdata *conn, bool *done) |
| 730 | { |
| 731 | struct smb_request *req = conn->data->req.protop; |
| 732 | struct smb_header *h; |
Elliott Hughes | cee0338 | 2017-06-23 12:17:18 -0700 | [diff] [blame] | 733 | struct smb_conn *smbc = &conn->proto.smbc; |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 734 | enum smb_req_state next_state = SMB_DONE; |
| 735 | unsigned short len; |
| 736 | unsigned short off; |
| 737 | CURLcode result; |
| 738 | void *msg = NULL; |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 739 | const struct smb_nt_create_response *smb_m; |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 740 | |
| 741 | /* Start the request */ |
| 742 | if(req->state == SMB_REQUESTING) { |
| 743 | result = smb_send_tree_connect(conn); |
| 744 | if(result) { |
| 745 | connclose(conn, "SMB: failed to send tree connect message"); |
| 746 | return result; |
| 747 | } |
| 748 | |
| 749 | request_state(conn, SMB_TREE_CONNECT); |
| 750 | } |
| 751 | |
| 752 | /* Send the previous message and check for a response */ |
| 753 | result = smb_send_and_recv(conn, &msg); |
| 754 | if(result && result != CURLE_AGAIN) { |
| 755 | connclose(conn, "SMB: failed to communicate"); |
| 756 | return result; |
| 757 | } |
| 758 | |
| 759 | if(!msg) |
| 760 | return CURLE_OK; |
| 761 | |
| 762 | h = msg; |
| 763 | |
| 764 | switch(req->state) { |
| 765 | case SMB_TREE_CONNECT: |
| 766 | if(h->status) { |
| 767 | req->result = CURLE_REMOTE_FILE_NOT_FOUND; |
| 768 | if(h->status == smb_swap32(SMB_ERR_NOACCESS)) |
| 769 | req->result = CURLE_REMOTE_ACCESS_DENIED; |
| 770 | break; |
| 771 | } |
| 772 | req->tid = smb_swap16(h->tid); |
| 773 | next_state = SMB_OPEN; |
| 774 | break; |
| 775 | |
| 776 | case SMB_OPEN: |
Elliott Hughes | cee0338 | 2017-06-23 12:17:18 -0700 | [diff] [blame] | 777 | if(h->status || smbc->got < sizeof(struct smb_nt_create_response)) { |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 778 | req->result = CURLE_REMOTE_FILE_NOT_FOUND; |
| 779 | next_state = SMB_TREE_DISCONNECT; |
| 780 | break; |
| 781 | } |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 782 | smb_m = (const struct smb_nt_create_response*) msg; |
| 783 | req->fid = smb_swap16(smb_m->fid); |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 784 | conn->data->req.offset = 0; |
| 785 | if(conn->data->set.upload) { |
| 786 | conn->data->req.size = conn->data->state.infilesize; |
| 787 | Curl_pgrsSetUploadSize(conn->data, conn->data->req.size); |
| 788 | next_state = SMB_UPLOAD; |
| 789 | } |
| 790 | else { |
Elliott Hughes | 82be86d | 2017-09-20 17:00:17 -0700 | [diff] [blame] | 791 | smb_m = (const struct smb_nt_create_response*) msg; |
| 792 | conn->data->req.size = smb_swap64(smb_m->end_of_file); |
Elliott Hughes | 1ef06ba | 2018-05-30 15:43:58 -0700 | [diff] [blame] | 793 | if(conn->data->req.size < 0) { |
| 794 | req->result = CURLE_WEIRD_SERVER_REPLY; |
| 795 | next_state = SMB_CLOSE; |
| 796 | } |
| 797 | else { |
| 798 | Curl_pgrsSetDownloadSize(conn->data, conn->data->req.size); |
| 799 | if(conn->data->set.get_filetime) |
| 800 | get_posix_time(&conn->data->info.filetime, smb_m->last_change_time); |
| 801 | next_state = SMB_DOWNLOAD; |
| 802 | } |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 803 | } |
| 804 | break; |
| 805 | |
| 806 | case SMB_DOWNLOAD: |
Elliott Hughes | cee0338 | 2017-06-23 12:17:18 -0700 | [diff] [blame] | 807 | if(h->status || smbc->got < sizeof(struct smb_header) + 14) { |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 808 | req->result = CURLE_RECV_ERROR; |
| 809 | next_state = SMB_CLOSE; |
| 810 | break; |
| 811 | } |
Elliott Hughes | cee0338 | 2017-06-23 12:17:18 -0700 | [diff] [blame] | 812 | len = Curl_read16_le(((const unsigned char *) msg) + |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 813 | sizeof(struct smb_header) + 11); |
Elliott Hughes | cee0338 | 2017-06-23 12:17:18 -0700 | [diff] [blame] | 814 | off = Curl_read16_le(((const unsigned char *) msg) + |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 815 | sizeof(struct smb_header) + 13); |
| 816 | if(len > 0) { |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 817 | if(off + sizeof(unsigned int) + len > smbc->got) { |
| 818 | failf(conn->data, "Invalid input packet"); |
| 819 | result = CURLE_RECV_ERROR; |
| 820 | } |
| 821 | else |
| 822 | result = Curl_client_write(conn, CLIENTWRITE_BODY, |
| 823 | (char *)msg + off + sizeof(unsigned int), |
| 824 | len); |
| 825 | if(result) { |
| 826 | req->result = result; |
| 827 | next_state = SMB_CLOSE; |
| 828 | break; |
| 829 | } |
| 830 | } |
| 831 | conn->data->req.bytecount += len; |
| 832 | conn->data->req.offset += len; |
| 833 | Curl_pgrsSetDownloadCounter(conn->data, conn->data->req.bytecount); |
| 834 | next_state = (len < MAX_PAYLOAD_SIZE) ? SMB_CLOSE : SMB_DOWNLOAD; |
| 835 | break; |
| 836 | |
| 837 | case SMB_UPLOAD: |
Elliott Hughes | cee0338 | 2017-06-23 12:17:18 -0700 | [diff] [blame] | 838 | if(h->status || smbc->got < sizeof(struct smb_header) + 6) { |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 839 | req->result = CURLE_UPLOAD_FAILED; |
| 840 | next_state = SMB_CLOSE; |
| 841 | break; |
| 842 | } |
Elliott Hughes | cee0338 | 2017-06-23 12:17:18 -0700 | [diff] [blame] | 843 | len = Curl_read16_le(((const unsigned char *) msg) + |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 844 | sizeof(struct smb_header) + 5); |
| 845 | conn->data->req.bytecount += len; |
| 846 | conn->data->req.offset += len; |
| 847 | Curl_pgrsSetUploadCounter(conn->data, conn->data->req.bytecount); |
| 848 | if(conn->data->req.bytecount >= conn->data->req.size) |
| 849 | next_state = SMB_CLOSE; |
| 850 | else |
| 851 | next_state = SMB_UPLOAD; |
| 852 | break; |
| 853 | |
| 854 | case SMB_CLOSE: |
| 855 | /* We don't care if the close failed, proceed to tree disconnect anyway */ |
| 856 | next_state = SMB_TREE_DISCONNECT; |
| 857 | break; |
| 858 | |
| 859 | case SMB_TREE_DISCONNECT: |
| 860 | next_state = SMB_DONE; |
| 861 | break; |
| 862 | |
| 863 | default: |
| 864 | smb_pop_message(conn); |
| 865 | return CURLE_OK; /* ignore */ |
| 866 | } |
| 867 | |
| 868 | smb_pop_message(conn); |
| 869 | |
| 870 | switch(next_state) { |
| 871 | case SMB_OPEN: |
| 872 | result = smb_send_open(conn); |
| 873 | break; |
| 874 | |
| 875 | case SMB_DOWNLOAD: |
| 876 | result = smb_send_read(conn); |
| 877 | break; |
| 878 | |
| 879 | case SMB_UPLOAD: |
| 880 | result = smb_send_write(conn); |
| 881 | break; |
| 882 | |
| 883 | case SMB_CLOSE: |
| 884 | result = smb_send_close(conn); |
| 885 | break; |
| 886 | |
| 887 | case SMB_TREE_DISCONNECT: |
| 888 | result = smb_send_tree_disconnect(conn); |
| 889 | break; |
| 890 | |
| 891 | case SMB_DONE: |
| 892 | result = req->result; |
| 893 | *done = true; |
| 894 | break; |
| 895 | |
| 896 | default: |
| 897 | break; |
| 898 | } |
| 899 | |
| 900 | if(result) { |
| 901 | connclose(conn, "SMB: failed to send message"); |
| 902 | return result; |
| 903 | } |
| 904 | |
| 905 | request_state(conn, next_state); |
| 906 | |
| 907 | return CURLE_OK; |
| 908 | } |
| 909 | |
| 910 | static CURLcode smb_done(struct connectdata *conn, CURLcode status, |
| 911 | bool premature) |
| 912 | { |
| 913 | struct smb_request *req = conn->data->req.protop; |
| 914 | |
| 915 | (void) premature; |
| 916 | |
| 917 | Curl_safefree(req->share); |
| 918 | Curl_safefree(conn->data->req.protop); |
| 919 | |
| 920 | return status; |
| 921 | } |
| 922 | |
| 923 | static CURLcode smb_disconnect(struct connectdata *conn, bool dead) |
| 924 | { |
| 925 | struct smb_conn *smbc = &conn->proto.smbc; |
| 926 | struct smb_request *req = conn->data->req.protop; |
| 927 | |
| 928 | (void) dead; |
| 929 | |
| 930 | Curl_safefree(smbc->domain); |
| 931 | Curl_safefree(smbc->recv_buf); |
| 932 | |
| 933 | /* smb_done is not always called, so cleanup the request */ |
| 934 | if(req) { |
| 935 | Curl_safefree(req->share); |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 936 | } |
| 937 | |
| 938 | return CURLE_OK; |
| 939 | } |
| 940 | |
| 941 | static int smb_getsock(struct connectdata *conn, curl_socket_t *socks, |
| 942 | int numsocks) |
| 943 | { |
| 944 | struct smb_conn *smbc = &conn->proto.smbc; |
| 945 | |
| 946 | if(!numsocks) |
| 947 | return GETSOCK_BLANK; |
| 948 | |
| 949 | socks[0] = conn->sock[FIRSTSOCKET]; |
| 950 | |
| 951 | if(smbc->send_size || smbc->upload_size) |
| 952 | return GETSOCK_WRITESOCK(0); |
| 953 | |
| 954 | return GETSOCK_READSOCK(0); |
| 955 | } |
| 956 | |
| 957 | static CURLcode smb_parse_url_path(struct connectdata *conn) |
| 958 | { |
| 959 | CURLcode result = CURLE_OK; |
Alex Deymo | e3149cc | 2016-10-05 11:18:42 -0700 | [diff] [blame] | 960 | struct Curl_easy *data = conn->data; |
Bertrand SIMONNET | e6cd738 | 2015-07-01 15:39:44 -0700 | [diff] [blame] | 961 | struct smb_request *req = data->req.protop; |
| 962 | char *path; |
| 963 | char *slash; |
| 964 | |
| 965 | /* URL decode the path */ |
| 966 | result = Curl_urldecode(data, data->state.path, 0, &path, NULL, TRUE); |
| 967 | if(result) |
| 968 | return result; |
| 969 | |
| 970 | /* Parse the path for the share */ |
| 971 | req->share = strdup((*path == '/' || *path == '\\') ? path + 1 : path); |
| 972 | if(!req->share) { |
| 973 | free(path); |
| 974 | |
| 975 | return CURLE_OUT_OF_MEMORY; |
| 976 | } |
| 977 | |
| 978 | slash = strchr(req->share, '/'); |
| 979 | if(!slash) |
| 980 | slash = strchr(req->share, '\\'); |
| 981 | |
| 982 | /* The share must be present */ |
| 983 | if(!slash) { |
| 984 | free(path); |
| 985 | |
| 986 | return CURLE_URL_MALFORMAT; |
| 987 | } |
| 988 | |
| 989 | /* Parse the path for the file path converting any forward slashes into |
| 990 | backslashes */ |
| 991 | *slash++ = 0; |
| 992 | req->path = slash; |
| 993 | for(; *slash; slash++) { |
| 994 | if(*slash == '/') |
| 995 | *slash = '\\'; |
| 996 | } |
| 997 | |
| 998 | free(path); |
| 999 | |
| 1000 | return CURLE_OK; |
| 1001 | } |
| 1002 | |
| 1003 | #endif /* !USE_WINDOWS_SSPI || USE_WIN32_CRYPTO */ |
| 1004 | |
| 1005 | #endif /* CURL_DISABLE_SMB && USE_NTLM && CURL_SIZEOF_CURL_OFF_T > 4 */ |