1: /*
2: * $Id: libnet_init.c,v 1.1.1.1 2012/02/21 22:14:23 misho Exp $
3: *
4: * libnet
5: * libnet_init.c - Initilization routines.
6: *
7: * Copyright (c) 1998 - 2004 Mike D. Schiffman <mike@infonexus.com>
8: * All rights reserved.
9: *
10: * Redistribution and use in source and binary forms, with or without
11: * modification, are permitted provided that the following conditions
12: * are met:
13: * 1. Redistributions of source code must retain the above copyright
14: * notice, this list of conditions and the following disclaimer.
15: * 2. Redistributions in binary form must reproduce the above copyright
16: * notice, this list of conditions and the following disclaimer in the
17: * documentation and/or other materials provided with the distribution.
18: *
19: * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
20: * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21: * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22: * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
23: * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24: * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25: * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26: * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27: * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28: * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29: * SUCH DAMAGE.
30: *
31: */
32:
33: #if (HAVE_CONFIG_H)
34: #include "../include/config.h"
35: #endif
36: #if (!(_WIN32) || (__CYGWIN__))
37: #include "../include/libnet.h"
38: #else
39: #include "../include/win32/libnet.h"
40: #endif
41:
42: libnet_t *
43: libnet_init(int injection_type, char *device, char *err_buf)
44: {
45: libnet_t *l = NULL;
46:
47: #if !defined(__WIN32__)
48: if (getuid() && geteuid())
49: {
50: snprintf(err_buf, LIBNET_ERRBUF_SIZE,
51: "%s(): UID or EUID of 0 required\n", __func__);
52: goto bad;
53: }
54: #else
55: WSADATA wsaData;
56:
57: if ((WSAStartup(0x0202, &wsaData)) != 0)
58: {
59: snprintf(err_buf, LIBNET_ERRBUF_SIZE,
60: "%s(): unable to initialize winsock 2\n", __func__);
61: goto bad;
62: }
63: #endif
64:
65: l = (libnet_t *)malloc(sizeof (libnet_t));
66: if (l == NULL)
67: {
68: snprintf(err_buf, LIBNET_ERRBUF_SIZE, "%s(): malloc(): %s\n", __func__,
69: strerror(errno));
70: goto bad;
71: }
72:
73: memset(l, 0, sizeof (*l));
74:
75: l->injection_type = injection_type;
76: l->ptag_state = LIBNET_PTAG_INITIALIZER;
77: l->device = (device ? strdup(device) : NULL);
78:
79: strncpy(l->label, LIBNET_LABEL_DEFAULT, LIBNET_LABEL_SIZE);
80: l->label[sizeof(l->label)] = '\0';
81:
82: switch (l->injection_type)
83: {
84: case LIBNET_LINK:
85: case LIBNET_LINK_ADV:
86: if (libnet_select_device(l) == -1)
87: {
88: snprintf(err_buf, LIBNET_ERRBUF_SIZE, l->err_buf);
89: goto bad;
90: }
91: if (libnet_open_link(l) == -1)
92: {
93: snprintf(err_buf, LIBNET_ERRBUF_SIZE, l->err_buf);
94: goto bad;
95: }
96: break;
97: case LIBNET_RAW4:
98: case LIBNET_RAW4_ADV:
99: if (libnet_open_raw4(l) == -1)
100: {
101: snprintf(err_buf, LIBNET_ERRBUF_SIZE, l->err_buf);
102: goto bad;
103: }
104: break;
105: case LIBNET_RAW6:
106: case LIBNET_RAW6_ADV:
107: if (libnet_open_raw6(l) == -1)
108: {
109: snprintf(err_buf, LIBNET_ERRBUF_SIZE, l->err_buf);
110: goto bad;
111: }
112: break;
113: default:
114: snprintf(err_buf, LIBNET_ERRBUF_SIZE,
115: "%s(): unsupported injection type\n", __func__);
116: goto bad;
117: break;
118: }
119:
120: return (l);
121:
122: bad:
123: if (l)
124: {
125: libnet_destroy(l);
126: }
127: return (NULL);
128: }
129:
130: void
131: libnet_destroy(libnet_t *l)
132: {
133: if (l)
134: {
135: close(l->fd);
136: if (l->device)
137: {
138: free(l->device);
139: }
140: libnet_clear_packet(l);
141: free(l);
142: }
143: }
144:
145: void
146: libnet_clear_packet(libnet_t *l)
147: {
148: libnet_pblock_t *p;
149: libnet_pblock_t *next;
150:
151: if (l)
152: {
153: p = l->protocol_blocks;
154: if (p)
155: {
156: for (; p; p = next)
157: {
158: next = p->next;
159: if (p->buf)
160: {
161: free(p->buf);
162: }
163: free(p);
164: }
165: }
166: l->protocol_blocks = NULL;
167: l->total_size = 0;
168: }
169: }
170:
171: void
172: libnet_stats(libnet_t *l, struct libnet_stats *ls)
173: {
174: if (l == NULL)
175: {
176: return;
177: }
178:
179: ls->packets_sent = l->stats.packets_sent;
180: ls->packet_errors = l->stats.packet_errors;
181: ls->bytes_written = l->stats.bytes_written;
182: }
183:
184: int
185: libnet_getfd(libnet_t *l)
186: {
187: if (l == NULL)
188: {
189: return (-1);
190: }
191:
192: return (l->fd);
193: }
194:
195: int8_t *
196: libnet_getdevice(libnet_t *l)
197: {
198: if (l == NULL)
199: {
200: return (NULL);
201: }
202:
203: return (l->device);
204: }
205:
206: u_int8_t *
207: libnet_getpbuf(libnet_t *l, libnet_ptag_t ptag)
208: {
209: libnet_pblock_t *p;
210:
211: if (l == NULL)
212: {
213: return (NULL);
214: }
215:
216: p = libnet_pblock_find(l, ptag);
217: if (p == NULL)
218: {
219: /* err msg set in libnet_pblock_find() */
220: return (NULL);
221: }
222: else
223: {
224: return (p->buf);
225: }
226: }
227:
228: u_int32_t
229: libnet_getpbuf_size(libnet_t *l, libnet_ptag_t ptag)
230: {
231: libnet_pblock_t *p;
232:
233: if (l == NULL)
234: {
235: return (0);
236: }
237:
238: p = libnet_pblock_find(l, ptag);
239: if (p == NULL)
240: {
241: /* err msg set in libnet_pblock_find() */
242: return (0);
243: }
244: else
245: {
246: return (p->b_len);
247: }
248: }
249:
250: u_int32_t
251: libnet_getpacket_size(libnet_t *l)
252: {
253: libnet_pblock_t *p;
254: u_int32_t n;
255:
256: if (l == NULL)
257: {
258: return (0);
259: }
260:
261: n = 0;
262: p = l->protocol_blocks;
263: if (p)
264: {
265: for (; p; p = p->next)
266: {
267: n += p->b_len;
268: }
269: }
270: return (n);
271: }
272:
273: /* EOF */
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to libnet_init.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / libnet / src