smbXcli: rebuild smb1.recv_iov array if we expect more than one response
[Samba/gebeck_regimport.git] / source3 / utils / log2pcaphex.c
blob0b1230efe7992b58a86c1c585cfd493e4258d2a2
1 /*
2 Unix SMB/CIFS implementation.
3 Utility to extract pcap files from samba (log level 10) log files
5 Copyright (C) Jelmer Vernooij 2003
6 Thanks to Tim Potter for the genial idea
8 Portions (from capconvert.c) (C) Andrew Tridgell 1997
9 Portions (from text2pcap.c) (C) Ashok Narayanan 2001
11 Example:
12 Output NBSS(SMB) packets in hex and convert to pcap adding
13 Eth/IP/TCP headers
15 log2pcap -h < samba.log | text2pcap -T 139,139 - samba.pcap
17 Output directly to pcap format without Eth headers or TCP
18 sequence numbers
20 log2pcap samba.log samba.pcap
22 TODO:
23 - Hex to text2pcap outputs are not properly parsed in Wireshark
24 the NBSS or SMB level. This is a bug.
25 - Writing directly to pcap format doesn't include sequence numbers
26 in the TCP packets
27 - Check if a packet is a response or request and set IP to/from
28 addresses accordingly. Currently all packets come from the same
29 dummy IP and go to the same dummy IP
30 - Add a message when done parsing about the number of pacekts
31 processed
32 - Parse NBSS packet header data from log file
33 - Have correct IP and TCP checksums.
35 Warning:
36 Samba log level 10 outputs a max of 512 bytes from the packet data
37 section. Packets larger than this will be truncated.
39 This program is free software; you can redistribute it and/or modify
40 it under the terms of the GNU General Public License as published by
41 the Free Software Foundation; either version 3 of the License, or
42 (at your option) any later version.
44 This program is distributed in the hope that it will be useful,
45 but WITHOUT ANY WARRANTY; without even the implied warranty of
46 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
47 GNU General Public License for more details.
49 You should have received a copy of the GNU General Public License
50 along with this program. If not, see <http://www.gnu.org/licenses/>.
53 #include "includes.h"
54 #include "popt_common.h"
56 /* We don't care about the paranoid malloc checker in this standalone
57 program */
58 #undef malloc
60 #include <assert.h>
62 int quiet = 0;
63 int hexformat = 0;
65 #define itoa(a) ((a) < 0xa?'0'+(a):'A' + (a-0xa))
67 #include <stdlib.h>
68 #include <unistd.h>
69 #include <sys/time.h>
70 #include <stdio.h>
71 #include <fcntl.h>
73 #define TCPDUMP_MAGIC 0xa1b2c3d4
75 /* tcpdump file format */
76 struct tcpdump_file_header {
77 uint32 magic;
78 uint16 major;
79 uint16 minor;
80 int32 zone;
81 uint32 sigfigs;
82 uint32 snaplen;
83 uint32 linktype;
86 struct tcpdump_packet {
87 struct timeval ts;
88 uint32 caplen;
89 uint32 len;
92 typedef struct {
93 uint8 ver_hdrlen;
94 uint8 dscp;
95 uint16 packet_length;
96 uint16 identification;
97 uint8 flags;
98 uint8 fragment;
99 uint8 ttl;
100 uint8 protocol;
101 uint16 hdr_checksum;
102 uint32 src_addr;
103 uint32 dest_addr;
104 } hdr_ip_t;
106 static hdr_ip_t HDR_IP = {0x45, 0, 0, 0x3412, 0, 0, 0xff, 6, 0, 0x01010101, 0x02020202};
108 typedef struct {
109 uint16 source_port;
110 uint16 dest_port;
111 uint32 seq_num;
112 uint32 ack_num;
113 uint8 hdr_length;
114 uint8 flags;
115 uint16 window;
116 uint16 checksum;
117 uint16 urg;
118 } hdr_tcp_t;
120 static hdr_tcp_t HDR_TCP = {139, 139, 0, 0, 0x50, 0, 0, 0, 0};
122 static void print_pcap_header(FILE *out)
124 struct tcpdump_file_header h;
125 h.magic = TCPDUMP_MAGIC;
126 h.major = 2;
127 h.minor = 4;
128 h.zone = 0;
129 h.sigfigs = 0;
130 h.snaplen = 102400; /* As long packets as possible */
131 h.linktype = 101; /* Raw IP */
132 fwrite(&h, sizeof(struct tcpdump_file_header), 1, out);
135 static void print_pcap_packet(FILE *out, unsigned char *data, long length,
136 long caplen)
138 static int i = 0;
139 struct tcpdump_packet p;
140 i++;
141 p.ts.tv_usec = 0;
142 p.ts.tv_sec = 0;
143 p.caplen = caplen;
144 p.len = length;
145 fwrite(&p, sizeof(struct tcpdump_packet), 1, out);
146 fwrite(data, sizeof(unsigned char), caplen, out);
149 static void print_hex_packet(FILE *out, unsigned char *data, long length)
151 long i,cur = 0;
152 while(cur < length) {
153 fprintf(out, "%06lX ", cur);
154 for(i = cur; i < length && i < cur + 16; i++) {
155 fprintf(out, "%02x ", data[i]);
158 cur = i;
159 fprintf(out, "\n");
163 static void print_netbios_packet(FILE *out, unsigned char *data, long length,
164 long actual_length)
166 unsigned char *newdata; long offset = 0;
167 long newlen;
169 newlen = length+sizeof(HDR_IP)+sizeof(HDR_TCP);
170 newdata = (unsigned char *)malloc(newlen);
172 HDR_IP.packet_length = htons(newlen);
173 HDR_TCP.window = htons(0x2000);
174 HDR_TCP.source_port = HDR_TCP.dest_port = htons(139);
176 memcpy(newdata+offset, &HDR_IP, sizeof(HDR_IP));offset+=sizeof(HDR_IP);
177 memcpy(newdata+offset, &HDR_TCP, sizeof(HDR_TCP));offset+=sizeof(HDR_TCP);
178 memcpy(newdata+offset,data,length);
180 print_pcap_packet(out, newdata, newlen, actual_length+offset);
181 free(newdata);
184 unsigned char *curpacket = NULL;
185 unsigned short curpacket_len = 0;
186 long line_num = 0;
188 /* Read the log message produced by lib/util.c:show_msg() containing the:
189 * SMB_HEADER
190 * SMB_PARAMETERS
191 * SMB_DATA.ByteCount
193 * Example:
194 * [2007/04/08 20:41:39, 5] lib/util.c:show_msg(516)
195 * size=144
196 * smb_com=0x73
197 * smb_rcls=0
198 * smb_reh=0
199 * smb_err=0
200 * smb_flg=136
201 * smb_flg2=49153
202 * smb_tid=1
203 * smb_pid=65279
204 * smb_uid=0
205 * smb_mid=64
206 * smt_wct=3
207 * smb_vwv[ 0]= 117 (0x75)
208 * smb_vwv[ 1]= 128 (0x80)
209 * smb_vwv[ 2]= 1 (0x1)
210 * smb_bcc=87
212 static void read_log_msg(FILE *in, unsigned char **_buffer,
213 unsigned short *buffersize, long *data_offset,
214 long *data_length)
216 unsigned char *buffer;
217 int tmp; long i;
218 assert(fscanf(in, " size=%hu\n", buffersize)); line_num++;
219 buffer = (unsigned char *)malloc(*buffersize+4); /* +4 for NBSS Header */
220 memset(buffer, 0, *buffersize+4);
221 /* NetBIOS Session Service */
222 buffer[0] = 0x00;
223 buffer[1] = 0x00;
224 memcpy(buffer+2, &buffersize, 2); /* TODO: need to copy as little-endian regardless of platform */
225 /* SMB Packet */
226 buffer[4] = 0xFF;
227 buffer[5] = 'S';
228 buffer[6] = 'M';
229 buffer[7] = 'B';
230 assert(fscanf(in, " smb_com=0x%x\n", &tmp)); buffer[smb_com] = tmp; line_num++;
231 assert(fscanf(in, " smb_rcls=%d\n", &tmp)); buffer[smb_rcls] = tmp; line_num++;
232 assert(fscanf(in, " smb_reh=%d\n", &tmp)); buffer[smb_reh] = tmp; line_num++;
233 assert(fscanf(in, " smb_err=%d\n", &tmp)); memcpy(buffer+smb_err, &tmp, 2); line_num++;
234 assert(fscanf(in, " smb_flg=%d\n", &tmp)); buffer[smb_flg] = tmp; line_num++;
235 assert(fscanf(in, " smb_flg2=%d\n", &tmp)); memcpy(buffer+smb_flg2, &tmp, 2); line_num++;
236 assert(fscanf(in, " smb_tid=%d\n", &tmp)); memcpy(buffer+smb_tid, &tmp, 2); line_num++;
237 assert(fscanf(in, " smb_pid=%d\n", &tmp)); memcpy(buffer+smb_pid, &tmp, 2); line_num++;
238 assert(fscanf(in, " smb_uid=%d\n", &tmp)); memcpy(buffer+smb_uid, &tmp, 2); line_num++;
239 assert(fscanf(in, " smb_mid=%d\n", &tmp)); memcpy(buffer+smb_mid, &tmp, 2); line_num++;
240 assert(fscanf(in, " smt_wct=%d\n", &tmp)); buffer[smb_wct] = tmp; line_num++;
241 for(i = 0; i < buffer[smb_wct]; i++) {
242 assert(fscanf(in, " smb_vwv[%*3d]=%*5d (0x%X)\n", &tmp)); line_num++;
243 memcpy(buffer+smb_vwv+i*2, &tmp, 2);
246 *data_offset = smb_vwv+buffer[smb_wct]*2;
247 assert(fscanf(in, " smb_bcc=%ld\n", data_length)); buffer[(*data_offset)] = *data_length; line_num++;
248 (*data_offset)+=2;
249 *_buffer = buffer;
252 /* Read the log message produced by lib/util.c:dump_data() containing:
253 * SMB_DATA.Bytes
255 * Example:
256 * [2007/04/08 20:41:39, 10] lib/util.c:dump_data(2243)
257 * [000] 00 55 00 6E 00 69 00 78 00 00 00 53 00 61 00 6D .U.n.i.x ...S.a.m
258 * [010] 00 62 00 61 00 20 00 33 00 2E 00 30 00 2E 00 32 .b.a. .3 ...0...2
259 * [020] 00 34 00 2D 00 49 00 73 00 69 00 6C 00 6F 00 6E .4.-.I.s .i.l.o.n
260 * [030] 00 20 00 4F 00 6E 00 65 00 46 00 53 00 20 00 76 . .O.n.e .F.S. .v
261 * [040] 00 34 00 2E 00 30 00 00 00 49 00 53 00 49 00 4C .4...0.. .I.S.I.L
262 * [050] 00 4F 00 4E 00 00 00 .O.N...
264 static long read_log_data(FILE *in, unsigned char *buffer, long data_length)
266 long i, addr; char real[2][16]; int ret;
267 unsigned int tmp;
268 for(i = 0; i < data_length; i++) {
269 if(i % 16 == 0){
270 if(i != 0) {
271 /* Read and discard the ascii data after each line. */
272 assert(fscanf(in, " %8c %8c\n", real[0], real[1]) == 2);
274 ret = fscanf(in, " [%03lX]", &addr); line_num++;
275 if(!ret) {
276 if(!quiet)
277 fprintf(stderr, "%ld: Only first %ld bytes are logged, "
278 "packet trace will be incomplete\n", line_num, i-1);
279 return i-1;
281 assert(addr == i);
283 if(!fscanf(in, "%02X", &tmp)) {
284 if(!quiet)
285 fprintf(stderr, "%ld: Log message formated incorrectly. "
286 "Only first %ld bytes are logged, packet trace will "
287 "be incomplete\n", line_num, i-1);
288 while ((tmp = getc(in)) != '\n');
289 return i-1;
291 buffer[i] = tmp;
294 /* Consume the newline so we don't increment num_lines twice */
295 while ((tmp = getc(in)) != '\n');
296 return data_length;
299 int main (int argc, char **argv)
301 const char *infile, *outfile;
302 FILE *out, *in;
303 int opt;
304 poptContext pc;
305 char buffer[4096];
306 long data_offset, data_length;
307 long data_bytes_read = 0;
308 int in_packet = 0;
309 struct poptOption long_options[] = {
310 POPT_AUTOHELP
311 { "quiet", 'q', POPT_ARG_NONE, &quiet, 0, "Be quiet, don't output warnings" },
312 { "hex", 'h', POPT_ARG_NONE, &hexformat, 0, "Output format readable by text2pcap" },
313 POPT_TABLEEND
316 pc = poptGetContext(NULL, argc, (const char **) argv, long_options,
317 POPT_CONTEXT_KEEP_FIRST);
318 poptSetOtherOptionHelp(pc, "[<infile> [<outfile>]]");
321 while((opt = poptGetNextOpt(pc)) != -1) {
322 switch (opt) {
326 poptGetArg(pc); /* Drop argv[0], the program name */
328 infile = poptGetArg(pc);
330 if(infile) {
331 in = fopen(infile, "r");
332 if(!in) {
333 perror("fopen");
334 return 1;
336 } else in = stdin;
338 outfile = poptGetArg(pc);
340 if(outfile) {
341 out = fopen(outfile, "w+");
342 if(!out) {
343 perror("fopen");
344 fprintf(stderr, "Can't find %s, using stdout...\n", outfile);
345 return 1;
349 if(!outfile) out = stdout;
351 if(!hexformat)print_pcap_header(out);
353 while(!feof(in)) {
354 fgets(buffer, sizeof(buffer), in); line_num++;
355 if(buffer[0] == '[') { /* Header */
356 if(strstr(buffer, "show_msg")) {
357 in_packet++;
358 if(in_packet == 1)continue;
359 read_log_msg(in, &curpacket, &curpacket_len, &data_offset, &data_length);
360 } else if(in_packet && strstr(buffer, "dump_data")) {
361 data_bytes_read = read_log_data(in, curpacket+data_offset, data_length);
362 } else {
363 if(in_packet){
364 if(hexformat) print_hex_packet(out, curpacket, curpacket_len);
365 else print_netbios_packet(out, curpacket, curpacket_len, data_bytes_read+data_offset);
366 free(curpacket);
368 in_packet = 0;
373 if (in != stdin) {
374 fclose(in);
377 if (out != stdout) {
378 fclose(out);
381 return 0;