2 Unix SMB/Netbios implementation.
4 SMB filter/socket plugin
5 Copyright (C) Andrew Tridgell 1999
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 2 of the License, or
10 (at your option) any later version.
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with this program; if not, write to the Free Software
19 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
25 #define SECURITY_MASK 0
26 #define SECURITY_SET 0
28 /* this forces non-unicode */
29 #define CAPABILITY_MASK CAP_UNICODE
30 #define CAPABILITY_SET 0
32 /* and non-unicode for the client too */
33 #define CLI_CAPABILITY_MASK CAP_UNICODE
34 #define CLI_CAPABILITY_SET 0
36 static char *netbiosname
;
37 static char packet
[BUFFER_SIZE
];
39 extern int DEBUGLEVEL
;
41 static void filter_reply(char *buf
)
43 int msg_type
= CVAL(buf
,0);
44 int type
= CVAL(buf
,smb_com
);
52 /* force the security bits */
53 x
= CVAL(buf
, smb_vwv1
);
54 x
= (x
| SECURITY_SET
) & ~SECURITY_MASK
;
55 SCVAL(buf
, smb_vwv1
, x
);
57 /* force the capabilities */
58 x
= IVAL(buf
,smb_vwv9
+1);
59 x
= (x
| CAPABILITY_SET
) & ~CAPABILITY_MASK
;
60 SIVAL(buf
, smb_vwv9
+1, x
);
66 static void filter_request(char *buf
)
68 int msg_type
= CVAL(buf
,0);
69 int type
= CVAL(buf
,smb_com
);
74 /* it's a netbios special */
78 name_extract(buf
,4,name1
);
79 name_extract(buf
,4 + name_len(buf
+ 4),name2
);
80 DEBUG(0,("sesion_request: %s -> %s\n",
83 /* replace the destination netbios name */
84 name_mangle(netbiosname
, buf
+4, 0x20);
90 /* it's an ordinary SMB request */
93 /* force the client capabilities */
94 x
= IVAL(buf
,smb_vwv11
);
95 x
= (x
| CLI_CAPABILITY_SET
) & ~CLI_CAPABILITY_MASK
;
96 SIVAL(buf
, smb_vwv11
, x
);
103 static void filter_child(int c
, struct in_addr dest_ip
)
107 /* we have a connection from a new client, now connect to the server */
108 s
= open_socket_out(SOCK_STREAM
, &dest_ip
, 139, LONG_CONNECT_TIMEOUT
);
111 DEBUG(0,("Unable to connect to %s\n", inet_ntoa(dest_ip
)));
115 while (c
!= -1 || s
!= -1) {
120 if (s
!= -1) FD_SET(s
, &fds
);
121 if (c
!= -1) FD_SET(c
, &fds
);
123 num
= sys_select(MAX(s
+1, c
+1),&fds
,NULL
);
124 if (num
<= 0) continue;
126 if (c
!= -1 && FD_ISSET(c
, &fds
)) {
127 if (!receive_smb(c
, packet
, 0)) {
128 DEBUG(0,("client closed connection\n"));
131 filter_request(packet
);
132 if (!send_smb(s
, packet
)) {
133 DEBUG(0,("server is dead\n"));
137 if (s
!= -1 && FD_ISSET(s
, &fds
)) {
138 if (!receive_smb(s
, packet
, 0)) {
139 DEBUG(0,("server closed connection\n"));
142 filter_reply(packet
);
143 if (!send_smb(c
, packet
)) {
144 DEBUG(0,("client is dead\n"));
149 DEBUG(0,("Connection closed\n"));
154 static void start_filter(char *desthost
)
157 struct in_addr dest_ip
;
161 /* start listening on port 139 locally */
162 s
= open_socket_in(SOCK_STREAM
, 139, 0, 0, True
);
165 DEBUG(0,("bind failed\n"));
169 if (listen(s
, 5) == -1) {
170 DEBUG(0,("listen failed\n"));
173 if (!resolve_name(desthost
, &dest_ip
, 0x20)) {
174 DEBUG(0,("Unable to resolve host %s\n", desthost
));
181 struct sockaddr addr
;
182 int in_addrlen
= sizeof(addr
);
187 num
= sys_select(s
+1,&fds
,NULL
);
189 c
= accept(s
, &addr
, &in_addrlen
);
193 filter_child(c
, dest_ip
);
204 int main(int argc
, char *argv
[])
211 setup_logging(argv
[0],True
);
213 charset_initialise();
215 pstrcpy(configfile
,CONFIGFILE
);
218 fprintf(stderr
,"smbfilter <desthost> <netbiosname>\n");
224 netbiosname
= argv
[2];
227 if (!lp_load(configfile
,True
,False
,False
)) {
228 DEBUG(0,("Unable to load config file\n"));
231 start_filter(desthost
);