shmif_dumpbus.c revision 1.8 1 /* $NetBSD: shmif_dumpbus.c,v 1.8 2011/09/16 15:39:29 joerg Exp $ */
2
3 /*-
4 * Copyright (c) 2010 Antti Kantee. All Rights Reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
16 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
17 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
18 * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
21 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 * SUCH DAMAGE.
26 */
27
28 /*
29 * Convert shmif bus traffic to a pcap file which can be then
30 * examined with tcpdump -r, wireshark, etc.
31 */
32
33 #include <sys/types.h>
34 #include <sys/mman.h>
35 #include <sys/stat.h>
36
37 #include <machine/bswap.h>
38
39 #include <assert.h>
40 #include <err.h>
41 #include <fcntl.h>
42 #include <inttypes.h>
43 #include <pcap.h>
44 #include <stdbool.h>
45 #include <stdio.h>
46 #include <stdlib.h>
47 #include <string.h>
48 #include <unistd.h>
49
50 #include "shmifvar.h"
51
52 __dead static void
53 usage(void)
54 {
55
56 fprintf(stderr, "usage: %s [-h] [-p pcapfile] buspath\n",getprogname());
57 exit(1);
58 }
59
60 #define BUFSIZE 64*1024
61 #define SWAPME(a) (doswap ? bswap32(a) : (a))
62 #define SWAPME64(a) (doswap ? bswap64(a) : (a))
63 int
64 main(int argc, char *argv[])
65 {
66 struct stat sb;
67 void *busmem;
68 const char *pcapfile = NULL;
69 uint32_t curbus, buslast;
70 struct shmif_mem *bmem;
71 int fd, i, ch;
72 int bonus;
73 char *buf;
74 bool hflag = false, doswap = false, isstdout;
75 pcap_dumper_t *pdump;
76
77 setprogname(argv[0]);
78 while ((ch = getopt(argc, argv, "hp:")) != -1) {
79 switch (ch) {
80 case 'h':
81 hflag = true;
82 break;
83 case 'p':
84 pcapfile = optarg;
85 break;
86 default:
87 usage();
88 }
89 }
90
91 argc -= optind;
92 argv += optind;
93
94 if (argc != 1)
95 usage();
96
97 buf = malloc(BUFSIZE);
98 if (buf == NULL)
99 err(1, "malloc");
100
101 fd = open(argv[0], O_RDONLY);
102 if (fd == -1)
103 err(1, "open bus");
104
105 if (fstat(fd, &sb) == -1)
106 err(1, "stat");
107
108 busmem = mmap(NULL, sb.st_size, PROT_READ, MAP_FILE|MAP_SHARED, fd, 0);
109 if (busmem == MAP_FAILED)
110 err(1, "mmap");
111 bmem = busmem;
112
113 if (bmem->shm_magic != SHMIF_MAGIC) {
114 if (bmem->shm_magic != bswap32(SHMIF_MAGIC))
115 errx(1, "%s not a shmif bus", argv[0]);
116 doswap = 1;
117 }
118 if (SWAPME(bmem->shm_version) != SHMIF_VERSION)
119 errx(1, "bus vesrsion %d, program %d",
120 SWAPME(bmem->shm_version), SHMIF_VERSION);
121 printf("bus version %d, lock: %d, generation: %" PRIu64
122 ", firstoff: 0x%04x, lastoff: 0x%04x\n",
123 SWAPME(bmem->shm_version), SWAPME(bmem->shm_lock),
124 SWAPME64(bmem->shm_gen),
125 SWAPME(bmem->shm_first), SWAPME(bmem->shm_last));
126
127 if (hflag)
128 exit(0);
129
130 if (pcapfile) {
131 isstdout = strcmp(pcapfile, "-") == 0;
132 pcap_t *pcap = pcap_open_dead(DLT_EN10MB, 1518);
133 pdump = pcap_dump_open(pcap, pcapfile);
134 if (pdump == NULL)
135 err(1, "cannot open pcap dump file");
136 } else {
137 /* XXXgcc */
138 isstdout = false;
139 pdump = NULL;
140 }
141
142 curbus = SWAPME(bmem->shm_first);
143 buslast = SWAPME(bmem->shm_last);
144 if (curbus == BUSMEM_DATASIZE)
145 curbus = 0;
146
147 bonus = 0;
148 if (buslast < curbus)
149 bonus = 1;
150
151 i = 0;
152 while (curbus <= buslast || bonus) {
153 struct pcap_pkthdr packhdr;
154 struct shmif_pkthdr sp;
155 uint32_t oldoff;
156 uint32_t curlen;
157 bool wrap;
158
159 assert(curbus < sb.st_size);
160
161 wrap = false;
162 oldoff = curbus;
163 curbus = shmif_busread(bmem, &sp, oldoff, sizeof(sp), &wrap);
164 if (wrap)
165 bonus = 0;
166
167 assert(curbus < sb.st_size);
168 curlen = SWAPME(sp.sp_len);
169
170 if (curlen == 0) {
171 continue;
172 }
173
174 if (!(pcapfile && isstdout))
175 printf("packet %d, offset 0x%04x, length 0x%04x, "
176 "ts %d/%06d\n", i++, curbus,
177 curlen, SWAPME(sp.sp_sec), SWAPME(sp.sp_usec));
178
179 if (!pcapfile) {
180 curbus = shmif_busread(bmem,
181 buf, curbus, curlen, &wrap);
182 if (wrap)
183 bonus = 0;
184 continue;
185 }
186
187 memset(&packhdr, 0, sizeof(packhdr));
188 packhdr.caplen = packhdr.len = curlen;
189 packhdr.ts.tv_sec = SWAPME(sp.sp_sec);
190 packhdr.ts.tv_usec = SWAPME(sp.sp_usec);
191 assert(curlen <= BUFSIZE);
192
193 curbus = shmif_busread(bmem, buf, curbus, curlen, &wrap);
194 pcap_dump((u_char *)pdump, &packhdr, (u_char *)buf);
195 if (wrap)
196 bonus = 0;
197 }
198
199 if (pcapfile)
200 pcap_dump_close(pdump);
201
202 return 0;
203 }
204