blob: 734b16ec6b199db71be8694bc322b60f2936fcdc [file] [log] [blame]
Mike McCormackfc932612002-03-12 19:24:04 +00001/*
2 * Copyright (C) 2002 Mike McCormack
3 *
4 * CIFS implementation for WINE
5 *
6 * This is a WINE's implementation of the Common Internet File System
7 *
8 * for specification see:
9 *
10 * http://www.codefx.com/CIFS_Explained.htm
11 * http://www.ubiqx.org/cifs/rfc-draft/rfc1002.html
12 * http://www.ubiqx.org/cifs/rfc-draft/draft-leach-cifs-v1-spec-02.html
13 * http://ubiqx.org/cifs/
14 * http://www.samba.org
15 *
16 * This library is free software; you can redistribute it and/or
17 * modify it under the terms of the GNU Lesser General Public
18 * License as published by the Free Software Foundation; either
19 * version 2.1 of the License, or (at your option) any later version.
20 *
21 * This library is distributed in the hope that it will be useful,
22 * but WITHOUT ANY WARRANTY; without even the implied warranty of
23 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
24 * Lesser General Public License for more details.
25 *
26 * You should have received a copy of the GNU Lesser General Public
27 * License along with this library; if not, write to the Free Software
28 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
Mike McCormack963985b2002-07-19 03:17:19 +000029 *
30 *
31 * FIXME:
32 *
33 * - There is a race condition when two threads try to read from the same
34 * SMB handle. Either we need to lock the SMB handle for the time we
35 * use it in the client, or do all reading and writing to the socket
36 * fd in the server.
37 *
38 * - Each new handle opens up a new connection to the SMB server. This
39 * is not ideal, since operations can be multiplexed on one socket. For
40 * this to work properly we would need to have some way of discovering
41 * connections that are already open.
42 *
43 * - All access is currently anonymous. Password protected shares cannot
44 * be accessed. We need some way of organising passwords, storing them
45 * in the config file, or putting up a dialog box for the user.
46 *
47 * - We don't deal with SMB dialects at all.
48 *
49 * - SMB supports passing unicode over the wire, should use this if possible.
50 *
51 * - Implement ability to read named pipes over the network. Would require
52 * integrate this code with the named pipes code in the server, and
53 * possibly implementing some support for security tokens.
Mike McCormackfc932612002-03-12 19:24:04 +000054 */
55
56#include "config.h"
57#include "wine/port.h"
58
59#include <assert.h>
60#include <ctype.h>
Mike McCormackfc932612002-03-12 19:24:04 +000061#include <fcntl.h>
62#include <stdlib.h>
Alexandre Julliarde37c6e12003-09-05 23:08:26 +000063#include <stdarg.h>
Mike McCormackfc932612002-03-12 19:24:04 +000064#include <stdio.h>
65#include <string.h>
Mike McCormackfc932612002-03-12 19:24:04 +000066#include <sys/types.h>
67#include <sys/stat.h>
68#ifdef HAVE_SYS_MMAN_H
69#include <sys/mman.h>
70#endif
Patrik Stridvalld016f812002-08-17 00:43:16 +000071#ifdef HAVE_SYS_TIME_H
72# include <sys/time.h>
73#endif
Patrik Stridvall57bf4502002-08-26 21:53:24 +000074#ifdef HAVE_SYS_POLL_H
75# include <sys/poll.h>
76#endif
Mike McCormackfc932612002-03-12 19:24:04 +000077#include <time.h>
Patrik Stridvalld016f812002-08-17 00:43:16 +000078#ifdef HAVE_UNISTD_H
79# include <unistd.h>
80#endif
Patrik Stridvall57bf4502002-08-26 21:53:24 +000081#ifdef HAVE_UTIME_H
82# include <utime.h>
83#endif
Patrik Stridvallf89d4a82002-03-23 21:39:05 +000084#ifdef HAVE_SYS_SOCKET_H
85# include <sys/socket.h>
86#endif
Mike McCormackfc932612002-03-12 19:24:04 +000087#include <sys/types.h>
Gerald Pfeiferc8764e42002-03-19 02:06:25 +000088#ifdef HAVE_NETINET_IN_SYSTM_H
89#include <netinet/in_systm.h>
90#endif
91#ifdef HAVE_NETINET_IN_H
92#include <netinet/in.h>
93#endif
94#ifdef HAVE_NETINET_IP_H
Mike McCormackfc932612002-03-12 19:24:04 +000095#include <netinet/ip.h>
Gerald Pfeiferc8764e42002-03-19 02:06:25 +000096#endif
97#ifdef HAVE_ARPA_INET_H
Mike McCormackfc932612002-03-12 19:24:04 +000098#include <arpa/inet.h>
Gerald Pfeiferc8764e42002-03-19 02:06:25 +000099#endif
Mike McCormack9414adf2002-05-05 20:29:15 +0000100#ifdef HAVE_NETDB_H
101#include <netdb.h>
102#endif
Mike McCormackfc932612002-03-12 19:24:04 +0000103
Eric Pouech9bd4f6b2003-06-26 02:08:17 +0000104#define NONAMELESSUNION
105#define NONAMELESSSTRUCT
Mike McCormackfc932612002-03-12 19:24:04 +0000106#include "winerror.h"
Alexandre Julliarde37c6e12003-09-05 23:08:26 +0000107#include "ntstatus.h"
Mike McCormackfc932612002-03-12 19:24:04 +0000108#include "windef.h"
109#include "winbase.h"
Alexandre Julliard43690e92003-01-24 00:54:58 +0000110#include "winnls.h"
Mike McCormackfc932612002-03-12 19:24:04 +0000111#include "file.h"
Mike McCormackfc932612002-03-12 19:24:04 +0000112
113#include "smb.h"
Eric Pouechc962a692003-06-23 18:12:28 +0000114#include "winternl.h"
115#include "ntdll_misc.h"
Mike McCormackfc932612002-03-12 19:24:04 +0000116
117#include "wine/server.h"
118#include "wine/debug.h"
119
120WINE_DEFAULT_DEBUG_CHANNEL(file);
121
Bill Medlandaf81a022003-06-24 19:23:25 +0000122#define NBR_ADDWORD(p,word) { (p)[1] = (word & 0xff); (p)[0] = ((word)>>8)&0xff; }
123#define NBR_GETWORD(p) ( (((p)[0])<<8) | ((p)[1]) )
124
125#define SMB_ADDWORD(p,word) { (p)[0] = (word & 0xff); (p)[1] = ((word)>>8)&0xff; }
126#define SMB_GETWORD(p) ( (((p)[1])<<8) | ((p)[0]) )
127#define SMB_ADDDWORD(p,w) { (p)[3]=((w)>>24)&0xff; (p)[2]=((w)>>16)&0xff; (p)[1]=((w)>>8)&0xff; (p)[0]=(w)&0xff; }
128#define SMB_GETDWORD(p) ( (((p)[3])<<24) | (((p)[2])<<16) | (((p)[1])<<8) | ((p)[0]) )
129
130#define SMB_COM_CREATE_DIRECTORY 0x00
131#define SMB_COM_DELETE_DIRECTORY 0x01
132#define SMB_COM_OPEN 0x02
133#define SMB_COM_CREATE 0x03
134#define SMB_COM_CLOSE 0x04
135#define SMB_COM_FLUSH 0x05
136#define SMB_COM_DELETE 0x06
137#define SMB_COM_RENAME 0x07
138#define SMB_COM_QUERY_INFORMATION 0x08
139#define SMB_COM_SET_INFORMATION 0x09
140#define SMB_COM_READ 0x0A
141#define SMB_COM_WRITE 0x0B
142#define SMB_COM_LOCK_BYTE_RANGE 0x0C
143#define SMB_COM_UNLOCK_BYTE_RANGE 0x0D
144#define SMB_COM_CREATE_TEMPORARY 0x0E
145#define SMB_COM_CREATE_NEW 0x0F
146#define SMB_COM_CHECK_DIRECTORY 0x10
147#define SMB_COM_PROCESS_EXIT 0x11
148#define SMB_COM_SEEK 0x12
149#define SMB_COM_LOCK_AND_READ 0x13
150#define SMB_COM_WRITE_AND_UNLOCK 0x14
151#define SMB_COM_READ_RAW 0x1A
152#define SMB_COM_READ_MPX 0x1B
153#define SMB_COM_READ_MPX_SECONDARY 0x1C
154#define SMB_COM_WRITE_RAW 0x1D
155#define SMB_COM_WRITE_MPX 0x1E
156#define SMB_COM_WRITE_COMPLETE 0x20
157#define SMB_COM_SET_INFORMATION2 0x22
158#define SMB_COM_QUERY_INFORMATION2 0x23
159#define SMB_COM_LOCKING_ANDX 0x24
160#define SMB_COM_TRANSACTION 0x25
161#define SMB_COM_TRANSACTION_SECONDARY 0x26
162#define SMB_COM_IOCTL 0x27
163#define SMB_COM_IOCTL_SECONDARY 0x28
164#define SMB_COM_COPY 0x29
165#define SMB_COM_MOVE 0x2A
166#define SMB_COM_ECHO 0x2B
167#define SMB_COM_WRITE_AND_CLOSE 0x2C
168#define SMB_COM_OPEN_ANDX 0x2D
169#define SMB_COM_READ_ANDX 0x2E
170#define SMB_COM_WRITE_ANDX 0x2F
171#define SMB_COM_CLOSE_AND_TREE_DISC 0x31
172#define SMB_COM_TRANSACTION2 0x32
173#define SMB_COM_TRANSACTION2_SECONDARY 0x33
174#define SMB_COM_FIND_CLOSE2 0x34
175#define SMB_COM_FIND_NOTIFY_CLOSE 0x35
176#define SMB_COM_TREE_CONNECT 0x70
177#define SMB_COM_TREE_DISCONNECT 0x71
178#define SMB_COM_NEGOTIATE 0x72
179#define SMB_COM_SESSION_SETUP_ANDX 0x73
180#define SMB_COM_LOGOFF_ANDX 0x74
181#define SMB_COM_TREE_CONNECT_ANDX 0x75
182#define SMB_COM_QUERY_INFORMATION_DISK 0x80
183#define SMB_COM_SEARCH 0x81
184#define SMB_COM_FIND 0x82
185#define SMB_COM_FIND_UNIQUE 0x83
186#define SMB_COM_NT_TRANSACT 0xA0
187#define SMB_COM_NT_TRANSACT_SECONDARY 0xA1
188#define SMB_COM_NT_CREATE_ANDX 0xA2
189#define SMB_COM_NT_CANCEL 0xA4
190#define SMB_COM_OPEN_PRINT_FILE 0xC0
191#define SMB_COM_WRITE_PRINT_FILE 0xC1
192#define SMB_COM_CLOSE_PRINT_FILE 0xC2
193#define SMB_COM_GET_PRINT_QUEUE 0xC3
194
195#define TRANS2_FIND_FIRST2 0x01
196#define TRANS2_FIND_NEXT2 0x02
197
Mike McCormackfc932612002-03-12 19:24:04 +0000198#define MAX_HOST_NAME 15
199#define NB_TIMEOUT 10000
200
Bill Medlandaf81a022003-06-24 19:23:25 +0000201/* We only need the A versions locally currently */
202static inline int SMB_isSepA (CHAR c) {return (c == '\\' || c == '/');}
203static inline int SMB_isUNCA (LPCSTR filename) {return (filename && SMB_isSepW (filename[0]) && SMB_isSepW (filename[1]));}
204static inline CHAR *SMB_nextSepA (CHAR *s) {while (*s && !SMB_isSepA (*s)) s++; return (*s? s : 0);}
205/* NB SM_nextSepA cannot return const CHAR * since it is going to be used for
206 * replacing separators with null characters
207 */
208
Eric Pouech9bd4f6b2003-06-26 02:08:17 +0000209static USHORT SMB_MultiplexId = 0;
Mike McCormackfc932612002-03-12 19:24:04 +0000210
Mike McCormack963985b2002-07-19 03:17:19 +0000211struct NB_Buffer
212{
213 unsigned char *buffer;
214 int len;
215};
216
Mike McCormackfc932612002-03-12 19:24:04 +0000217static int netbios_name(const char *p, unsigned char *buffer)
218{
219 char ch;
220 int i,len=0;
221
222 buffer[len++]=' ';
223 for(i=0; i<=MAX_HOST_NAME; i++)
224 {
225 if(i<MAX_HOST_NAME)
226 {
227 if(*p)
228 ch = *p++&0xdf; /* add character from hostname */
229 else
230 ch = ' '; /* add padding */
231 }
232 else
233 ch = 0; /* add terminator */
234 buffer[len++] = ((ch&0xf0) >> 4) + 'A';
235 buffer[len++] = (ch&0x0f) + 'A';
236 }
237 buffer[len++] = 0; /* add second terminator */
238 return len;
239}
240
241static DWORD NB_NameReq(LPCSTR host, unsigned char *buffer, int len)
242{
243 int trn = 1234,i=0;
244
245 NBR_ADDWORD(&buffer[i],trn); i+=2;
246 NBR_ADDWORD(&buffer[i],0x0110); i+=2;
247 NBR_ADDWORD(&buffer[i],0x0001); i+=2;
248 NBR_ADDWORD(&buffer[i],0x0000); i+=2;
249 NBR_ADDWORD(&buffer[i],0x0000); i+=2;
250 NBR_ADDWORD(&buffer[i],0x0000); i+=2;
251
252 i += netbios_name(host,&buffer[i]);
Vincent Béron9a624912002-05-31 23:06:46 +0000253
Mike McCormackfc932612002-03-12 19:24:04 +0000254 NBR_ADDWORD(&buffer[i],0x0020); i+=2;
255 NBR_ADDWORD(&buffer[i],0x0001); i+=2;
256
Mike McCormack963985b2002-07-19 03:17:19 +0000257 TRACE("packet is %d bytes in length\n",i);
Mike McCormackfc932612002-03-12 19:24:04 +0000258
259 {
260 int j;
261 for(j=0; j<i; j++)
262 printf("%02x%c",buffer[j],(((j+1)%16)&&((j+1)!=j))?' ':'\n');
263 }
264
265 return i;
266}
267
268/* unc = \\hostname\share\file... */
269static BOOL UNC_SplitName(LPSTR unc, LPSTR *hostname, LPSTR *share, LPSTR *file)
270{
271 char *p;
272
Mike McCormack963985b2002-07-19 03:17:19 +0000273 TRACE("%s\n",unc);
Mike McCormackfc932612002-03-12 19:24:04 +0000274
Bill Medlandaf81a022003-06-24 19:23:25 +0000275 if (!SMB_isUNCA (unc))
Mike McCormackfc932612002-03-12 19:24:04 +0000276 return FALSE;
Bill Medlandaf81a022003-06-24 19:23:25 +0000277 p = unc + 2;
278 *hostname=p;
Mike McCormackfc932612002-03-12 19:24:04 +0000279
Bill Medlandaf81a022003-06-24 19:23:25 +0000280 p = SMB_nextSepA (p);
Mike McCormackfc932612002-03-12 19:24:04 +0000281 if(!p)
282 return FALSE;
283 *p=0;
284 *share = ++p;
285
Bill Medlandaf81a022003-06-24 19:23:25 +0000286 p = SMB_nextSepA (p);
Mike McCormackfc932612002-03-12 19:24:04 +0000287 if(!p)
288 return FALSE;
289 *p=0;
290 *file = ++p;
291
292 return TRUE;
293}
294
295static BOOL NB_Lookup(LPCSTR host, struct sockaddr_in *addr)
296{
Mike McCormack9414adf2002-05-05 20:29:15 +0000297 int fd,on=1,r,len,i,fromsize;
Mike McCormackfc932612002-03-12 19:24:04 +0000298 struct pollfd fds;
Mike McCormack9414adf2002-05-05 20:29:15 +0000299 struct sockaddr_in sin,fromaddr;
Mike McCormackfc932612002-03-12 19:24:04 +0000300 unsigned char buffer[256];
301
302 fd = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP);
303 if(fd<0)
304 return FALSE;
305
Francois Gouget95918362003-06-18 19:45:22 +0000306 r = setsockopt(fd, SOL_SOCKET, SO_BROADCAST, &on, sizeof(on));
Mike McCormackfc932612002-03-12 19:24:04 +0000307 if(r<0)
Mike McCormack9414adf2002-05-05 20:29:15 +0000308 goto err;
Mike McCormackfc932612002-03-12 19:24:04 +0000309
310 if(0==inet_aton("255.255.255.255", (struct in_addr *)&sin.sin_addr.s_addr))
311 {
312 FIXME("Error getting bcast address\n");
Mike McCormack9414adf2002-05-05 20:29:15 +0000313 goto err;
Mike McCormackfc932612002-03-12 19:24:04 +0000314 }
315 sin.sin_family = AF_INET;
316 sin.sin_port = htons(137);
317
Francois Gouget95918362003-06-18 19:45:22 +0000318 len = NB_NameReq(host,buffer,sizeof(buffer));
Mike McCormackfc932612002-03-12 19:24:04 +0000319 if(len<=0)
Mike McCormack9414adf2002-05-05 20:29:15 +0000320 goto err;
Mike McCormackfc932612002-03-12 19:24:04 +0000321
Francois Gouget95918362003-06-18 19:45:22 +0000322 r = sendto(fd, buffer, len, 0, (struct sockaddr*)&sin, sizeof(sin));
Mike McCormackfc932612002-03-12 19:24:04 +0000323 if(r<0)
324 {
325 FIXME("Error sending packet\n");
Mike McCormack9414adf2002-05-05 20:29:15 +0000326 goto err;
Mike McCormackfc932612002-03-12 19:24:04 +0000327 }
328
329 fds.fd = fd;
330 fds.events = POLLIN;
331 fds.revents = 0;
332
Mike McCormack9414adf2002-05-05 20:29:15 +0000333 /* FIXME: this is simple and easily fooled logic
334 * we should loop until we receive the correct packet or timeout
335 */
Mike McCormackfc932612002-03-12 19:24:04 +0000336 r = poll(&fds,1,NB_TIMEOUT);
337 if(r!=1)
Mike McCormack9414adf2002-05-05 20:29:15 +0000338 goto err;
339
340 TRACE("Got response!\n");
341
342 fromsize = sizeof (fromaddr);
Francois Gouget95918362003-06-18 19:45:22 +0000343 r = recvfrom(fd, buffer, sizeof(buffer), 0, (struct sockaddr*)&fromaddr, &fromsize);
Mike McCormack9414adf2002-05-05 20:29:15 +0000344 if(r<0)
345 goto err;
346
Mike McCormack963985b2002-07-19 03:17:19 +0000347 TRACE("%d bytes received\n",r);
Mike McCormack9414adf2002-05-05 20:29:15 +0000348
349 if(r!=62)
350 goto err;
351
352 for(i=0; i<r; i++)
353 DPRINTF("%02X%c",buffer[i],(((i+1)!=r)&&((i+1)%16))?' ':'\n');
354 DPRINTF("\n");
355
356 if(0x0f & buffer[3])
357 goto err;
358
Mike McCormack963985b2002-07-19 03:17:19 +0000359 TRACE("packet is OK\n");
Mike McCormack9414adf2002-05-05 20:29:15 +0000360
Francois Gouget95918362003-06-18 19:45:22 +0000361 memcpy(&addr->sin_addr, &buffer[58], sizeof(addr->sin_addr));
Mike McCormackfc932612002-03-12 19:24:04 +0000362
363 close(fd);
Mike McCormackfc932612002-03-12 19:24:04 +0000364 return TRUE;
Mike McCormack9414adf2002-05-05 20:29:15 +0000365
366err:
367 close(fd);
368 return FALSE;
Mike McCormackfc932612002-03-12 19:24:04 +0000369}
370
371#define NB_FIRST 0x40
372
373#define NB_HDRSIZE 4
374
375#define NB_SESSION_MSG 0x00
376#define NB_SESSION_REQ 0x81
377
378/* RFC 1002, section 4.3.2 */
379static BOOL NB_SessionReq(int fd, char *called, char *calling)
380{
381 unsigned char buffer[0x100];
382 int len = 0,r;
383 struct pollfd fds;
384
Mike McCormack963985b2002-07-19 03:17:19 +0000385 TRACE("called %s, calling %s\n",called,calling);
Mike McCormackfc932612002-03-12 19:24:04 +0000386
387 buffer[0] = NB_SESSION_REQ;
388 buffer[1] = NB_FIRST;
389
390 netbios_name(called, &buffer[NB_HDRSIZE]);
391 len += 34;
392 netbios_name(calling, &buffer[NB_HDRSIZE+len]);
393 len += 34;
394
395 NBR_ADDWORD(&buffer[2],len);
396
397 /* for(i=0; i<(len+NB_HDRSIZE); i++)
398 DPRINTF("%02X%c",buffer[i],(((i+1)!=(len+4))&&((i+1)%16))?' ':'\n'); */
399
400 r = write(fd,buffer,len+4);
401 if(r<0)
402 {
403 ERR("Write failed\n");
404 return FALSE;
405 }
406
407 fds.fd = fd;
408 fds.events = POLLIN;
409 fds.revents = 0;
410
411 r = poll(&fds,1,NB_TIMEOUT);
412 if(r!=1)
413 {
414 ERR("Poll failed\n");
415 return FALSE;
416 }
417
418 r = read(fd, buffer, NB_HDRSIZE);
419 if((r!=NB_HDRSIZE) || (buffer[0]!=0x82))
420 {
Mike McCormack963985b2002-07-19 03:17:19 +0000421 TRACE("Received %d bytes\n",r);
422 TRACE("%02x %02x %02x %02x\n", buffer[0],buffer[1],buffer[2],buffer[3]);
Mike McCormackfc932612002-03-12 19:24:04 +0000423 return FALSE;
424 }
425
426 return TRUE;
427}
428
Mike McCormack963985b2002-07-19 03:17:19 +0000429static BOOL NB_SendData(int fd, struct NB_Buffer *out)
Mike McCormackfc932612002-03-12 19:24:04 +0000430{
431 unsigned char buffer[NB_HDRSIZE];
432 int r;
433
434 /* CHECK: is it always OK to do this in two writes? */
435 /* perhaps use scatter gather sendmsg instead? */
436
437 buffer[0] = NB_SESSION_MSG;
438 buffer[1] = NB_FIRST;
Mike McCormack963985b2002-07-19 03:17:19 +0000439 NBR_ADDWORD(&buffer[2],out->len);
Mike McCormackfc932612002-03-12 19:24:04 +0000440
441 r = write(fd, buffer, NB_HDRSIZE);
442 if(r!=NB_HDRSIZE)
443 return FALSE;
444
Mike McCormack963985b2002-07-19 03:17:19 +0000445 r = write(fd, out->buffer, out->len);
446 if(r!=out->len)
Mike McCormackfc932612002-03-12 19:24:04 +0000447 {
448 ERR("write failed\n");
449 return FALSE;
450 }
451
452 return TRUE;
453}
454
Mike McCormack963985b2002-07-19 03:17:19 +0000455static BOOL NB_RecvData(int fd, struct NB_Buffer *rx)
Mike McCormackfc932612002-03-12 19:24:04 +0000456{
Mike McCormack963985b2002-07-19 03:17:19 +0000457 int r;
Mike McCormackfc932612002-03-12 19:24:04 +0000458 unsigned char buffer[NB_HDRSIZE];
459
460 r = read(fd, buffer, NB_HDRSIZE);
461 if((r!=NB_HDRSIZE) || (buffer[0]!=NB_SESSION_MSG))
462 {
463 ERR("Received %d bytes\n",r);
464 return FALSE;
465 }
466
Mike McCormack963985b2002-07-19 03:17:19 +0000467 rx->len = NBR_GETWORD(&buffer[2]);
468
Eric Pouechc962a692003-06-23 18:12:28 +0000469 rx->buffer = RtlAllocateHeap(ntdll_get_process_heap(), 0, rx->len);
Mike McCormack963985b2002-07-19 03:17:19 +0000470 if(!rx->buffer)
471 return FALSE;
472
473 r = read(fd, rx->buffer, rx->len);
474 if(rx->len!=r)
Mike McCormackfc932612002-03-12 19:24:04 +0000475 {
Mike McCormack963985b2002-07-19 03:17:19 +0000476 TRACE("Received %d bytes\n",r);
Eric Pouechc962a692003-06-23 18:12:28 +0000477 RtlFreeHeap(ntdll_get_process_heap(), 0, rx->buffer);
Mike McCormack963985b2002-07-19 03:17:19 +0000478 rx->buffer = 0;
479 rx->len = 0;
Mike McCormackfc932612002-03-12 19:24:04 +0000480 return FALSE;
481 }
Mike McCormackfc932612002-03-12 19:24:04 +0000482
483 return TRUE;
484}
485
Mike McCormack963985b2002-07-19 03:17:19 +0000486static BOOL NB_Transaction(int fd, struct NB_Buffer *in, struct NB_Buffer *out)
Mike McCormackfc932612002-03-12 19:24:04 +0000487{
Mike McCormack963985b2002-07-19 03:17:19 +0000488 int r;
Mike McCormackfc932612002-03-12 19:24:04 +0000489 struct pollfd fds;
490
Mike McCormack963985b2002-07-19 03:17:19 +0000491 if(TRACE_ON(file))
492 {
493 int i;
Mike McCormackfc932612002-03-12 19:24:04 +0000494 DPRINTF("Sending request:\n");
Mike McCormack963985b2002-07-19 03:17:19 +0000495 for(i=0; i<in->len; i++)
496 DPRINTF("%02X%c",in->buffer[i],(((i+1)!=in->len)&&((i+1)%16))?' ':'\n');
497 }
Mike McCormackfc932612002-03-12 19:24:04 +0000498
Mike McCormack963985b2002-07-19 03:17:19 +0000499 if(!NB_SendData(fd,in))
Mike McCormackfc932612002-03-12 19:24:04 +0000500 return FALSE;
501
502 fds.fd = fd;
503 fds.events = POLLIN;
504 fds.revents = 0;
505
506 r = poll(&fds,1,NB_TIMEOUT);
507 if(r!=1)
508 {
509 ERR("Poll failed\n");
510 return FALSE;
511 }
512
Mike McCormack963985b2002-07-19 03:17:19 +0000513 if(!NB_RecvData(fd, out))
Mike McCormackfc932612002-03-12 19:24:04 +0000514 return FALSE;
515
Mike McCormack963985b2002-07-19 03:17:19 +0000516 if(TRACE_ON(file))
517 {
518 int i;
Mike McCormackfc932612002-03-12 19:24:04 +0000519 DPRINTF("Got response:\n");
Mike McCormack963985b2002-07-19 03:17:19 +0000520 for(i=0; i<out->len; i++)
521 DPRINTF("%02X%c",out->buffer[i],(((i+1)!=out->len)&&((i+1)%16))?' ':'\n');
522 }
Mike McCormackfc932612002-03-12 19:24:04 +0000523
524 return TRUE;
525}
526
527#define SMB_ADDHEADER(b,l) { b[(l)++]=0xff; b[(l)++]='S'; b[(l)++]='M'; b[(l)++]='B'; }
528#define SMB_ADDERRINFO(b,l) { b[(l)++]=0; b[(l)++]=0; b[(l)++]=0; b[(l)++]=0; }
529#define SMB_ADDPADSIG(b,l) { memset(&b[l],0,12); l+=12; }
530
531#define SMB_ERRCLASS 5
532#define SMB_ERRCODE 7
533#define SMB_TREEID 24
534#define SMB_PROCID 26
535#define SMB_USERID 28
536#define SMB_PLEXID 30
537#define SMB_PCOUNT 32
538#define SMB_HDRSIZE 33
539
540static DWORD SMB_GetError(unsigned char *buffer)
541{
Mike McCormack963985b2002-07-19 03:17:19 +0000542 char *err_class;
543
544 switch(buffer[SMB_ERRCLASS])
545 {
546 case 0:
Mike McCormackfc932612002-03-12 19:24:04 +0000547 return STATUS_SUCCESS;
Mike McCormack963985b2002-07-19 03:17:19 +0000548 case 1:
549 err_class = "DOS";
550 break;
551 case 2:
552 err_class = "net server";
553 break;
554 case 3:
555 err_class = "hardware";
556 break;
557 case 0xff:
558 err_class = "smb";
559 break;
560 default:
561 err_class = "unknown";
562 break;
563 }
564
565 ERR("%s error %d \n",err_class, buffer[SMB_ERRCODE]);
566
Mike McCormackfc932612002-03-12 19:24:04 +0000567 /* FIXME: return propper error codes */
568 return STATUS_INVALID_PARAMETER;
569}
570
571static int SMB_Header(unsigned char *buffer, unsigned char command, USHORT tree_id, USHORT user_id)
572{
573 int len = 0;
574 DWORD id;
575
576 /* 0 */
Vincent Béron9a624912002-05-31 23:06:46 +0000577 SMB_ADDHEADER(buffer,len);
Mike McCormackfc932612002-03-12 19:24:04 +0000578
579 /* 4 */
580 buffer[len++] = command;
581
582 /* 5 */
583 SMB_ADDERRINFO(buffer,len)
584
585 /* 9 */
586 buffer[len++] = 0x00; /* flags */
587 SMB_ADDWORD(&buffer[len],1); len += 2; /* flags2 */
588
589 /* 12 */
590 SMB_ADDPADSIG(buffer,len)
591
592 /* 24 */
593 SMB_ADDWORD(&buffer[len],tree_id); len += 2; /* treeid */
594 id = GetCurrentThreadId();
595 SMB_ADDWORD(&buffer[len],id); len += 2; /* process id */
596 SMB_ADDWORD(&buffer[len],user_id); len += 2; /* user id */
597 SMB_ADDWORD(&buffer[len],SMB_MultiplexId); len += 2; /* multiplex id */
598 SMB_MultiplexId++;
599
600 return len;
601}
602
603static const char *SMB_ProtocolDialect = "NT LM 0.12";
604/* = "Windows for Workgroups 3.1a"; */
605
606/* FIXME: support multiple SMB dialects */
607static BOOL SMB_NegotiateProtocol(int fd, USHORT *dialect)
608{
Mike McCormack963985b2002-07-19 03:17:19 +0000609 unsigned char buf[0x100];
610 int buflen = 0;
611 struct NB_Buffer tx, rx;
Mike McCormackfc932612002-03-12 19:24:04 +0000612
Mike McCormack963985b2002-07-19 03:17:19 +0000613 TRACE("\n");
Mike McCormackfc932612002-03-12 19:24:04 +0000614
Francois Gouget95918362003-06-18 19:45:22 +0000615 memset(buf,0,sizeof(buf));
Mike McCormackfc932612002-03-12 19:24:04 +0000616
Mike McCormack963985b2002-07-19 03:17:19 +0000617 tx.buffer = buf;
618 tx.len = SMB_Header(tx.buffer, SMB_COM_NEGOTIATE, 0, 0);
Vincent Béron9a624912002-05-31 23:06:46 +0000619
Mike McCormackfc932612002-03-12 19:24:04 +0000620 /* parameters */
Mike McCormack963985b2002-07-19 03:17:19 +0000621 tx.buffer[tx.len++] = 0; /* no parameters */
Mike McCormackfc932612002-03-12 19:24:04 +0000622
623 /* command buffer */
624 buflen = strlen(SMB_ProtocolDialect)+2; /* include type and nul byte */
Mike McCormack963985b2002-07-19 03:17:19 +0000625 SMB_ADDWORD(&tx.buffer[tx.len],buflen); tx.len += 2;
Mike McCormackfc932612002-03-12 19:24:04 +0000626
Mike McCormack963985b2002-07-19 03:17:19 +0000627 tx.buffer[tx.len] = 0x02;
628 strcpy(&tx.buffer[tx.len+1],SMB_ProtocolDialect);
629 tx.len += buflen;
Mike McCormackfc932612002-03-12 19:24:04 +0000630
Mike McCormack963985b2002-07-19 03:17:19 +0000631 rx.buffer = NULL;
632 rx.len = 0;
633 if(!NB_Transaction(fd, &tx, &rx))
Mike McCormackfc932612002-03-12 19:24:04 +0000634 {
635 ERR("Failed\n");
636 return FALSE;
637 }
638
Mike McCormack963985b2002-07-19 03:17:19 +0000639 if(!rx.buffer)
640 return FALSE;
641
Mike McCormackfc932612002-03-12 19:24:04 +0000642 /* FIXME: check response */
Mike McCormack963985b2002-07-19 03:17:19 +0000643 if(SMB_GetError(rx.buffer))
Mike McCormackfc932612002-03-12 19:24:04 +0000644 {
645 ERR("returned error\n");
Eric Pouechc962a692003-06-23 18:12:28 +0000646 RtlFreeHeap(ntdll_get_process_heap(),0,rx.buffer);
Mike McCormackfc932612002-03-12 19:24:04 +0000647 return FALSE;
648 }
649
Eric Pouechc962a692003-06-23 18:12:28 +0000650 RtlFreeHeap(ntdll_get_process_heap(),0,rx.buffer);
Mike McCormack963985b2002-07-19 03:17:19 +0000651
Mike McCormackfc932612002-03-12 19:24:04 +0000652 *dialect = 0;
653
654 return TRUE;
655}
656
657#define SMB_PARAM_COUNT(buffer) ((buffer)[SMB_PCOUNT])
658#define SMB_PARAM(buffer,n) SMB_GETWORD(&(buffer)[SMB_HDRSIZE+2*(n)])
659#define SMB_BUFFER_COUNT(buffer) SMB_GETWORD(buffer+SMB_HDRSIZE+2*SMB_PARAM_COUNT(buffer))
660#define SMB_BUFFER(buffer,n) ((buffer)[SMB_HDRSIZE + 2*SMB_PARAM_COUNT(buffer) + 2 + (n) ])
661
662static BOOL SMB_SessionSetup(int fd, USHORT *userid)
663{
Mike McCormack963985b2002-07-19 03:17:19 +0000664 unsigned char buf[0x100];
665 int pcount,bcount;
666 struct NB_Buffer rx, tx;
Mike McCormackfc932612002-03-12 19:24:04 +0000667
Francois Gouget95918362003-06-18 19:45:22 +0000668 memset(buf,0,sizeof(buf));
Mike McCormack963985b2002-07-19 03:17:19 +0000669 tx.buffer = buf;
Mike McCormackfc932612002-03-12 19:24:04 +0000670
Mike McCormack963985b2002-07-19 03:17:19 +0000671 tx.len = SMB_Header(tx.buffer, SMB_COM_SESSION_SETUP_ANDX, 0, 0);
Mike McCormackfc932612002-03-12 19:24:04 +0000672
Mike McCormack963985b2002-07-19 03:17:19 +0000673 tx.buffer[tx.len++] = 0; /* no parameters? */
Mike McCormackfc932612002-03-12 19:24:04 +0000674
Mike McCormack963985b2002-07-19 03:17:19 +0000675 tx.buffer[tx.len++] = 0xff; /* AndXCommand: secondary request */
676 tx.buffer[tx.len++] = 0x00; /* AndXReserved */
677 SMB_ADDWORD(&tx.buffer[tx.len],0); /* AndXOffset */
678 tx.len += 2;
679 SMB_ADDWORD(&tx.buffer[tx.len],0x400); /* MaxBufferSize */
680 tx.len += 2;
681 SMB_ADDWORD(&tx.buffer[tx.len],1); /* MaxMpxCount */
682 tx.len += 2;
683 SMB_ADDWORD(&tx.buffer[tx.len],0); /* VcNumber */
684 tx.len += 2;
685 SMB_ADDWORD(&tx.buffer[tx.len],0); /* SessionKey */
686 tx.len += 2;
687 SMB_ADDWORD(&tx.buffer[tx.len],0); /* SessionKey */
688 tx.len += 2;
689 SMB_ADDWORD(&tx.buffer[tx.len],0); /* Password length */
690 tx.len += 2;
691 SMB_ADDWORD(&tx.buffer[tx.len],0); /* Reserved */
692 tx.len += 2;
693 SMB_ADDWORD(&tx.buffer[tx.len],0); /* Reserved */
694 tx.len += 2;
Mike McCormackfc932612002-03-12 19:24:04 +0000695
696 /* FIXME: add name and password here */
Mike McCormack963985b2002-07-19 03:17:19 +0000697 tx.buffer[tx.len++] = 0; /* number of bytes in password */
Mike McCormackfc932612002-03-12 19:24:04 +0000698
Mike McCormack963985b2002-07-19 03:17:19 +0000699 rx.buffer = NULL;
700 rx.len = 0;
701 if(!NB_Transaction(fd, &tx, &rx))
Mike McCormackfc932612002-03-12 19:24:04 +0000702 return FALSE;
703
Mike McCormack963985b2002-07-19 03:17:19 +0000704 if(!rx.buffer)
Mike McCormackfc932612002-03-12 19:24:04 +0000705 return FALSE;
706
Mike McCormack963985b2002-07-19 03:17:19 +0000707 if(SMB_GetError(rx.buffer))
708 goto done;
Mike McCormackfc932612002-03-12 19:24:04 +0000709
Mike McCormack963985b2002-07-19 03:17:19 +0000710 pcount = SMB_PARAM_COUNT(rx.buffer);
711
712 if( (SMB_HDRSIZE+pcount*2) > rx.len )
Mike McCormackfc932612002-03-12 19:24:04 +0000713 {
714 ERR("Bad parameter count %d\n",pcount);
Mike McCormack963985b2002-07-19 03:17:19 +0000715 goto done;
Mike McCormackfc932612002-03-12 19:24:04 +0000716 }
717
Mike McCormack963985b2002-07-19 03:17:19 +0000718 if(TRACE_ON(file))
719 {
720 int i;
Mike McCormackfc932612002-03-12 19:24:04 +0000721 DPRINTF("SMB_COM_SESSION_SETUP response, %d args: ",pcount);
722 for(i=0; i<pcount; i++)
Mike McCormack963985b2002-07-19 03:17:19 +0000723 DPRINTF("%04x ",SMB_PARAM(rx.buffer,i));
Mike McCormackfc932612002-03-12 19:24:04 +0000724 DPRINTF("\n");
Mike McCormackfc932612002-03-12 19:24:04 +0000725 }
726
Mike McCormack963985b2002-07-19 03:17:19 +0000727 bcount = SMB_BUFFER_COUNT(rx.buffer);
728 if( (SMB_HDRSIZE+pcount*2+2+bcount) > rx.len )
729 {
730 ERR("parameter count %x, buffer count %x, len %x\n",pcount,bcount,rx.len);
731 goto done;
732 }
733
734 if(TRACE_ON(file))
735 {
736 int i;
Mike McCormackfc932612002-03-12 19:24:04 +0000737 DPRINTF("response buffer %d bytes: ",bcount);
738 for(i=0; i<bcount; i++)
739 {
Mike McCormack963985b2002-07-19 03:17:19 +0000740 unsigned char ch = SMB_BUFFER(rx.buffer,i);
Mike McCormackfc932612002-03-12 19:24:04 +0000741 DPRINTF("%c", isprint(ch)?ch:' ');
742 }
743 DPRINTF("\n");
Mike McCormack963985b2002-07-19 03:17:19 +0000744 }
Mike McCormackfc932612002-03-12 19:24:04 +0000745
Mike McCormack963985b2002-07-19 03:17:19 +0000746 *userid = SMB_GETWORD(&rx.buffer[SMB_USERID]);
Mike McCormackfc932612002-03-12 19:24:04 +0000747
Eric Pouechc962a692003-06-23 18:12:28 +0000748 RtlFreeHeap(ntdll_get_process_heap(),0,rx.buffer);
Mike McCormackfc932612002-03-12 19:24:04 +0000749 return TRUE;
Mike McCormack963985b2002-07-19 03:17:19 +0000750
751done:
Eric Pouechc962a692003-06-23 18:12:28 +0000752 RtlFreeHeap(ntdll_get_process_heap(),0,rx.buffer);
Mike McCormack963985b2002-07-19 03:17:19 +0000753 return FALSE;
Mike McCormackfc932612002-03-12 19:24:04 +0000754}
755
Mike McCormack963985b2002-07-19 03:17:19 +0000756
Mike McCormackfc932612002-03-12 19:24:04 +0000757static BOOL SMB_TreeConnect(int fd, USHORT user_id, LPCSTR share_name, USHORT *treeid)
758{
Mike McCormack963985b2002-07-19 03:17:19 +0000759 unsigned char buf[0x100];
760 int slen;
761 struct NB_Buffer rx,tx;
Mike McCormackfc932612002-03-12 19:24:04 +0000762
Mike McCormack963985b2002-07-19 03:17:19 +0000763 TRACE("%s\n",share_name);
Mike McCormackfc932612002-03-12 19:24:04 +0000764
Francois Gouget95918362003-06-18 19:45:22 +0000765 memset(buf,0,sizeof(buf));
Mike McCormack963985b2002-07-19 03:17:19 +0000766 tx.buffer = buf;
Mike McCormackfc932612002-03-12 19:24:04 +0000767
Mike McCormack963985b2002-07-19 03:17:19 +0000768 tx.len = SMB_Header(tx.buffer, SMB_COM_TREE_CONNECT, 0, user_id);
Mike McCormackfc932612002-03-12 19:24:04 +0000769
Mike McCormack963985b2002-07-19 03:17:19 +0000770 tx.buffer[tx.len++] = 4; /* parameters */
Mike McCormackfc932612002-03-12 19:24:04 +0000771
Mike McCormack963985b2002-07-19 03:17:19 +0000772 tx.buffer[tx.len++] = 0xff; /* AndXCommand: secondary request */
773 tx.buffer[tx.len++] = 0x00; /* AndXReserved */
774 SMB_ADDWORD(&tx.buffer[tx.len],0); /* AndXOffset */
775 tx.len += 2;
776 SMB_ADDWORD(&tx.buffer[tx.len],0); /* Flags */
777 tx.len += 2;
778 SMB_ADDWORD(&tx.buffer[tx.len],1); /* Password length */
779 tx.len += 2;
Vincent Béron9a624912002-05-31 23:06:46 +0000780
Mike McCormackfc932612002-03-12 19:24:04 +0000781 /* SMB command buffer */
Mike McCormack963985b2002-07-19 03:17:19 +0000782 SMB_ADDWORD(&tx.buffer[tx.len],3); /* command buffer len */
783 tx.len += 2;
784 tx.buffer[tx.len++] = 0; /* null terminated password */
Mike McCormackfc932612002-03-12 19:24:04 +0000785
786 slen = strlen(share_name);
Francois Gouget95918362003-06-18 19:45:22 +0000787 if(slen<(sizeof(buf)-tx.len))
Mike McCormack963985b2002-07-19 03:17:19 +0000788 strcpy(&tx.buffer[tx.len], share_name);
Mike McCormackfc932612002-03-12 19:24:04 +0000789 else
790 return FALSE;
Mike McCormack963985b2002-07-19 03:17:19 +0000791 tx.len += slen+1;
Vincent Béron9a624912002-05-31 23:06:46 +0000792
Mike McCormackfc932612002-03-12 19:24:04 +0000793 /* name of the service */
Mike McCormack963985b2002-07-19 03:17:19 +0000794 tx.buffer[tx.len++] = 0;
Mike McCormackfc932612002-03-12 19:24:04 +0000795
Mike McCormack963985b2002-07-19 03:17:19 +0000796 rx.buffer = NULL;
797 rx.len = 0;
798 if(!NB_Transaction(fd, &tx, &rx))
Mike McCormackfc932612002-03-12 19:24:04 +0000799 return FALSE;
800
Mike McCormack963985b2002-07-19 03:17:19 +0000801 if(!rx.buffer)
Mike McCormackfc932612002-03-12 19:24:04 +0000802 return FALSE;
803
Mike McCormack963985b2002-07-19 03:17:19 +0000804 if(SMB_GetError(rx.buffer))
805 {
Eric Pouechc962a692003-06-23 18:12:28 +0000806 RtlFreeHeap(ntdll_get_process_heap(),0,rx.buffer);
Mike McCormack963985b2002-07-19 03:17:19 +0000807 return FALSE;
808 }
Mike McCormackfc932612002-03-12 19:24:04 +0000809
Mike McCormack963985b2002-07-19 03:17:19 +0000810 *treeid = SMB_GETWORD(&rx.buffer[SMB_TREEID]);
811
Eric Pouechc962a692003-06-23 18:12:28 +0000812 RtlFreeHeap(ntdll_get_process_heap(),0,rx.buffer);
Mike McCormack963985b2002-07-19 03:17:19 +0000813 TRACE("OK, treeid = %04x\n", *treeid);
Mike McCormackfc932612002-03-12 19:24:04 +0000814
815 return TRUE;
816}
817
Alexandre Julliard958732d2002-07-02 02:09:39 +0000818#if 0 /* not yet */
Mike McCormackfc932612002-03-12 19:24:04 +0000819static BOOL SMB_NtCreateOpen(int fd, USHORT tree_id, USHORT user_id, USHORT dialect,
820 LPCSTR filename, DWORD access, DWORD sharing,
821 LPSECURITY_ATTRIBUTES sa, DWORD creation,
822 DWORD attributes, HANDLE template, USHORT *file_id )
823{
824 unsigned char buffer[0x100];
825 int len = 0,slen;
826
Mike McCormack963985b2002-07-19 03:17:19 +0000827 TRACE("%s\n",filename);
Mike McCormackfc932612002-03-12 19:24:04 +0000828
Francois Gouget95918362003-06-18 19:45:22 +0000829 memset(buffer,0,sizeof(buffer));
Mike McCormackfc932612002-03-12 19:24:04 +0000830
831 len = SMB_Header(buffer, SMB_COM_NT_CREATE_ANDX, tree_id, user_id);
832
833 /* 0 */
834 buffer[len++] = 24; /* parameters */
835
836 buffer[len++] = 0xff; /* AndXCommand: secondary request */
837 buffer[len++] = 0x00; /* AndXReserved */
838 SMB_ADDWORD(&buffer[len],0); len += 2; /* AndXOffset */
839
840 buffer[len++] = 0; /* reserved */
841 slen = strlen(filename);
842 SMB_ADDWORD(&buffer[len],slen); len += 2; /* name length */
843
844 /* 0x08 */
845 SMB_ADDDWORD(&buffer[len],0); len += 4; /* flags */
846 SMB_ADDDWORD(&buffer[len],0); len += 4; /* root directory fid */
847 /* 0x10 */
848 SMB_ADDDWORD(&buffer[len],access); len += 4; /* access */
849 SMB_ADDDWORD(&buffer[len],0); len += 4; /* allocation size */
850 /* 0x18 */
851 SMB_ADDDWORD(&buffer[len],0); len += 4; /* root directory fid */
852
853 /* 0x1c */
854 SMB_ADDDWORD(&buffer[len],0); len += 4; /* initial allocation */
Vincent Béron9a624912002-05-31 23:06:46 +0000855 SMB_ADDDWORD(&buffer[len],0); len += 4;
856
Mike McCormackfc932612002-03-12 19:24:04 +0000857 /* 0x24 */
858 SMB_ADDDWORD(&buffer[len],attributes); len += 4; /* ExtFileAttributes*/
859
860 /* 0x28 */
861 SMB_ADDDWORD(&buffer[len],sharing); len += 4; /* ShareAccess */
Vincent Béron9a624912002-05-31 23:06:46 +0000862
Mike McCormackfc932612002-03-12 19:24:04 +0000863 /* 0x2c */
Mike McCormack963985b2002-07-19 03:17:19 +0000864 TRACE("creation = %08lx\n",creation);
Mike McCormackfc932612002-03-12 19:24:04 +0000865 SMB_ADDDWORD(&buffer[len],creation); len += 4; /* CreateDisposition */
Vincent Béron9a624912002-05-31 23:06:46 +0000866
Mike McCormackfc932612002-03-12 19:24:04 +0000867 /* 0x30 */
868 SMB_ADDDWORD(&buffer[len],creation); len += 4; /* CreateOptions */
Vincent Béron9a624912002-05-31 23:06:46 +0000869
Mike McCormackfc932612002-03-12 19:24:04 +0000870 /* 0x34 */
871 SMB_ADDDWORD(&buffer[len],0); len += 4; /* Impersonation */
872
873 /* 0x38 */
874 buffer[len++] = 0; /* security flags */
875
876 /* 0x39 */
877 SMB_ADDWORD(&buffer[len],slen); len += 2; /* size of buffer */
878
Francois Gouget95918362003-06-18 19:45:22 +0000879 if(slen<(sizeof(buffer)-len))
Mike McCormackfc932612002-03-12 19:24:04 +0000880 strcpy(&buffer[len], filename);
881 else
882 return FALSE;
883 len += slen+1;
Vincent Béron9a624912002-05-31 23:06:46 +0000884
Mike McCormackfc932612002-03-12 19:24:04 +0000885 /* name of the file */
886 buffer[len++] = 0;
887
888 if(!NB_Transaction(fd, buffer, len, &len))
889 return FALSE;
890
891 if(SMB_GetError(buffer))
892 return FALSE;
893
Mike McCormack963985b2002-07-19 03:17:19 +0000894 TRACE("OK\n");
Mike McCormackfc932612002-03-12 19:24:04 +0000895
896 /* FIXME */
897 /* *file_id = SMB_GETWORD(&buffer[xxx]); */
898 *file_id = 0;
899 return FALSE;
900
901 return TRUE;
902}
Alexandre Julliard958732d2002-07-02 02:09:39 +0000903#endif
Mike McCormackfc932612002-03-12 19:24:04 +0000904
905static USHORT SMB_GetMode(DWORD access, DWORD sharing)
906{
907 USHORT mode=0;
908
909 switch(access&(GENERIC_READ|GENERIC_WRITE))
910 {
911 case GENERIC_READ:
912 mode |= OF_READ;
913 break;
914 case GENERIC_WRITE:
915 mode |= OF_WRITE;
916 break;
917 case (GENERIC_READ|GENERIC_WRITE):
918 mode |= OF_READWRITE;
919 break;
920 }
921
922 switch(sharing&(FILE_SHARE_READ|FILE_SHARE_WRITE))
923 {
924 case (FILE_SHARE_READ|FILE_SHARE_WRITE):
925 mode |= OF_SHARE_DENY_NONE;
926 break;
927 case FILE_SHARE_READ:
928 mode |= OF_SHARE_DENY_WRITE;
929 break;
930 case FILE_SHARE_WRITE:
931 mode |= OF_SHARE_DENY_READ;
932 break;
933 default:
934 mode |= OF_SHARE_EXCLUSIVE;
935 break;
936 }
937
938 return mode;
939}
940
Alexandre Julliard958732d2002-07-02 02:09:39 +0000941#if 0 /* not yet */
Mike McCormackfc932612002-03-12 19:24:04 +0000942/* inverse of FILE_ConvertOFMode */
943static BOOL SMB_OpenAndX(int fd, USHORT tree_id, USHORT user_id, USHORT dialect,
944 LPCSTR filename, DWORD access, DWORD sharing,
945 DWORD creation, DWORD attributes, USHORT *file_id )
946{
947 unsigned char buffer[0x100];
948 int len = 0;
949 USHORT mode;
950
Mike McCormack963985b2002-07-19 03:17:19 +0000951 TRACE("%s\n",filename);
Mike McCormackfc932612002-03-12 19:24:04 +0000952
953 mode = SMB_GetMode(access,sharing);
954
Francois Gouget95918362003-06-18 19:45:22 +0000955 memset(buffer,0,sizeof(buffer));
Mike McCormackfc932612002-03-12 19:24:04 +0000956
957 len = SMB_Header(buffer, SMB_COM_OPEN_ANDX, tree_id, user_id);
958
959 /* 0 */
960 buffer[len++] = 15; /* parameters */
961 buffer[len++] = 0xff; /* AndXCommand: secondary request */
962 buffer[len++] = 0x00; /* AndXReserved */
963 SMB_ADDWORD(buffer+len,0); len+=2; /* AndXOffset */
964 SMB_ADDWORD(buffer+len,0); len+=2; /* Flags */
965 SMB_ADDWORD(buffer+len,mode); len+=2; /* desired access */
966 SMB_ADDWORD(buffer+len,0); len+=2; /* search attributes */
967 SMB_ADDWORD(buffer+len,0); len+=2;
968
969 /*FIXME: complete */
970 return FALSE;
971}
Alexandre Julliard958732d2002-07-02 02:09:39 +0000972#endif
Mike McCormackfc932612002-03-12 19:24:04 +0000973
Mike McCormack963985b2002-07-19 03:17:19 +0000974
Mike McCormackfc932612002-03-12 19:24:04 +0000975static BOOL SMB_Open(int fd, USHORT tree_id, USHORT user_id, USHORT dialect,
976 LPCSTR filename, DWORD access, DWORD sharing,
977 DWORD creation, DWORD attributes, USHORT *file_id )
978{
Mike McCormack963985b2002-07-19 03:17:19 +0000979 unsigned char buf[0x100];
980 int slen,pcount,i;
Mike McCormackfc932612002-03-12 19:24:04 +0000981 USHORT mode = SMB_GetMode(access,sharing);
Mike McCormack963985b2002-07-19 03:17:19 +0000982 struct NB_Buffer rx,tx;
Mike McCormackfc932612002-03-12 19:24:04 +0000983
Mike McCormack963985b2002-07-19 03:17:19 +0000984 TRACE("%s\n",filename);
Mike McCormackfc932612002-03-12 19:24:04 +0000985
Francois Gouget95918362003-06-18 19:45:22 +0000986 memset(buf,0,sizeof(buf));
Mike McCormackfc932612002-03-12 19:24:04 +0000987
Mike McCormack963985b2002-07-19 03:17:19 +0000988 tx.buffer = buf;
989 tx.len = SMB_Header(tx.buffer, SMB_COM_OPEN, tree_id, user_id);
Mike McCormackfc932612002-03-12 19:24:04 +0000990
991 /* 0 */
Mike McCormack963985b2002-07-19 03:17:19 +0000992 tx.buffer[tx.len++] = 2; /* parameters */
993 SMB_ADDWORD(tx.buffer+tx.len,mode); tx.len+=2;
994 SMB_ADDWORD(tx.buffer+tx.len,0); tx.len+=2; /* search attributes */
Mike McCormackfc932612002-03-12 19:24:04 +0000995
996 slen = strlen(filename)+2; /* inc. nul and BufferFormat */
Mike McCormack963985b2002-07-19 03:17:19 +0000997 SMB_ADDWORD(tx.buffer+tx.len,slen); tx.len+=2;
Mike McCormackfc932612002-03-12 19:24:04 +0000998
Mike McCormack963985b2002-07-19 03:17:19 +0000999 tx.buffer[tx.len] = 0x04; /* BufferFormat */
1000 strcpy(&tx.buffer[tx.len+1],filename);
1001 tx.len += slen;
Mike McCormackfc932612002-03-12 19:24:04 +00001002
Mike McCormack963985b2002-07-19 03:17:19 +00001003 rx.buffer = NULL;
1004 rx.len = 0;
1005 if(!NB_Transaction(fd, &tx, &rx))
Mike McCormackfc932612002-03-12 19:24:04 +00001006 return FALSE;
1007
Mike McCormack963985b2002-07-19 03:17:19 +00001008 if(!rx.buffer)
Mike McCormackfc932612002-03-12 19:24:04 +00001009 return FALSE;
1010
Mike McCormack963985b2002-07-19 03:17:19 +00001011 if(SMB_GetError(rx.buffer))
1012 return FALSE;
Mike McCormackfc932612002-03-12 19:24:04 +00001013
Mike McCormack963985b2002-07-19 03:17:19 +00001014 pcount = SMB_PARAM_COUNT(rx.buffer);
1015
1016 if( (SMB_HDRSIZE+pcount*2) > rx.len )
Mike McCormackfc932612002-03-12 19:24:04 +00001017 {
1018 ERR("Bad parameter count %d\n",pcount);
1019 return FALSE;
1020 }
1021
Mike McCormack963985b2002-07-19 03:17:19 +00001022 TRACE("response, %d args: ",pcount);
Mike McCormackfc932612002-03-12 19:24:04 +00001023 for(i=0; i<pcount; i++)
Tony Lambregts2e24a142003-03-15 00:12:42 +00001024 TRACE("%04x ",SMB_PARAM(rx.buffer,i));
1025 TRACE("\n");
Mike McCormackfc932612002-03-12 19:24:04 +00001026
Mike McCormack963985b2002-07-19 03:17:19 +00001027 *file_id = SMB_PARAM(rx.buffer,0);
Mike McCormackfc932612002-03-12 19:24:04 +00001028
Mike McCormack963985b2002-07-19 03:17:19 +00001029 TRACE("file_id = %04x\n",*file_id);
Mike McCormackfc932612002-03-12 19:24:04 +00001030
1031 return TRUE;
1032}
1033
Mike McCormackfc932612002-03-12 19:24:04 +00001034
Mike McCormack963985b2002-07-19 03:17:19 +00001035static BOOL SMB_Read(int fd, USHORT tree_id, USHORT user_id, USHORT dialect,
Francois Gougetbba4bb12002-09-17 01:35:09 +00001036 USHORT file_id, DWORD offset, LPVOID out, USHORT count, USHORT* read)
Mike McCormack963985b2002-07-19 03:17:19 +00001037{
1038 int buf_size,n,i;
1039 struct NB_Buffer rx,tx;
1040
1041 TRACE("user %04x tree %04x file %04x count %04x offset %08lx\n",
Mike McCormackfc932612002-03-12 19:24:04 +00001042 user_id, tree_id, file_id, count, offset);
1043
1044 buf_size = count+0x100;
Eric Pouechc962a692003-06-23 18:12:28 +00001045 tx.buffer = (unsigned char *) RtlAllocateHeap(ntdll_get_process_heap(),0,buf_size);
Mike McCormackfc932612002-03-12 19:24:04 +00001046
Mike McCormack963985b2002-07-19 03:17:19 +00001047 memset(tx.buffer,0,buf_size);
Mike McCormackfc932612002-03-12 19:24:04 +00001048
Mike McCormack963985b2002-07-19 03:17:19 +00001049 tx.len = SMB_Header(tx.buffer, SMB_COM_READ, tree_id, user_id);
Mike McCormackfc932612002-03-12 19:24:04 +00001050
Mike McCormack963985b2002-07-19 03:17:19 +00001051 tx.buffer[tx.len++] = 5;
1052 SMB_ADDWORD(&tx.buffer[tx.len],file_id); tx.len += 2;
1053 SMB_ADDWORD(&tx.buffer[tx.len],count); tx.len += 2;
1054 SMB_ADDDWORD(&tx.buffer[tx.len],offset); tx.len += 4;
1055 SMB_ADDWORD(&tx.buffer[tx.len],0); tx.len += 2; /* how many more bytes will be read */
Mike McCormackfc932612002-03-12 19:24:04 +00001056
Mike McCormack963985b2002-07-19 03:17:19 +00001057 tx.buffer[tx.len++] = 0;
Mike McCormackfc932612002-03-12 19:24:04 +00001058
Mike McCormack963985b2002-07-19 03:17:19 +00001059 rx.buffer = NULL;
1060 rx.len = 0;
1061 if(!NB_Transaction(fd, &tx, &rx))
Mike McCormackfc932612002-03-12 19:24:04 +00001062 {
Eric Pouechc962a692003-06-23 18:12:28 +00001063 RtlFreeHeap(ntdll_get_process_heap(),0,tx.buffer);
Mike McCormackfc932612002-03-12 19:24:04 +00001064 return FALSE;
1065 }
1066
Mike McCormack963985b2002-07-19 03:17:19 +00001067 if(SMB_GetError(rx.buffer))
Mike McCormackfc932612002-03-12 19:24:04 +00001068 {
Eric Pouechc962a692003-06-23 18:12:28 +00001069 RtlFreeHeap(ntdll_get_process_heap(),0,rx.buffer);
1070 RtlFreeHeap(ntdll_get_process_heap(),0,tx.buffer);
Mike McCormackfc932612002-03-12 19:24:04 +00001071 return FALSE;
1072 }
1073
Mike McCormack963985b2002-07-19 03:17:19 +00001074 n = SMB_PARAM_COUNT(rx.buffer);
Mike McCormackfc932612002-03-12 19:24:04 +00001075
Mike McCormack963985b2002-07-19 03:17:19 +00001076 if( (SMB_HDRSIZE+n*2) > rx.len )
Mike McCormackfc932612002-03-12 19:24:04 +00001077 {
Eric Pouechc962a692003-06-23 18:12:28 +00001078 RtlFreeHeap(ntdll_get_process_heap(),0,rx.buffer);
1079 RtlFreeHeap(ntdll_get_process_heap(),0,tx.buffer);
Mike McCormackfc932612002-03-12 19:24:04 +00001080 ERR("Bad parameter count %d\n",n);
1081 return FALSE;
1082 }
1083
Mike McCormack963985b2002-07-19 03:17:19 +00001084 TRACE("response, %d args: ",n);
Mike McCormackfc932612002-03-12 19:24:04 +00001085 for(i=0; i<n; i++)
Tony Lambregts2e24a142003-03-15 00:12:42 +00001086 TRACE("%04x ",SMB_PARAM(rx.buffer,i));
1087 TRACE("\n");
Mike McCormackfc932612002-03-12 19:24:04 +00001088
Mike McCormack963985b2002-07-19 03:17:19 +00001089 n = SMB_PARAM(rx.buffer,5) - 3;
Mike McCormackfc932612002-03-12 19:24:04 +00001090 if(n>count)
1091 n=count;
1092
Mike McCormack963985b2002-07-19 03:17:19 +00001093 memcpy( out, &SMB_BUFFER(rx.buffer,3), n);
Mike McCormackfc932612002-03-12 19:24:04 +00001094
Mike McCormack963985b2002-07-19 03:17:19 +00001095 TRACE("Read %d bytes\n",n);
Mike McCormackfc932612002-03-12 19:24:04 +00001096 *read = n;
1097
Eric Pouechc962a692003-06-23 18:12:28 +00001098 RtlFreeHeap(ntdll_get_process_heap(),0,tx.buffer);
1099 RtlFreeHeap(ntdll_get_process_heap(),0,rx.buffer);
Vincent Béron9a624912002-05-31 23:06:46 +00001100
Mike McCormackfc932612002-03-12 19:24:04 +00001101 return TRUE;
1102}
1103
Mike McCormack963985b2002-07-19 03:17:19 +00001104
1105/*
1106 * setup_count : number of USHORTs in the setup string
1107 */
1108struct SMB_Trans2Info
1109{
1110 struct NB_Buffer buf;
1111 unsigned char *setup;
1112 int setup_count;
1113 unsigned char *params;
1114 int param_count;
1115 unsigned char *data;
1116 int data_count;
1117};
1118
1119/*
1120 * Do an SMB transaction
1121 *
1122 * This function allocates memory in the recv structure. It is
1123 * the caller's responsibility to free the memory if it finds
1124 * that recv->buf.buffer is nonzero.
1125 */
1126static BOOL SMB_Transaction2(int fd, int tree_id, int user_id,
1127 struct SMB_Trans2Info *send,
1128 struct SMB_Trans2Info *recv)
1129{
1130 int buf_size;
1131 const int retmaxparams = 0xf000;
1132 const int retmaxdata = 1024;
1133 const int retmaxsetup = 0; /* FIXME */
1134 const int flags = 0;
1135 const int timeout = 0;
1136 int param_ofs, data_ofs;
1137 struct NB_Buffer tx;
1138 BOOL ret = FALSE;
1139
1140 buf_size = 0x100 + send->setup_count*2 + send->param_count + send->data_count ;
Eric Pouechc962a692003-06-23 18:12:28 +00001141 tx.buffer = (unsigned char *) RtlAllocateHeap(ntdll_get_process_heap(),0,buf_size);
Mike McCormack963985b2002-07-19 03:17:19 +00001142
1143 tx.len = SMB_Header(tx.buffer, SMB_COM_TRANSACTION2, tree_id, user_id);
1144
1145 tx.buffer[tx.len++] = 14 + send->setup_count;
1146 SMB_ADDWORD(&tx.buffer[tx.len],send->param_count); /* total param bytes sent */
1147 tx.len += 2;
1148 SMB_ADDWORD(&tx.buffer[tx.len],send->data_count); /* total data bytes sent */
1149 tx.len += 2;
1150 SMB_ADDWORD(&tx.buffer[tx.len],retmaxparams); /*max parameter bytes to return */
1151 tx.len += 2;
1152 SMB_ADDWORD(&tx.buffer[tx.len],retmaxdata); /* max data bytes to return */
1153 tx.len += 2;
1154 tx.buffer[tx.len++] = retmaxsetup;
1155 tx.buffer[tx.len++] = 0; /* reserved1 */
1156
1157 SMB_ADDWORD(&tx.buffer[tx.len],flags); /* flags */
1158 tx.len += 2;
1159 SMB_ADDDWORD(&tx.buffer[tx.len],timeout); /* timeout */
1160 tx.len += 4;
1161 SMB_ADDWORD(&tx.buffer[tx.len],0); /* reserved2 */
1162 tx.len += 2;
1163 SMB_ADDWORD(&tx.buffer[tx.len],send->param_count); /* parameter count - this buffer */
1164 tx.len += 2;
1165
1166 param_ofs = tx.len; /* parameter offset */
1167 tx.len += 2;
1168 SMB_ADDWORD(&tx.buffer[tx.len],send->data_count); /* data count */
1169 tx.len += 2;
1170
1171 data_ofs = tx.len; /* data offset */
1172 tx.len += 2;
1173 tx.buffer[tx.len++] = send->setup_count; /* setup count */
1174 tx.buffer[tx.len++] = 0; /* reserved3 */
1175
1176 memcpy(&tx.buffer[tx.len], send->setup, send->setup_count*2); /* setup */
1177 tx.len += send->setup_count*2;
1178
1179 /* add string here when implementing SMB_COM_TRANS */
1180
1181 SMB_ADDWORD(&tx.buffer[param_ofs], tx.len);
1182 memcpy(&tx.buffer[tx.len], send->params, send->param_count); /* parameters */
1183 tx.len += send->param_count;
1184 if(tx.len%2)
1185 tx.len ++; /* pad2 */
1186
1187 SMB_ADDWORD(&tx.buffer[data_ofs], tx.len);
1188 if(send->data_count && send->data)
1189 {
1190 memcpy(&tx.buffer[tx.len], send->data, send->data_count); /* data */
1191 tx.len += send->data_count;
1192 }
1193
1194 recv->buf.buffer = NULL;
1195 recv->buf.len = 0;
1196 if(!NB_Transaction(fd, &tx, &recv->buf))
1197 goto done;
1198
1199 if(!recv->buf.buffer)
1200 goto done;
1201
1202 if(SMB_GetError(recv->buf.buffer))
1203 goto done;
1204
1205 /* reuse these two offsets to check the received message */
1206 param_ofs = SMB_PARAM(recv->buf.buffer,4);
1207 data_ofs = SMB_PARAM(recv->buf.buffer,7);
1208
1209 if( (recv->param_count + param_ofs) > recv->buf.len )
1210 goto done;
1211
1212 if( (recv->data_count + data_ofs) > recv->buf.len )
1213 goto done;
1214
1215 TRACE("Success\n");
1216
1217 recv->setup = NULL;
1218 recv->setup_count = 0;
1219
1220 recv->param_count = SMB_PARAM(recv->buf.buffer,0);
1221 recv->params = &recv->buf.buffer[param_ofs];
1222
1223 recv->data_count = SMB_PARAM(recv->buf.buffer,6);
1224 recv->data = &recv->buf.buffer[data_ofs];
1225
1226 /*
1227 TRACE("%d words\n",SMB_PARAM_COUNT(recv->buf.buffer));
1228 TRACE("total parameters = %d\n",SMB_PARAM(recv->buf.buffer,0));
1229 TRACE("total data = %d\n",SMB_PARAM(recv->buf.buffer,1));
1230 TRACE("parameters = %d\n",SMB_PARAM(recv->buf.buffer,3));
1231 TRACE("parameter offset = %d\n",SMB_PARAM(recv->buf.buffer,4));
1232 TRACE("param displace = %d\n",SMB_PARAM(recv->buf.buffer,5));
1233
1234 TRACE("data count = %d\n",SMB_PARAM(recv->buf.buffer,6));
1235 TRACE("data offset = %d\n",SMB_PARAM(recv->buf.buffer,7));
1236 TRACE("data displace = %d\n",SMB_PARAM(recv->buf.buffer,8));
1237 */
1238
1239 ret = TRUE;
1240
1241done:
1242 if(tx.buffer)
Eric Pouechc962a692003-06-23 18:12:28 +00001243 RtlFreeHeap(ntdll_get_process_heap(),0,tx.buffer);
Mike McCormack963985b2002-07-19 03:17:19 +00001244
1245 return ret;
1246}
1247
1248static BOOL SMB_SetupFindFirst(struct SMB_Trans2Info *send, LPSTR filename)
1249{
1250 int search_attribs = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_SYSTEM;
1251 int search_count = 10;
1252 int flags = 0;
1253 int infolevel = 0x104; /* SMB_FILE_BOTH_DIRECTORY_INFO */
1254 int storagetype = 0;
1255 int len, buf_size;
1256
Francois Gouget95918362003-06-18 19:45:22 +00001257 memset(send,0,sizeof(send));
Mike McCormack963985b2002-07-19 03:17:19 +00001258
1259 send->setup_count = 1;
Eric Pouechc962a692003-06-23 18:12:28 +00001260 send->setup = RtlAllocateHeap(ntdll_get_process_heap(),0,send->setup_count*2);
Mike McCormack963985b2002-07-19 03:17:19 +00001261 if(!send->setup)
1262 return FALSE;
1263
Eric Pouechc962a692003-06-23 18:12:28 +00001264 buf_size = 0x10 + strlen(filename);
1265 send->params = RtlAllocateHeap(ntdll_get_process_heap(),0,buf_size);
Mike McCormack963985b2002-07-19 03:17:19 +00001266 if(!send->params)
1267 {
Eric Pouechc962a692003-06-23 18:12:28 +00001268 RtlFreeHeap(ntdll_get_process_heap(),0,send->setup);
Mike McCormack963985b2002-07-19 03:17:19 +00001269 return FALSE;
1270 }
1271
1272 SMB_ADDWORD(send->setup,TRANS2_FIND_FIRST2);
1273
1274 len = 0;
1275 memset(send->params,0,buf_size);
1276 SMB_ADDWORD(&send->params[len],search_attribs); len += 2;
1277 SMB_ADDWORD(&send->params[len],search_count); len += 2;
1278 SMB_ADDWORD(&send->params[len],flags); len += 2;
1279 SMB_ADDWORD(&send->params[len],infolevel); len += 2;
1280 SMB_ADDDWORD(&send->params[len],storagetype); len += 4;
1281
1282 strcpy(&send->params[len],filename);
Eric Pouechc962a692003-06-23 18:12:28 +00001283 len += strlen(filename)+1;
Mike McCormack963985b2002-07-19 03:17:19 +00001284
1285 send->param_count = len;
1286 send->data = NULL;
1287 send->data_count = 0;
1288
1289 return TRUE;
1290}
1291
1292static SMB_DIR *SMB_Trans2FindFirst(int fd, USHORT tree_id,
1293 USHORT user_id, USHORT dialect, LPSTR filename )
1294{
1295 int num;
1296 BOOL ret;
1297 /* char *filename = "\\*"; */
1298 struct SMB_Trans2Info send, recv;
1299 SMB_DIR *smbdir = NULL;
1300
Bill Medlandaf81a022003-06-24 19:23:25 +00001301 TRACE("pattern = %s\n",filename);
Mike McCormack963985b2002-07-19 03:17:19 +00001302
1303 if(!SMB_SetupFindFirst(&send, filename))
1304 return FALSE;
1305
Francois Gouget95918362003-06-18 19:45:22 +00001306 memset(&recv,0,sizeof(recv));
Mike McCormack963985b2002-07-19 03:17:19 +00001307
1308 ret = SMB_Transaction2(fd, tree_id, user_id, &send, &recv);
Eric Pouechc962a692003-06-23 18:12:28 +00001309 RtlFreeHeap(ntdll_get_process_heap(),0,send.params);
1310 RtlFreeHeap(ntdll_get_process_heap(),0,send.setup);
Mike McCormack963985b2002-07-19 03:17:19 +00001311
1312 if(!ret)
1313 goto done;
1314
1315 if(recv.setup_count)
1316 goto done;
1317
1318 if(recv.param_count != 10)
1319 goto done;
1320
1321 num = SMB_GETWORD(&recv.params[2]);
1322 TRACE("Success, search id: %d\n",num);
1323
1324 if(SMB_GETWORD(&recv.params[4]))
1325 FIXME("need to read more!\n");
1326
Eric Pouechc962a692003-06-23 18:12:28 +00001327 smbdir = RtlAllocateHeap(ntdll_get_process_heap(),0,sizeof(*smbdir));
Mike McCormack963985b2002-07-19 03:17:19 +00001328 if(smbdir)
1329 {
1330 int i, ofs=0;
1331
1332 smbdir->current = 0;
1333 smbdir->num_entries = num;
Eric Pouechc962a692003-06-23 18:12:28 +00001334 smbdir->entries = RtlAllocateHeap(ntdll_get_process_heap(), 0, sizeof(unsigned char*)*num);
Mike McCormack963985b2002-07-19 03:17:19 +00001335 if(!smbdir->entries)
1336 goto done;
1337 smbdir->buffer = recv.buf.buffer; /* save to free later */
1338
1339 for(i=0; i<num; i++)
1340 {
1341 int size = SMB_GETDWORD(&recv.data[ofs]);
1342
1343 smbdir->entries[i] = &recv.data[ofs];
1344
1345 if(TRACE_ON(file))
1346 {
1347 int j;
1348 for(j=0; j<size; j++)
1349 DPRINTF("%02x%c",recv.data[ofs+j],((j+1)%16)?' ':'\n');
1350 }
1351 TRACE("file %d : %s\n", i, &recv.data[ofs+0x5e]);
1352 ofs += size;
1353 if(ofs>recv.data_count)
1354 goto done;
1355 }
1356
1357 ret = TRUE;
1358 }
1359
1360done:
1361 if(!ret)
1362 {
1363 if( recv.buf.buffer )
Eric Pouechc962a692003-06-23 18:12:28 +00001364 RtlFreeHeap(ntdll_get_process_heap(),0,recv.buf.buffer);
Mike McCormack963985b2002-07-19 03:17:19 +00001365 if( smbdir )
1366 {
1367 if( smbdir->entries )
Eric Pouechc962a692003-06-23 18:12:28 +00001368 RtlFreeHeap(ntdll_get_process_heap(),0,smbdir->entries);
1369 RtlFreeHeap(ntdll_get_process_heap(),0,smbdir);
Mike McCormack963985b2002-07-19 03:17:19 +00001370 }
1371 smbdir = NULL;
1372 }
1373
1374 return smbdir;
1375}
1376
Mike McCormack9414adf2002-05-05 20:29:15 +00001377static int SMB_GetSocket(LPCSTR host)
Mike McCormackfc932612002-03-12 19:24:04 +00001378{
1379 int fd=-1,r;
1380 struct sockaddr_in sin;
Mike McCormack9414adf2002-05-05 20:29:15 +00001381 struct hostent *he;
1382
Mike McCormack963985b2002-07-19 03:17:19 +00001383 TRACE("host %s\n",host);
Mike McCormack9414adf2002-05-05 20:29:15 +00001384
1385 he = gethostbyname(host);
1386 if(he)
1387 {
1388 memcpy(&sin.sin_addr,he->h_addr, sizeof (sin.sin_addr));
1389 goto connect;
1390 }
1391
Mike McCormack963985b2002-07-19 03:17:19 +00001392 if(NB_Lookup(host,&sin))
1393 goto connect;
1394
Mike McCormack9414adf2002-05-05 20:29:15 +00001395 /* FIXME: resolve by WINS too */
1396
1397 ERR("couldn't resolve SMB host %s\n", host);
1398
1399 return -1;
1400
1401connect:
1402 sin.sin_family = AF_INET;
1403 sin.sin_port = htons(139); /* netbios session */
1404
1405 fd = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
1406 if(fd<0)
1407 return fd;
1408
1409 {
1410 unsigned char *x = (unsigned char *)&sin.sin_addr;
Mike McCormack963985b2002-07-19 03:17:19 +00001411 TRACE("Connecting to %d.%d.%d.%d ...\n", x[0],x[1],x[2],x[3]);
Mike McCormack9414adf2002-05-05 20:29:15 +00001412 }
Francois Gouget95918362003-06-18 19:45:22 +00001413 r = connect(fd, (struct sockaddr*)&sin, sizeof(sin));
Mike McCormack9414adf2002-05-05 20:29:15 +00001414
1415 if(!NB_SessionReq(fd, "*SMBSERVER", "WINE"))
1416 {
1417 close(fd);
1418 return -1;
1419 }
1420
1421 return fd;
1422}
1423
1424static BOOL SMB_LoginAndConnect(int fd, LPCSTR host, LPCSTR share, USHORT *tree_id, USHORT *user_id, USHORT *dialect)
1425{
Mike McCormackfc932612002-03-12 19:24:04 +00001426 LPSTR name=NULL;
1427
Mike McCormack963985b2002-07-19 03:17:19 +00001428 TRACE("host %s share %s\n",host,share);
Mike McCormackfc932612002-03-12 19:24:04 +00001429
Mike McCormackfc932612002-03-12 19:24:04 +00001430 if(!SMB_NegotiateProtocol(fd, dialect))
Mike McCormack9414adf2002-05-05 20:29:15 +00001431 return FALSE;
Mike McCormackfc932612002-03-12 19:24:04 +00001432
1433 if(!SMB_SessionSetup(fd, user_id))
Mike McCormack9414adf2002-05-05 20:29:15 +00001434 return FALSE;
Mike McCormackfc932612002-03-12 19:24:04 +00001435
Eric Pouechc962a692003-06-23 18:12:28 +00001436 name = RtlAllocateHeap(ntdll_get_process_heap(),0,strlen(host)+strlen(share)+5);
Mike McCormackfc932612002-03-12 19:24:04 +00001437 if(!name)
Mike McCormack9414adf2002-05-05 20:29:15 +00001438 return FALSE;
1439
Mike McCormackfc932612002-03-12 19:24:04 +00001440 sprintf(name,"\\\\%s\\%s",host,share);
1441 if(!SMB_TreeConnect(fd,*user_id,name,tree_id))
Mike McCormack9414adf2002-05-05 20:29:15 +00001442 {
Eric Pouechc962a692003-06-23 18:12:28 +00001443 RtlFreeHeap(ntdll_get_process_heap(),0,name);
Mike McCormack9414adf2002-05-05 20:29:15 +00001444 return FALSE;
1445 }
1446
1447 return TRUE;
Mike McCormackfc932612002-03-12 19:24:04 +00001448}
1449
1450static HANDLE SMB_RegisterFile( int fd, USHORT tree_id, USHORT user_id, USHORT dialect, USHORT file_id)
1451{
1452 int r;
1453 HANDLE ret;
1454
1455 wine_server_send_fd( fd );
1456
1457 SERVER_START_REQ( create_smb )
1458 {
1459 req->tree_id = tree_id;
1460 req->user_id = user_id;
1461 req->file_id = file_id;
1462 req->dialect = 0;
1463 req->fd = fd;
1464 SetLastError(0);
1465 r = wine_server_call_err( req );
1466 ret = reply->handle;
1467 }
1468 SERVER_END_REQ;
1469
1470 if(!r)
Andrew John Hughesed800c62002-11-21 03:45:01 +00001471 TRACE("created wineserver smb object, handle = %p\n",ret);
Mike McCormackfc932612002-03-12 19:24:04 +00001472 else
1473 SetLastError( ERROR_PATH_NOT_FOUND );
1474
1475 return ret;
1476}
1477
Dmitry Timoshkovd75aed22002-08-27 01:13:58 +00001478HANDLE WINAPI SMB_CreateFileW( LPCWSTR uncname, DWORD access, DWORD sharing,
Mike McCormackfc932612002-03-12 19:24:04 +00001479 LPSECURITY_ATTRIBUTES sa, DWORD creation,
1480 DWORD attributes, HANDLE template )
1481{
1482 int fd;
1483 USHORT tree_id=0, user_id=0, dialect=0, file_id=0;
1484 LPSTR name,host,share,file;
Mike McCormack9414adf2002-05-05 20:29:15 +00001485 HANDLE handle = INVALID_HANDLE_VALUE;
Dmitry Timoshkovd75aed22002-08-27 01:13:58 +00001486 INT len;
Mike McCormackfc932612002-03-12 19:24:04 +00001487
Dmitry Timoshkovd75aed22002-08-27 01:13:58 +00001488 len = WideCharToMultiByte(CP_ACP, 0, uncname, -1, NULL, 0, NULL, NULL);
Eric Pouechc962a692003-06-23 18:12:28 +00001489 name = RtlAllocateHeap(ntdll_get_process_heap(), 0, len);
Mike McCormackfc932612002-03-12 19:24:04 +00001490 if(!name)
Mike McCormack9414adf2002-05-05 20:29:15 +00001491 return handle;
Mike McCormackfc932612002-03-12 19:24:04 +00001492
Dmitry Timoshkovd75aed22002-08-27 01:13:58 +00001493 WideCharToMultiByte(CP_ACP, 0, uncname, -1, name, len, NULL, NULL);
Mike McCormackfc932612002-03-12 19:24:04 +00001494
1495 if( !UNC_SplitName(name, &host, &share, &file) )
1496 {
Eric Pouechc962a692003-06-23 18:12:28 +00001497 RtlFreeHeap(ntdll_get_process_heap(),0,name);
Mike McCormackfc932612002-03-12 19:24:04 +00001498 return handle;
1499 }
1500
Mike McCormack963985b2002-07-19 03:17:19 +00001501 TRACE("server is %s, share is %s, file is %s\n", host, share, file);
Mike McCormack9414adf2002-05-05 20:29:15 +00001502
1503 fd = SMB_GetSocket(host);
Mike McCormackfc932612002-03-12 19:24:04 +00001504 if(fd < 0)
Mike McCormack9414adf2002-05-05 20:29:15 +00001505 goto done;
1506
1507 if(!SMB_LoginAndConnect(fd, host, share, &tree_id, &user_id, &dialect))
1508 goto done;
Mike McCormackfc932612002-03-12 19:24:04 +00001509
1510#if 0
Vincent Béron9a624912002-05-31 23:06:46 +00001511 if(!SMB_NtCreateOpen(fd, tree_id, user_id, dialect, file,
Mike McCormackfc932612002-03-12 19:24:04 +00001512 access, sharing, sa, creation, attributes, template, &file_id ))
1513 {
1514 close(fd);
Mike McCormackfc932612002-03-12 19:24:04 +00001515 ERR("CreateOpen failed\n");
Mike McCormack9414adf2002-05-05 20:29:15 +00001516 goto done;
Mike McCormackfc932612002-03-12 19:24:04 +00001517 }
1518#endif
Vincent Béron9a624912002-05-31 23:06:46 +00001519 if(!SMB_Open(fd, tree_id, user_id, dialect, file,
Mike McCormackfc932612002-03-12 19:24:04 +00001520 access, sharing, creation, attributes, &file_id ))
1521 {
1522 close(fd);
Mike McCormackfc932612002-03-12 19:24:04 +00001523 ERR("CreateOpen failed\n");
Mike McCormack9414adf2002-05-05 20:29:15 +00001524 goto done;
Mike McCormackfc932612002-03-12 19:24:04 +00001525 }
1526
Mike McCormackfc932612002-03-12 19:24:04 +00001527 handle = SMB_RegisterFile(fd, tree_id, user_id, dialect, file_id);
1528 if(!handle)
1529 {
1530 ERR("register failed\n");
1531 close(fd);
1532 }
Vincent Béron9a624912002-05-31 23:06:46 +00001533
Mike McCormack9414adf2002-05-05 20:29:15 +00001534done:
Eric Pouechc962a692003-06-23 18:12:28 +00001535 RtlFreeHeap(ntdll_get_process_heap(),0,name);
Mike McCormackfc932612002-03-12 19:24:04 +00001536 return handle;
1537}
1538
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001539static NTSTATUS SMB_GetSmbInfo(HANDLE hFile, USHORT *tree_id, USHORT *user_id, USHORT *dialect, USHORT *file_id, LPDWORD offset)
Mike McCormackfc932612002-03-12 19:24:04 +00001540{
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001541 NTSTATUS status;
Mike McCormackfc932612002-03-12 19:24:04 +00001542
1543 SERVER_START_REQ( get_smb_info )
1544 {
1545 req->handle = hFile;
1546 req->flags = 0;
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001547 status = wine_server_call( req );
Mike McCormackfc932612002-03-12 19:24:04 +00001548 if(tree_id)
1549 *tree_id = reply->tree_id;
1550 if(user_id)
1551 *user_id = reply->user_id;
1552 if(file_id)
1553 *file_id = reply->file_id;
1554 if(dialect)
1555 *dialect = reply->dialect;
1556 if(offset)
1557 *offset = reply->offset;
1558 }
1559 SERVER_END_REQ;
1560
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001561 return status;
Mike McCormackfc932612002-03-12 19:24:04 +00001562}
1563
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001564static NTSTATUS SMB_SetOffset(HANDLE hFile, DWORD offset)
Mike McCormackfc932612002-03-12 19:24:04 +00001565{
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001566 NTSTATUS status;
Mike McCormackfc932612002-03-12 19:24:04 +00001567
Mike McCormack963985b2002-07-19 03:17:19 +00001568 TRACE("offset = %08lx\n",offset);
Mike McCormackfc932612002-03-12 19:24:04 +00001569
1570 SERVER_START_REQ( get_smb_info )
1571 {
1572 req->handle = hFile;
1573 req->flags = SMBINFO_SET_OFFSET;
1574 req->offset = offset;
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001575 status = wine_server_call( req );
Mike McCormackfc932612002-03-12 19:24:04 +00001576 /* if(offset)
1577 *offset = reply->offset; */
1578 }
1579 SERVER_END_REQ;
1580
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001581 return status;
Mike McCormackfc932612002-03-12 19:24:04 +00001582}
1583
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001584NTSTATUS WINAPI SMB_ReadFile(HANDLE hFile, LPVOID buffer, DWORD bytesToRead,
1585 PIO_STATUS_BLOCK io_status)
Mike McCormackfc932612002-03-12 19:24:04 +00001586{
1587 int fd;
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001588 DWORD count, offset;
Mike McCormackfc932612002-03-12 19:24:04 +00001589 USHORT user_id, tree_id, dialect, file_id, read;
Mike McCormackfc932612002-03-12 19:24:04 +00001590
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001591 TRACE("%p %p %ld %p\n", hFile, buffer, bytesToRead, io_status);
Mike McCormackfc932612002-03-12 19:24:04 +00001592
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001593 io_status->Information = 0;
1594
1595 io_status->u.Status = SMB_GetSmbInfo(hFile, &tree_id, &user_id, &dialect, &file_id, &offset);
1596 if (io_status->u.Status) return io_status->u.Status;
Mike McCormackfc932612002-03-12 19:24:04 +00001597
1598 fd = FILE_GetUnixHandle(hFile, GENERIC_READ);
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001599 if (fd<0) return io_status->u.Status = STATUS_INVALID_HANDLE;
Mike McCormackfc932612002-03-12 19:24:04 +00001600
Mike McCormackfc932612002-03-12 19:24:04 +00001601 while(1)
1602 {
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001603 count = bytesToRead - io_status->Information;
Mike McCormackfc932612002-03-12 19:24:04 +00001604 if(count>0x400)
1605 count = 0x400;
1606 if(count==0)
1607 break;
1608 read = 0;
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001609 if (!SMB_Read(fd, tree_id, user_id, dialect, file_id, offset, buffer, count, &read))
Mike McCormackfc932612002-03-12 19:24:04 +00001610 break;
1611 if(!read)
1612 break;
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001613 io_status->Information += read;
Gregg Mattinson7c4cb512002-07-03 21:10:43 +00001614 buffer = (char*)buffer + read;
Mike McCormackfc932612002-03-12 19:24:04 +00001615 offset += read;
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001616 if(io_status->Information >= bytesToRead)
Mike McCormackfc932612002-03-12 19:24:04 +00001617 break;
1618 }
1619 close(fd);
1620
Eric Pouech9bd4f6b2003-06-26 02:08:17 +00001621 return io_status->u.Status = SMB_SetOffset(hFile, offset);
Mike McCormackfc932612002-03-12 19:24:04 +00001622}
Mike McCormack963985b2002-07-19 03:17:19 +00001623
Dmitry Timoshkovd75aed22002-08-27 01:13:58 +00001624SMB_DIR* WINAPI SMB_FindFirst(LPCWSTR name)
Mike McCormack963985b2002-07-19 03:17:19 +00001625{
1626 int fd = -1;
1627 LPSTR host,share,file;
1628 USHORT tree_id=0, user_id=0, dialect=0;
1629 SMB_DIR *ret = NULL;
1630 LPSTR filename;
Dmitry Timoshkovd75aed22002-08-27 01:13:58 +00001631 DWORD len;
Mike McCormack963985b2002-07-19 03:17:19 +00001632
Dmitry Timoshkovd75aed22002-08-27 01:13:58 +00001633 TRACE("Find %s\n",debugstr_w(name));
Mike McCormack963985b2002-07-19 03:17:19 +00001634
Dmitry Timoshkovd75aed22002-08-27 01:13:58 +00001635 len = WideCharToMultiByte( CP_ACP, 0, name, -1, NULL, 0, NULL, NULL );
Eric Pouechc962a692003-06-23 18:12:28 +00001636 filename = RtlAllocateHeap(ntdll_get_process_heap(),0,len);
Mike McCormack963985b2002-07-19 03:17:19 +00001637 if(!filename)
1638 return ret;
Dmitry Timoshkovd75aed22002-08-27 01:13:58 +00001639 WideCharToMultiByte( CP_ACP, 0, name, -1, filename, len, NULL, NULL );
Mike McCormack963985b2002-07-19 03:17:19 +00001640
1641 if( !UNC_SplitName(filename, &host, &share, &file) )
1642 goto done;
1643
1644 fd = SMB_GetSocket(host);
1645 if(fd < 0)
1646 goto done;
1647
1648 if(!SMB_LoginAndConnect(fd, host, share, &tree_id, &user_id, &dialect))
1649 goto done;
1650
1651 TRACE("server is %s, share is %s, file is %s\n", host, share, file);
1652
1653 ret = SMB_Trans2FindFirst(fd, tree_id, user_id, dialect, file);
1654
1655done:
1656 /* disconnect */
1657 if(fd != -1)
1658 close(fd);
1659
1660 if(filename)
Eric Pouechc962a692003-06-23 18:12:28 +00001661 RtlFreeHeap(ntdll_get_process_heap(),0,filename);
Mike McCormack963985b2002-07-19 03:17:19 +00001662
1663 return ret;
1664}
1665
1666
Dmitry Timoshkovd75aed22002-08-27 01:13:58 +00001667BOOL WINAPI SMB_FindNext(SMB_DIR *dir, WIN32_FIND_DATAW *data )
Mike McCormack963985b2002-07-19 03:17:19 +00001668{
1669 unsigned char *ent;
1670 int len, fnlen;
1671
1672 TRACE("%d of %d\n",dir->current,dir->num_entries);
1673
1674 if(dir->current >= dir->num_entries)
1675 return FALSE;
1676
Francois Gouget95918362003-06-18 19:45:22 +00001677 memset(data, 0, sizeof(*data));
Mike McCormack963985b2002-07-19 03:17:19 +00001678
1679 ent = dir->entries[dir->current];
1680 len = SMB_GETDWORD(&ent[0]);
1681 if(len<0x5e)
1682 return FALSE;
1683
1684 memcpy(&data->ftCreationTime, &ent[8], 8);
1685 memcpy(&data->ftLastAccessTime, &ent[0x10], 8);
1686 memcpy(&data->ftLastWriteTime, &ent[0x18], 8);
1687 data->nFileSizeHigh = SMB_GETDWORD(&ent[0x30]);
1688 data->nFileSizeLow = SMB_GETDWORD(&ent[0x34]);
1689 data->dwFileAttributes = SMB_GETDWORD(&ent[0x38]);
1690
1691 /* copy the long filename */
1692 fnlen = SMB_GETDWORD(&ent[0x3c]);
Francois Gouget95918362003-06-18 19:45:22 +00001693 if ( fnlen > (sizeof(data->cFileName)/sizeof(WCHAR)) )
Mike McCormack963985b2002-07-19 03:17:19 +00001694 return FALSE;
Dmitry Timoshkovd75aed22002-08-27 01:13:58 +00001695 MultiByteToWideChar( CP_ACP, 0, &ent[0x5e], fnlen, data->cFileName,
1696 sizeof(data->cFileName)/sizeof(WCHAR) );
Mike McCormack963985b2002-07-19 03:17:19 +00001697
1698 /* copy the short filename */
Francois Gouget95918362003-06-18 19:45:22 +00001699 if ( ent[0x44] > (sizeof(data->cAlternateFileName)/sizeof(WCHAR)) )
Mike McCormack963985b2002-07-19 03:17:19 +00001700 return FALSE;
Dmitry Timoshkovd75aed22002-08-27 01:13:58 +00001701 MultiByteToWideChar( CP_ACP, 0, &ent[0x5e + len], ent[0x44], data->cAlternateFileName,
1702 sizeof(data->cAlternateFileName)/sizeof(WCHAR) );
Mike McCormack963985b2002-07-19 03:17:19 +00001703
1704 dir->current++;
1705
1706 return TRUE;
1707}
1708
1709BOOL WINAPI SMB_CloseDir(SMB_DIR *dir)
1710{
Eric Pouechc962a692003-06-23 18:12:28 +00001711 RtlFreeHeap(ntdll_get_process_heap(),0,dir->buffer);
1712 RtlFreeHeap(ntdll_get_process_heap(),0,dir->entries);
Francois Gouget95918362003-06-18 19:45:22 +00001713 memset(dir,0,sizeof(*dir));
Eric Pouechc962a692003-06-23 18:12:28 +00001714 RtlFreeHeap(ntdll_get_process_heap(),0,dir);
Mike McCormack963985b2002-07-19 03:17:19 +00001715 return TRUE;
1716}