2 * Copyright (C) 2002 Mike McCormack
4 * CIFS implementation for WINE
6 * This is a WINE's implementation of the Common Internet File System
8 * for specification see:
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
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.
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.
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
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
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.
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.
47 * - We don't deal with SMB dialects at all.
49 * - SMB supports passing unicode over the wire, should use this if possible.
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.
57 #include "wine/port.h"
65 #include <sys/types.h>
67 #ifdef HAVE_SYS_MMAN_H
70 #ifdef HAVE_SYS_TIME_H
71 # include <sys/time.h>
73 #ifdef HAVE_SYS_POLL_H
74 # include <sys/poll.h>
83 #ifdef HAVE_SYS_SOCKET_H
84 # include <sys/socket.h>
86 #include <sys/types.h>
87 #ifdef HAVE_NETINET_IN_SYSTM_H
88 #include <netinet/in_systm.h>
90 #ifdef HAVE_NETINET_IN_H
91 #include <netinet/in.h>
93 #ifdef HAVE_NETINET_IP_H
94 #include <netinet/ip.h>
96 #ifdef HAVE_ARPA_INET_H
97 #include <arpa/inet.h>
103 #include "winerror.h"
111 #include "wine/server.h"
112 #include "wine/debug.h"
114 WINE_DEFAULT_DEBUG_CHANNEL(file
);
116 #define MAX_HOST_NAME 15
117 #define NB_TIMEOUT 10000
119 USHORT SMB_MultiplexId
= 0;
123 unsigned char *buffer
;
127 static int netbios_name(const char *p
, unsigned char *buffer
)
133 for(i
=0; i
<=MAX_HOST_NAME
; i
++)
138 ch
= *p
++&0xdf; /* add character from hostname */
140 ch
= ' '; /* add padding */
143 ch
= 0; /* add terminator */
144 buffer
[len
++] = ((ch
&0xf0) >> 4) + 'A';
145 buffer
[len
++] = (ch
&0x0f) + 'A';
147 buffer
[len
++] = 0; /* add second terminator */
151 static DWORD
NB_NameReq(LPCSTR host
, unsigned char *buffer
, int len
)
155 NBR_ADDWORD(&buffer
[i
],trn
); i
+=2;
156 NBR_ADDWORD(&buffer
[i
],0x0110); i
+=2;
157 NBR_ADDWORD(&buffer
[i
],0x0001); i
+=2;
158 NBR_ADDWORD(&buffer
[i
],0x0000); i
+=2;
159 NBR_ADDWORD(&buffer
[i
],0x0000); i
+=2;
160 NBR_ADDWORD(&buffer
[i
],0x0000); i
+=2;
162 i
+= netbios_name(host
,&buffer
[i
]);
164 NBR_ADDWORD(&buffer
[i
],0x0020); i
+=2;
165 NBR_ADDWORD(&buffer
[i
],0x0001); i
+=2;
167 TRACE("packet is %d bytes in length\n",i
);
172 printf("%02x%c",buffer
[j
],(((j
+1)%16)&&((j
+1)!=j
))?' ':'\n');
178 /* unc = \\hostname\share\file... */
179 static BOOL
UNC_SplitName(LPSTR unc
, LPSTR
*hostname
, LPSTR
*share
, LPSTR
*file
)
185 p
= strchr(unc
,'\\');
188 p
= strchr(p
+1,'\\');
208 static BOOL
NB_Lookup(LPCSTR host
, struct sockaddr_in
*addr
)
210 int fd
,on
=1,r
,len
,i
,fromsize
;
212 struct sockaddr_in sin
,fromaddr
;
213 unsigned char buffer
[256];
215 fd
= socket(PF_INET
, SOCK_DGRAM
, IPPROTO_UDP
);
219 r
= setsockopt(fd
, SOL_SOCKET
, SO_BROADCAST
, &on
, sizeof on
);
223 if(0==inet_aton("255.255.255.255", (struct in_addr
*)&sin
.sin_addr
.s_addr
))
225 FIXME("Error getting bcast address\n");
228 sin
.sin_family
= AF_INET
;
229 sin
.sin_port
= htons(137);
231 len
= NB_NameReq(host
,buffer
,sizeof buffer
);
235 r
= sendto(fd
, buffer
, len
, 0, (struct sockaddr
*)&sin
, sizeof sin
);
238 FIXME("Error sending packet\n");
246 /* FIXME: this is simple and easily fooled logic
247 * we should loop until we receive the correct packet or timeout
249 r
= poll(&fds
,1,NB_TIMEOUT
);
253 TRACE("Got response!\n");
255 fromsize
= sizeof (fromaddr
);
256 r
= recvfrom(fd
, buffer
, sizeof buffer
, 0, (struct sockaddr
*)&fromaddr
, &fromsize
);
260 TRACE("%d bytes received\n",r
);
266 DPRINTF("%02X%c",buffer
[i
],(((i
+1)!=r
)&&((i
+1)%16))?' ':'\n');
272 TRACE("packet is OK\n");
274 memcpy(&addr
->sin_addr
, &buffer
[58], sizeof addr
->sin_addr
);
284 #define NB_FIRST 0x40
288 #define NB_SESSION_MSG 0x00
289 #define NB_SESSION_REQ 0x81
291 /* RFC 1002, section 4.3.2 */
292 static BOOL
NB_SessionReq(int fd
, char *called
, char *calling
)
294 unsigned char buffer
[0x100];
298 TRACE("called %s, calling %s\n",called
,calling
);
300 buffer
[0] = NB_SESSION_REQ
;
301 buffer
[1] = NB_FIRST
;
303 netbios_name(called
, &buffer
[NB_HDRSIZE
]);
305 netbios_name(calling
, &buffer
[NB_HDRSIZE
+len
]);
308 NBR_ADDWORD(&buffer
[2],len
);
310 /* for(i=0; i<(len+NB_HDRSIZE); i++)
311 DPRINTF("%02X%c",buffer[i],(((i+1)!=(len+4))&&((i+1)%16))?' ':'\n'); */
313 r
= write(fd
,buffer
,len
+4);
316 ERR("Write failed\n");
324 r
= poll(&fds
,1,NB_TIMEOUT
);
327 ERR("Poll failed\n");
331 r
= read(fd
, buffer
, NB_HDRSIZE
);
332 if((r
!=NB_HDRSIZE
) || (buffer
[0]!=0x82))
334 TRACE("Received %d bytes\n",r
);
335 TRACE("%02x %02x %02x %02x\n", buffer
[0],buffer
[1],buffer
[2],buffer
[3]);
342 static BOOL
NB_SendData(int fd
, struct NB_Buffer
*out
)
344 unsigned char buffer
[NB_HDRSIZE
];
347 /* CHECK: is it always OK to do this in two writes? */
348 /* perhaps use scatter gather sendmsg instead? */
350 buffer
[0] = NB_SESSION_MSG
;
351 buffer
[1] = NB_FIRST
;
352 NBR_ADDWORD(&buffer
[2],out
->len
);
354 r
= write(fd
, buffer
, NB_HDRSIZE
);
358 r
= write(fd
, out
->buffer
, out
->len
);
361 ERR("write failed\n");
368 static BOOL
NB_RecvData(int fd
, struct NB_Buffer
*rx
)
371 unsigned char buffer
[NB_HDRSIZE
];
373 r
= read(fd
, buffer
, NB_HDRSIZE
);
374 if((r
!=NB_HDRSIZE
) || (buffer
[0]!=NB_SESSION_MSG
))
376 ERR("Received %d bytes\n",r
);
380 rx
->len
= NBR_GETWORD(&buffer
[2]);
382 rx
->buffer
= HeapAlloc(GetProcessHeap(), 0, rx
->len
);
386 r
= read(fd
, rx
->buffer
, rx
->len
);
389 TRACE("Received %d bytes\n",r
);
390 HeapFree(GetProcessHeap(), 0, rx
->buffer
);
399 static BOOL
NB_Transaction(int fd
, struct NB_Buffer
*in
, struct NB_Buffer
*out
)
407 DPRINTF("Sending request:\n");
408 for(i
=0; i
<in
->len
; i
++)
409 DPRINTF("%02X%c",in
->buffer
[i
],(((i
+1)!=in
->len
)&&((i
+1)%16))?' ':'\n');
412 if(!NB_SendData(fd
,in
))
419 r
= poll(&fds
,1,NB_TIMEOUT
);
422 ERR("Poll failed\n");
426 if(!NB_RecvData(fd
, out
))
432 DPRINTF("Got response:\n");
433 for(i
=0; i
<out
->len
; i
++)
434 DPRINTF("%02X%c",out
->buffer
[i
],(((i
+1)!=out
->len
)&&((i
+1)%16))?' ':'\n');
440 #define SMB_ADDHEADER(b,l) { b[(l)++]=0xff; b[(l)++]='S'; b[(l)++]='M'; b[(l)++]='B'; }
441 #define SMB_ADDERRINFO(b,l) { b[(l)++]=0; b[(l)++]=0; b[(l)++]=0; b[(l)++]=0; }
442 #define SMB_ADDPADSIG(b,l) { memset(&b[l],0,12); l+=12; }
444 #define SMB_ERRCLASS 5
445 #define SMB_ERRCODE 7
446 #define SMB_TREEID 24
447 #define SMB_PROCID 26
448 #define SMB_USERID 28
449 #define SMB_PLEXID 30
450 #define SMB_PCOUNT 32
451 #define SMB_HDRSIZE 33
453 static DWORD
SMB_GetError(unsigned char *buffer
)
457 switch(buffer
[SMB_ERRCLASS
])
460 return STATUS_SUCCESS
;
465 err_class
= "net server";
468 err_class
= "hardware";
474 err_class
= "unknown";
478 ERR("%s error %d \n",err_class
, buffer
[SMB_ERRCODE
]);
480 /* FIXME: return propper error codes */
481 return STATUS_INVALID_PARAMETER
;
484 static int SMB_Header(unsigned char *buffer
, unsigned char command
, USHORT tree_id
, USHORT user_id
)
490 SMB_ADDHEADER(buffer
,len
);
493 buffer
[len
++] = command
;
496 SMB_ADDERRINFO(buffer
,len
)
499 buffer
[len
++] = 0x00; /* flags */
500 SMB_ADDWORD(&buffer
[len
],1); len
+= 2; /* flags2 */
503 SMB_ADDPADSIG(buffer
,len
)
506 SMB_ADDWORD(&buffer
[len
],tree_id
); len
+= 2; /* treeid */
507 id
= GetCurrentThreadId();
508 SMB_ADDWORD(&buffer
[len
],id
); len
+= 2; /* process id */
509 SMB_ADDWORD(&buffer
[len
],user_id
); len
+= 2; /* user id */
510 SMB_ADDWORD(&buffer
[len
],SMB_MultiplexId
); len
+= 2; /* multiplex id */
516 static const char *SMB_ProtocolDialect
= "NT LM 0.12";
517 /* = "Windows for Workgroups 3.1a"; */
519 /* FIXME: support multiple SMB dialects */
520 static BOOL
SMB_NegotiateProtocol(int fd
, USHORT
*dialect
)
522 unsigned char buf
[0x100];
524 struct NB_Buffer tx
, rx
;
528 memset(buf
,0,sizeof buf
);
531 tx
.len
= SMB_Header(tx
.buffer
, SMB_COM_NEGOTIATE
, 0, 0);
534 tx
.buffer
[tx
.len
++] = 0; /* no parameters */
537 buflen
= strlen(SMB_ProtocolDialect
)+2; /* include type and nul byte */
538 SMB_ADDWORD(&tx
.buffer
[tx
.len
],buflen
); tx
.len
+= 2;
540 tx
.buffer
[tx
.len
] = 0x02;
541 strcpy(&tx
.buffer
[tx
.len
+1],SMB_ProtocolDialect
);
546 if(!NB_Transaction(fd
, &tx
, &rx
))
555 /* FIXME: check response */
556 if(SMB_GetError(rx
.buffer
))
558 ERR("returned error\n");
559 HeapFree(GetProcessHeap(),0,rx
.buffer
);
563 HeapFree(GetProcessHeap(),0,rx
.buffer
);
570 #define SMB_PARAM_COUNT(buffer) ((buffer)[SMB_PCOUNT])
571 #define SMB_PARAM(buffer,n) SMB_GETWORD(&(buffer)[SMB_HDRSIZE+2*(n)])
572 #define SMB_BUFFER_COUNT(buffer) SMB_GETWORD(buffer+SMB_HDRSIZE+2*SMB_PARAM_COUNT(buffer))
573 #define SMB_BUFFER(buffer,n) ((buffer)[SMB_HDRSIZE + 2*SMB_PARAM_COUNT(buffer) + 2 + (n) ])
575 static BOOL
SMB_SessionSetup(int fd
, USHORT
*userid
)
577 unsigned char buf
[0x100];
579 struct NB_Buffer rx
, tx
;
581 memset(buf
,0,sizeof buf
);
584 tx
.len
= SMB_Header(tx
.buffer
, SMB_COM_SESSION_SETUP_ANDX
, 0, 0);
586 tx
.buffer
[tx
.len
++] = 0; /* no parameters? */
588 tx
.buffer
[tx
.len
++] = 0xff; /* AndXCommand: secondary request */
589 tx
.buffer
[tx
.len
++] = 0x00; /* AndXReserved */
590 SMB_ADDWORD(&tx
.buffer
[tx
.len
],0); /* AndXOffset */
592 SMB_ADDWORD(&tx
.buffer
[tx
.len
],0x400); /* MaxBufferSize */
594 SMB_ADDWORD(&tx
.buffer
[tx
.len
],1); /* MaxMpxCount */
596 SMB_ADDWORD(&tx
.buffer
[tx
.len
],0); /* VcNumber */
598 SMB_ADDWORD(&tx
.buffer
[tx
.len
],0); /* SessionKey */
600 SMB_ADDWORD(&tx
.buffer
[tx
.len
],0); /* SessionKey */
602 SMB_ADDWORD(&tx
.buffer
[tx
.len
],0); /* Password length */
604 SMB_ADDWORD(&tx
.buffer
[tx
.len
],0); /* Reserved */
606 SMB_ADDWORD(&tx
.buffer
[tx
.len
],0); /* Reserved */
609 /* FIXME: add name and password here */
610 tx
.buffer
[tx
.len
++] = 0; /* number of bytes in password */
614 if(!NB_Transaction(fd
, &tx
, &rx
))
620 if(SMB_GetError(rx
.buffer
))
623 pcount
= SMB_PARAM_COUNT(rx
.buffer
);
625 if( (SMB_HDRSIZE
+pcount
*2) > rx
.len
)
627 ERR("Bad parameter count %d\n",pcount
);
634 DPRINTF("SMB_COM_SESSION_SETUP response, %d args: ",pcount
);
635 for(i
=0; i
<pcount
; i
++)
636 DPRINTF("%04x ",SMB_PARAM(rx
.buffer
,i
));
640 bcount
= SMB_BUFFER_COUNT(rx
.buffer
);
641 if( (SMB_HDRSIZE
+pcount
*2+2+bcount
) > rx
.len
)
643 ERR("parameter count %x, buffer count %x, len %x\n",pcount
,bcount
,rx
.len
);
650 DPRINTF("response buffer %d bytes: ",bcount
);
651 for(i
=0; i
<bcount
; i
++)
653 unsigned char ch
= SMB_BUFFER(rx
.buffer
,i
);
654 DPRINTF("%c", isprint(ch
)?ch
:' ');
659 *userid
= SMB_GETWORD(&rx
.buffer
[SMB_USERID
]);
661 HeapFree(GetProcessHeap(),0,rx
.buffer
);
665 HeapFree(GetProcessHeap(),0,rx
.buffer
);
670 static BOOL
SMB_TreeConnect(int fd
, USHORT user_id
, LPCSTR share_name
, USHORT
*treeid
)
672 unsigned char buf
[0x100];
674 struct NB_Buffer rx
,tx
;
676 TRACE("%s\n",share_name
);
678 memset(buf
,0,sizeof buf
);
681 tx
.len
= SMB_Header(tx
.buffer
, SMB_COM_TREE_CONNECT
, 0, user_id
);
683 tx
.buffer
[tx
.len
++] = 4; /* parameters */
685 tx
.buffer
[tx
.len
++] = 0xff; /* AndXCommand: secondary request */
686 tx
.buffer
[tx
.len
++] = 0x00; /* AndXReserved */
687 SMB_ADDWORD(&tx
.buffer
[tx
.len
],0); /* AndXOffset */
689 SMB_ADDWORD(&tx
.buffer
[tx
.len
],0); /* Flags */
691 SMB_ADDWORD(&tx
.buffer
[tx
.len
],1); /* Password length */
694 /* SMB command buffer */
695 SMB_ADDWORD(&tx
.buffer
[tx
.len
],3); /* command buffer len */
697 tx
.buffer
[tx
.len
++] = 0; /* null terminated password */
699 slen
= strlen(share_name
);
700 if(slen
<(sizeof buf
-tx
.len
))
701 strcpy(&tx
.buffer
[tx
.len
], share_name
);
706 /* name of the service */
707 tx
.buffer
[tx
.len
++] = 0;
711 if(!NB_Transaction(fd
, &tx
, &rx
))
717 if(SMB_GetError(rx
.buffer
))
719 HeapFree(GetProcessHeap(),0,rx
.buffer
);
723 *treeid
= SMB_GETWORD(&rx
.buffer
[SMB_TREEID
]);
725 HeapFree(GetProcessHeap(),0,rx
.buffer
);
726 TRACE("OK, treeid = %04x\n", *treeid
);
732 static BOOL
SMB_NtCreateOpen(int fd
, USHORT tree_id
, USHORT user_id
, USHORT dialect
,
733 LPCSTR filename
, DWORD access
, DWORD sharing
,
734 LPSECURITY_ATTRIBUTES sa
, DWORD creation
,
735 DWORD attributes
, HANDLE
template, USHORT
*file_id
)
737 unsigned char buffer
[0x100];
740 TRACE("%s\n",filename
);
742 memset(buffer
,0,sizeof buffer
);
744 len
= SMB_Header(buffer
, SMB_COM_NT_CREATE_ANDX
, tree_id
, user_id
);
747 buffer
[len
++] = 24; /* parameters */
749 buffer
[len
++] = 0xff; /* AndXCommand: secondary request */
750 buffer
[len
++] = 0x00; /* AndXReserved */
751 SMB_ADDWORD(&buffer
[len
],0); len
+= 2; /* AndXOffset */
753 buffer
[len
++] = 0; /* reserved */
754 slen
= strlen(filename
);
755 SMB_ADDWORD(&buffer
[len
],slen
); len
+= 2; /* name length */
758 SMB_ADDDWORD(&buffer
[len
],0); len
+= 4; /* flags */
759 SMB_ADDDWORD(&buffer
[len
],0); len
+= 4; /* root directory fid */
761 SMB_ADDDWORD(&buffer
[len
],access
); len
+= 4; /* access */
762 SMB_ADDDWORD(&buffer
[len
],0); len
+= 4; /* allocation size */
764 SMB_ADDDWORD(&buffer
[len
],0); len
+= 4; /* root directory fid */
767 SMB_ADDDWORD(&buffer
[len
],0); len
+= 4; /* initial allocation */
768 SMB_ADDDWORD(&buffer
[len
],0); len
+= 4;
771 SMB_ADDDWORD(&buffer
[len
],attributes
); len
+= 4; /* ExtFileAttributes*/
774 SMB_ADDDWORD(&buffer
[len
],sharing
); len
+= 4; /* ShareAccess */
777 TRACE("creation = %08lx\n",creation
);
778 SMB_ADDDWORD(&buffer
[len
],creation
); len
+= 4; /* CreateDisposition */
781 SMB_ADDDWORD(&buffer
[len
],creation
); len
+= 4; /* CreateOptions */
784 SMB_ADDDWORD(&buffer
[len
],0); len
+= 4; /* Impersonation */
787 buffer
[len
++] = 0; /* security flags */
790 SMB_ADDWORD(&buffer
[len
],slen
); len
+= 2; /* size of buffer */
792 if(slen
<(sizeof buffer
-len
))
793 strcpy(&buffer
[len
], filename
);
798 /* name of the file */
801 if(!NB_Transaction(fd
, buffer
, len
, &len
))
804 if(SMB_GetError(buffer
))
810 /* *file_id = SMB_GETWORD(&buffer[xxx]); */
818 static USHORT
SMB_GetMode(DWORD access
, DWORD sharing
)
822 switch(access
&(GENERIC_READ
|GENERIC_WRITE
))
830 case (GENERIC_READ
|GENERIC_WRITE
):
831 mode
|= OF_READWRITE
;
835 switch(sharing
&(FILE_SHARE_READ
|FILE_SHARE_WRITE
))
837 case (FILE_SHARE_READ
|FILE_SHARE_WRITE
):
838 mode
|= OF_SHARE_DENY_NONE
;
840 case FILE_SHARE_READ
:
841 mode
|= OF_SHARE_DENY_WRITE
;
843 case FILE_SHARE_WRITE
:
844 mode
|= OF_SHARE_DENY_READ
;
847 mode
|= OF_SHARE_EXCLUSIVE
;
855 /* inverse of FILE_ConvertOFMode */
856 static BOOL
SMB_OpenAndX(int fd
, USHORT tree_id
, USHORT user_id
, USHORT dialect
,
857 LPCSTR filename
, DWORD access
, DWORD sharing
,
858 DWORD creation
, DWORD attributes
, USHORT
*file_id
)
860 unsigned char buffer
[0x100];
864 TRACE("%s\n",filename
);
866 mode
= SMB_GetMode(access
,sharing
);
868 memset(buffer
,0,sizeof buffer
);
870 len
= SMB_Header(buffer
, SMB_COM_OPEN_ANDX
, tree_id
, user_id
);
873 buffer
[len
++] = 15; /* parameters */
874 buffer
[len
++] = 0xff; /* AndXCommand: secondary request */
875 buffer
[len
++] = 0x00; /* AndXReserved */
876 SMB_ADDWORD(buffer
+len
,0); len
+=2; /* AndXOffset */
877 SMB_ADDWORD(buffer
+len
,0); len
+=2; /* Flags */
878 SMB_ADDWORD(buffer
+len
,mode
); len
+=2; /* desired access */
879 SMB_ADDWORD(buffer
+len
,0); len
+=2; /* search attributes */
880 SMB_ADDWORD(buffer
+len
,0); len
+=2;
888 static BOOL
SMB_Open(int fd
, USHORT tree_id
, USHORT user_id
, USHORT dialect
,
889 LPCSTR filename
, DWORD access
, DWORD sharing
,
890 DWORD creation
, DWORD attributes
, USHORT
*file_id
)
892 unsigned char buf
[0x100];
894 USHORT mode
= SMB_GetMode(access
,sharing
);
895 struct NB_Buffer rx
,tx
;
897 TRACE("%s\n",filename
);
899 memset(buf
,0,sizeof buf
);
902 tx
.len
= SMB_Header(tx
.buffer
, SMB_COM_OPEN
, tree_id
, user_id
);
905 tx
.buffer
[tx
.len
++] = 2; /* parameters */
906 SMB_ADDWORD(tx
.buffer
+tx
.len
,mode
); tx
.len
+=2;
907 SMB_ADDWORD(tx
.buffer
+tx
.len
,0); tx
.len
+=2; /* search attributes */
909 slen
= strlen(filename
)+2; /* inc. nul and BufferFormat */
910 SMB_ADDWORD(tx
.buffer
+tx
.len
,slen
); tx
.len
+=2;
912 tx
.buffer
[tx
.len
] = 0x04; /* BufferFormat */
913 strcpy(&tx
.buffer
[tx
.len
+1],filename
);
918 if(!NB_Transaction(fd
, &tx
, &rx
))
924 if(SMB_GetError(rx
.buffer
))
927 pcount
= SMB_PARAM_COUNT(rx
.buffer
);
929 if( (SMB_HDRSIZE
+pcount
*2) > rx
.len
)
931 ERR("Bad parameter count %d\n",pcount
);
935 TRACE("response, %d args: ",pcount
);
936 for(i
=0; i
<pcount
; i
++)
937 TRACE("%04x ",SMB_PARAM(rx
.buffer
,i
));
940 *file_id
= SMB_PARAM(rx
.buffer
,0);
942 TRACE("file_id = %04x\n",*file_id
);
948 static BOOL
SMB_Read(int fd
, USHORT tree_id
, USHORT user_id
, USHORT dialect
,
949 USHORT file_id
, DWORD offset
, LPVOID out
, USHORT count
, USHORT
* read
)
952 struct NB_Buffer rx
,tx
;
954 TRACE("user %04x tree %04x file %04x count %04x offset %08lx\n",
955 user_id
, tree_id
, file_id
, count
, offset
);
957 buf_size
= count
+0x100;
958 tx
.buffer
= (unsigned char *) HeapAlloc(GetProcessHeap(),0,buf_size
);
960 memset(tx
.buffer
,0,buf_size
);
962 tx
.len
= SMB_Header(tx
.buffer
, SMB_COM_READ
, tree_id
, user_id
);
964 tx
.buffer
[tx
.len
++] = 5;
965 SMB_ADDWORD(&tx
.buffer
[tx
.len
],file_id
); tx
.len
+= 2;
966 SMB_ADDWORD(&tx
.buffer
[tx
.len
],count
); tx
.len
+= 2;
967 SMB_ADDDWORD(&tx
.buffer
[tx
.len
],offset
); tx
.len
+= 4;
968 SMB_ADDWORD(&tx
.buffer
[tx
.len
],0); tx
.len
+= 2; /* how many more bytes will be read */
970 tx
.buffer
[tx
.len
++] = 0;
974 if(!NB_Transaction(fd
, &tx
, &rx
))
976 HeapFree(GetProcessHeap(),0,tx
.buffer
);
980 if(SMB_GetError(rx
.buffer
))
982 HeapFree(GetProcessHeap(),0,rx
.buffer
);
983 HeapFree(GetProcessHeap(),0,tx
.buffer
);
987 n
= SMB_PARAM_COUNT(rx
.buffer
);
989 if( (SMB_HDRSIZE
+n
*2) > rx
.len
)
991 HeapFree(GetProcessHeap(),0,rx
.buffer
);
992 HeapFree(GetProcessHeap(),0,tx
.buffer
);
993 ERR("Bad parameter count %d\n",n
);
997 TRACE("response, %d args: ",n
);
999 TRACE("%04x ",SMB_PARAM(rx
.buffer
,i
));
1002 n
= SMB_PARAM(rx
.buffer
,5) - 3;
1006 memcpy( out
, &SMB_BUFFER(rx
.buffer
,3), n
);
1008 TRACE("Read %d bytes\n",n
);
1011 HeapFree(GetProcessHeap(),0,tx
.buffer
);
1012 HeapFree(GetProcessHeap(),0,rx
.buffer
);
1019 * setup_count : number of USHORTs in the setup string
1021 struct SMB_Trans2Info
1023 struct NB_Buffer buf
;
1024 unsigned char *setup
;
1026 unsigned char *params
;
1028 unsigned char *data
;
1033 * Do an SMB transaction
1035 * This function allocates memory in the recv structure. It is
1036 * the caller's responsibility to free the memory if it finds
1037 * that recv->buf.buffer is nonzero.
1039 static BOOL
SMB_Transaction2(int fd
, int tree_id
, int user_id
,
1040 struct SMB_Trans2Info
*send
,
1041 struct SMB_Trans2Info
*recv
)
1044 const int retmaxparams
= 0xf000;
1045 const int retmaxdata
= 1024;
1046 const int retmaxsetup
= 0; /* FIXME */
1047 const int flags
= 0;
1048 const int timeout
= 0;
1049 int param_ofs
, data_ofs
;
1050 struct NB_Buffer tx
;
1053 buf_size
= 0x100 + send
->setup_count
*2 + send
->param_count
+ send
->data_count
;
1054 tx
.buffer
= (unsigned char *) HeapAlloc(GetProcessHeap(),0,buf_size
);
1056 tx
.len
= SMB_Header(tx
.buffer
, SMB_COM_TRANSACTION2
, tree_id
, user_id
);
1058 tx
.buffer
[tx
.len
++] = 14 + send
->setup_count
;
1059 SMB_ADDWORD(&tx
.buffer
[tx
.len
],send
->param_count
); /* total param bytes sent */
1061 SMB_ADDWORD(&tx
.buffer
[tx
.len
],send
->data_count
); /* total data bytes sent */
1063 SMB_ADDWORD(&tx
.buffer
[tx
.len
],retmaxparams
); /*max parameter bytes to return */
1065 SMB_ADDWORD(&tx
.buffer
[tx
.len
],retmaxdata
); /* max data bytes to return */
1067 tx
.buffer
[tx
.len
++] = retmaxsetup
;
1068 tx
.buffer
[tx
.len
++] = 0; /* reserved1 */
1070 SMB_ADDWORD(&tx
.buffer
[tx
.len
],flags
); /* flags */
1072 SMB_ADDDWORD(&tx
.buffer
[tx
.len
],timeout
); /* timeout */
1074 SMB_ADDWORD(&tx
.buffer
[tx
.len
],0); /* reserved2 */
1076 SMB_ADDWORD(&tx
.buffer
[tx
.len
],send
->param_count
); /* parameter count - this buffer */
1079 param_ofs
= tx
.len
; /* parameter offset */
1081 SMB_ADDWORD(&tx
.buffer
[tx
.len
],send
->data_count
); /* data count */
1084 data_ofs
= tx
.len
; /* data offset */
1086 tx
.buffer
[tx
.len
++] = send
->setup_count
; /* setup count */
1087 tx
.buffer
[tx
.len
++] = 0; /* reserved3 */
1089 memcpy(&tx
.buffer
[tx
.len
], send
->setup
, send
->setup_count
*2); /* setup */
1090 tx
.len
+= send
->setup_count
*2;
1092 /* add string here when implementing SMB_COM_TRANS */
1094 SMB_ADDWORD(&tx
.buffer
[param_ofs
], tx
.len
);
1095 memcpy(&tx
.buffer
[tx
.len
], send
->params
, send
->param_count
); /* parameters */
1096 tx
.len
+= send
->param_count
;
1098 tx
.len
++; /* pad2 */
1100 SMB_ADDWORD(&tx
.buffer
[data_ofs
], tx
.len
);
1101 if(send
->data_count
&& send
->data
)
1103 memcpy(&tx
.buffer
[tx
.len
], send
->data
, send
->data_count
); /* data */
1104 tx
.len
+= send
->data_count
;
1107 recv
->buf
.buffer
= NULL
;
1109 if(!NB_Transaction(fd
, &tx
, &recv
->buf
))
1112 if(!recv
->buf
.buffer
)
1115 if(SMB_GetError(recv
->buf
.buffer
))
1118 /* reuse these two offsets to check the received message */
1119 param_ofs
= SMB_PARAM(recv
->buf
.buffer
,4);
1120 data_ofs
= SMB_PARAM(recv
->buf
.buffer
,7);
1122 if( (recv
->param_count
+ param_ofs
) > recv
->buf
.len
)
1125 if( (recv
->data_count
+ data_ofs
) > recv
->buf
.len
)
1131 recv
->setup_count
= 0;
1133 recv
->param_count
= SMB_PARAM(recv
->buf
.buffer
,0);
1134 recv
->params
= &recv
->buf
.buffer
[param_ofs
];
1136 recv
->data_count
= SMB_PARAM(recv
->buf
.buffer
,6);
1137 recv
->data
= &recv
->buf
.buffer
[data_ofs
];
1140 TRACE("%d words\n",SMB_PARAM_COUNT(recv->buf.buffer));
1141 TRACE("total parameters = %d\n",SMB_PARAM(recv->buf.buffer,0));
1142 TRACE("total data = %d\n",SMB_PARAM(recv->buf.buffer,1));
1143 TRACE("parameters = %d\n",SMB_PARAM(recv->buf.buffer,3));
1144 TRACE("parameter offset = %d\n",SMB_PARAM(recv->buf.buffer,4));
1145 TRACE("param displace = %d\n",SMB_PARAM(recv->buf.buffer,5));
1147 TRACE("data count = %d\n",SMB_PARAM(recv->buf.buffer,6));
1148 TRACE("data offset = %d\n",SMB_PARAM(recv->buf.buffer,7));
1149 TRACE("data displace = %d\n",SMB_PARAM(recv->buf.buffer,8));
1156 HeapFree(GetProcessHeap(),0,tx
.buffer
);
1161 static BOOL
SMB_SetupFindFirst(struct SMB_Trans2Info
*send
, LPSTR filename
)
1163 int search_attribs
= FILE_ATTRIBUTE_DIRECTORY
| FILE_ATTRIBUTE_HIDDEN
| FILE_ATTRIBUTE_SYSTEM
;
1164 int search_count
= 10;
1166 int infolevel
= 0x104; /* SMB_FILE_BOTH_DIRECTORY_INFO */
1167 int storagetype
= 0;
1170 memset(send
,0,sizeof send
);
1172 send
->setup_count
= 1;
1173 send
->setup
= HeapAlloc(GetProcessHeap(),0,send
->setup_count
*2);
1177 buf_size
= 0x10 + lstrlenA(filename
);
1178 send
->params
= HeapAlloc(GetProcessHeap(),0,buf_size
);
1181 HeapFree(GetProcessHeap(),0,send
->setup
);
1185 SMB_ADDWORD(send
->setup
,TRANS2_FIND_FIRST2
);
1188 memset(send
->params
,0,buf_size
);
1189 SMB_ADDWORD(&send
->params
[len
],search_attribs
); len
+= 2;
1190 SMB_ADDWORD(&send
->params
[len
],search_count
); len
+= 2;
1191 SMB_ADDWORD(&send
->params
[len
],flags
); len
+= 2;
1192 SMB_ADDWORD(&send
->params
[len
],infolevel
); len
+= 2;
1193 SMB_ADDDWORD(&send
->params
[len
],storagetype
); len
+= 4;
1195 strcpy(&send
->params
[len
],filename
);
1196 len
+= lstrlenA(filename
)+1;
1198 send
->param_count
= len
;
1200 send
->data_count
= 0;
1205 static SMB_DIR
*SMB_Trans2FindFirst(int fd
, USHORT tree_id
,
1206 USHORT user_id
, USHORT dialect
, LPSTR filename
)
1210 /* char *filename = "\\*"; */
1211 struct SMB_Trans2Info send
, recv
;
1212 SMB_DIR
*smbdir
= NULL
;
1214 TRACE("patern = %s\n",filename
);
1216 if(!SMB_SetupFindFirst(&send
, filename
))
1219 memset(&recv
,0,sizeof recv
);
1221 ret
= SMB_Transaction2(fd
, tree_id
, user_id
, &send
, &recv
);
1222 HeapFree(GetProcessHeap(),0,send
.params
);
1223 HeapFree(GetProcessHeap(),0,send
.setup
);
1228 if(recv
.setup_count
)
1231 if(recv
.param_count
!= 10)
1234 num
= SMB_GETWORD(&recv
.params
[2]);
1235 TRACE("Success, search id: %d\n",num
);
1237 if(SMB_GETWORD(&recv
.params
[4]))
1238 FIXME("need to read more!\n");
1240 smbdir
= HeapAlloc(GetProcessHeap(),0,sizeof(*smbdir
));
1245 smbdir
->current
= 0;
1246 smbdir
->num_entries
= num
;
1247 smbdir
->entries
= HeapAlloc(GetProcessHeap(), 0, sizeof(unsigned char*)*num
);
1248 if(!smbdir
->entries
)
1250 smbdir
->buffer
= recv
.buf
.buffer
; /* save to free later */
1252 for(i
=0; i
<num
; i
++)
1254 int size
= SMB_GETDWORD(&recv
.data
[ofs
]);
1256 smbdir
->entries
[i
] = &recv
.data
[ofs
];
1261 for(j
=0; j
<size
; j
++)
1262 DPRINTF("%02x%c",recv
.data
[ofs
+j
],((j
+1)%16)?' ':'\n');
1264 TRACE("file %d : %s\n", i
, &recv
.data
[ofs
+0x5e]);
1266 if(ofs
>recv
.data_count
)
1276 if( recv
.buf
.buffer
)
1277 HeapFree(GetProcessHeap(),0,recv
.buf
.buffer
);
1280 if( smbdir
->entries
)
1281 HeapFree(GetProcessHeap(),0,smbdir
->entries
);
1282 HeapFree(GetProcessHeap(),0,smbdir
);
1290 static int SMB_GetSocket(LPCSTR host
)
1293 struct sockaddr_in sin
;
1296 TRACE("host %s\n",host
);
1298 he
= gethostbyname(host
);
1301 memcpy(&sin
.sin_addr
,he
->h_addr
, sizeof (sin
.sin_addr
));
1305 if(NB_Lookup(host
,&sin
))
1308 /* FIXME: resolve by WINS too */
1310 ERR("couldn't resolve SMB host %s\n", host
);
1315 sin
.sin_family
= AF_INET
;
1316 sin
.sin_port
= htons(139); /* netbios session */
1318 fd
= socket(PF_INET
, SOCK_STREAM
, IPPROTO_TCP
);
1323 unsigned char *x
= (unsigned char *)&sin
.sin_addr
;
1324 TRACE("Connecting to %d.%d.%d.%d ...\n", x
[0],x
[1],x
[2],x
[3]);
1326 r
= connect(fd
, (struct sockaddr
*)&sin
, sizeof sin
);
1328 if(!NB_SessionReq(fd
, "*SMBSERVER", "WINE"))
1337 static BOOL
SMB_LoginAndConnect(int fd
, LPCSTR host
, LPCSTR share
, USHORT
*tree_id
, USHORT
*user_id
, USHORT
*dialect
)
1341 TRACE("host %s share %s\n",host
,share
);
1343 if(!SMB_NegotiateProtocol(fd
, dialect
))
1346 if(!SMB_SessionSetup(fd
, user_id
))
1349 name
= HeapAlloc(GetProcessHeap(),0,strlen(host
)+strlen(share
)+5);
1353 sprintf(name
,"\\\\%s\\%s",host
,share
);
1354 if(!SMB_TreeConnect(fd
,*user_id
,name
,tree_id
))
1356 HeapFree(GetProcessHeap(),0,name
);
1363 static HANDLE
SMB_RegisterFile( int fd
, USHORT tree_id
, USHORT user_id
, USHORT dialect
, USHORT file_id
)
1368 wine_server_send_fd( fd
);
1370 SERVER_START_REQ( create_smb
)
1372 req
->tree_id
= tree_id
;
1373 req
->user_id
= user_id
;
1374 req
->file_id
= file_id
;
1378 r
= wine_server_call_err( req
);
1379 ret
= reply
->handle
;
1384 TRACE("created wineserver smb object, handle = %p\n",ret
);
1386 SetLastError( ERROR_PATH_NOT_FOUND
);
1391 HANDLE WINAPI
SMB_CreateFileW( LPCWSTR uncname
, DWORD access
, DWORD sharing
,
1392 LPSECURITY_ATTRIBUTES sa
, DWORD creation
,
1393 DWORD attributes
, HANDLE
template )
1396 USHORT tree_id
=0, user_id
=0, dialect
=0, file_id
=0;
1397 LPSTR name
,host
,share
,file
;
1398 HANDLE handle
= INVALID_HANDLE_VALUE
;
1401 len
= WideCharToMultiByte(CP_ACP
, 0, uncname
, -1, NULL
, 0, NULL
, NULL
);
1402 name
= HeapAlloc(GetProcessHeap(), 0, len
);
1406 WideCharToMultiByte(CP_ACP
, 0, uncname
, -1, name
, len
, NULL
, NULL
);
1408 if( !UNC_SplitName(name
, &host
, &share
, &file
) )
1410 HeapFree(GetProcessHeap(),0,name
);
1414 TRACE("server is %s, share is %s, file is %s\n", host
, share
, file
);
1416 fd
= SMB_GetSocket(host
);
1420 if(!SMB_LoginAndConnect(fd
, host
, share
, &tree_id
, &user_id
, &dialect
))
1424 if(!SMB_NtCreateOpen(fd
, tree_id
, user_id
, dialect
, file
,
1425 access
, sharing
, sa
, creation
, attributes
, template, &file_id
))
1428 ERR("CreateOpen failed\n");
1432 if(!SMB_Open(fd
, tree_id
, user_id
, dialect
, file
,
1433 access
, sharing
, creation
, attributes
, &file_id
))
1436 ERR("CreateOpen failed\n");
1440 handle
= SMB_RegisterFile(fd
, tree_id
, user_id
, dialect
, file_id
);
1443 ERR("register failed\n");
1448 HeapFree(GetProcessHeap(),0,name
);
1452 static BOOL
SMB_GetSmbInfo(HANDLE hFile
, USHORT
*tree_id
, USHORT
*user_id
, USHORT
*dialect
, USHORT
*file_id
, LPDWORD offset
)
1456 SERVER_START_REQ( get_smb_info
)
1458 req
->handle
= hFile
;
1461 r
= wine_server_call_err( req
);
1463 *tree_id
= reply
->tree_id
;
1465 *user_id
= reply
->user_id
;
1467 *file_id
= reply
->file_id
;
1469 *dialect
= reply
->dialect
;
1471 *offset
= reply
->offset
;
1478 static BOOL
SMB_SetOffset(HANDLE hFile
, DWORD offset
)
1482 TRACE("offset = %08lx\n",offset
);
1484 SERVER_START_REQ( get_smb_info
)
1486 req
->handle
= hFile
;
1487 req
->flags
= SMBINFO_SET_OFFSET
;
1488 req
->offset
= offset
;
1490 r
= wine_server_call_err( req
);
1492 *offset = reply->offset; */
1499 BOOL WINAPI
SMB_ReadFile(HANDLE hFile
, LPVOID buffer
, DWORD bytesToRead
, LPDWORD bytesRead
, LPOVERLAPPED lpOverlapped
)
1502 DWORD total
, count
, offset
;
1503 USHORT user_id
, tree_id
, dialect
, file_id
, read
;
1506 TRACE("%p %p %ld %p\n", hFile
, buffer
, bytesToRead
, bytesRead
);
1508 if(!SMB_GetSmbInfo(hFile
, &tree_id
, &user_id
, &dialect
, &file_id
, &offset
))
1511 fd
= FILE_GetUnixHandle(hFile
, GENERIC_READ
);
1518 count
= bytesToRead
- total
;
1524 r
= SMB_Read(fd
, tree_id
, user_id
, dialect
, file_id
, offset
, buffer
, count
, &read
);
1530 buffer
= (char*)buffer
+ read
;
1532 if(total
>=bytesToRead
)
1540 if(!SMB_SetOffset(hFile
, offset
))
1546 SMB_DIR
* WINAPI
SMB_FindFirst(LPCWSTR name
)
1549 LPSTR host
,share
,file
;
1550 USHORT tree_id
=0, user_id
=0, dialect
=0;
1551 SMB_DIR
*ret
= NULL
;
1555 TRACE("Find %s\n",debugstr_w(name
));
1557 len
= WideCharToMultiByte( CP_ACP
, 0, name
, -1, NULL
, 0, NULL
, NULL
);
1558 filename
= HeapAlloc(GetProcessHeap(),0,len
);
1561 WideCharToMultiByte( CP_ACP
, 0, name
, -1, filename
, len
, NULL
, NULL
);
1563 if( !UNC_SplitName(filename
, &host
, &share
, &file
) )
1566 fd
= SMB_GetSocket(host
);
1570 if(!SMB_LoginAndConnect(fd
, host
, share
, &tree_id
, &user_id
, &dialect
))
1573 TRACE("server is %s, share is %s, file is %s\n", host
, share
, file
);
1575 ret
= SMB_Trans2FindFirst(fd
, tree_id
, user_id
, dialect
, file
);
1583 HeapFree(GetProcessHeap(),0,filename
);
1589 BOOL WINAPI
SMB_FindNext(SMB_DIR
*dir
, WIN32_FIND_DATAW
*data
)
1594 TRACE("%d of %d\n",dir
->current
,dir
->num_entries
);
1596 if(dir
->current
>= dir
->num_entries
)
1599 memset(data
, 0, sizeof *data
);
1601 ent
= dir
->entries
[dir
->current
];
1602 len
= SMB_GETDWORD(&ent
[0]);
1606 memcpy(&data
->ftCreationTime
, &ent
[8], 8);
1607 memcpy(&data
->ftLastAccessTime
, &ent
[0x10], 8);
1608 memcpy(&data
->ftLastWriteTime
, &ent
[0x18], 8);
1609 data
->nFileSizeHigh
= SMB_GETDWORD(&ent
[0x30]);
1610 data
->nFileSizeLow
= SMB_GETDWORD(&ent
[0x34]);
1611 data
->dwFileAttributes
= SMB_GETDWORD(&ent
[0x38]);
1613 /* copy the long filename */
1614 fnlen
= SMB_GETDWORD(&ent
[0x3c]);
1615 if ( fnlen
> (sizeof data
->cFileName
/sizeof(WCHAR
)) )
1617 MultiByteToWideChar( CP_ACP
, 0, &ent
[0x5e], fnlen
, data
->cFileName
,
1618 sizeof(data
->cFileName
)/sizeof(WCHAR
) );
1620 /* copy the short filename */
1621 if ( ent
[0x44] > (sizeof data
->cAlternateFileName
/sizeof(WCHAR
)) )
1623 MultiByteToWideChar( CP_ACP
, 0, &ent
[0x5e + len
], ent
[0x44], data
->cAlternateFileName
,
1624 sizeof(data
->cAlternateFileName
)/sizeof(WCHAR
) );
1631 BOOL WINAPI
SMB_CloseDir(SMB_DIR
*dir
)
1633 HeapFree(GetProcessHeap(),0,dir
->buffer
);
1634 HeapFree(GetProcessHeap(),0,dir
->entries
);
1635 memset(dir
,0,sizeof *dir
);
1636 HeapFree(GetProcessHeap(),0,dir
);