sendpacket.c 46 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551
  1. /* $Id$ */
  2. /*
  3. * Copyright (c) 2001-2010 Aaron Turner <aturner at synfin dot net>
  4. * Copyright (c) 2013-2022 Fred Klassen <tcpreplay at appneta dot com> - AppNeta
  5. *
  6. * The Tcpreplay Suite of tools is free software: you can redistribute it
  7. * and/or modify it under the terms of the GNU General Public License as
  8. * published by the Free Software Foundation, either version 3 of the
  9. * License, or with the authors permission any later version.
  10. *
  11. * The Tcpreplay Suite is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14. * GNU General Public License for more details.
  15. *
  16. * You should have received a copy of the GNU General Public License
  17. * along with the Tcpreplay Suite. If not, see <http://www.gnu.org/licenses/>.
  18. */
  19. /* sendpacket.[ch] is my attempt to write a universal packet injection
  20. * API for BPF, libpcap, libdnet, and Linux's PF_PACKET. I got sick
  21. * and tired dealing with libnet bugs and its lack of active maintenance,
  22. * but unfortunately, libpcap frame injection support is relatively new
  23. * and not everyone uses Linux, so I decided to support all four as
  24. * best as possible. If your platform/OS/hardware supports an additional
  25. * injection method, then by all means add it here (and send me a patch).
  26. *
  27. * Anyways, long story short, for now the order of preference is:
  28. * 0. pcap_dump
  29. * 1. TX_RING
  30. * 2. PF_PACKET
  31. * 3. BPF
  32. * 4. libdnet
  33. * 5. pcap_inject()
  34. * 6. pcap_sendpacket()
  35. *
  36. * Right now, one big problem with the pcap_* methods is that libpcap
  37. * doesn't provide a reliable method of getting the MAC address of
  38. * an interface (required for tcpbridge).
  39. * You can use PF_PACKET or BPF to get that, but if your system supports
  40. * those, might as well inject directly without going through another
  41. * level of indirection.
  42. *
  43. * Please note that some of this code was copied from Libnet 1.1.3
  44. */
  45. #include "sendpacket.h"
  46. #include "defines.h"
  47. #include "config.h"
  48. #include "common.h"
  49. #include <errno.h>
  50. #include <stdarg.h>
  51. #include <stdio.h>
  52. #include <string.h>
  53. #include <sys/file.h>
  54. #include <sys/ioctl.h>
  55. #include <sys/socket.h>
  56. #include <sys/time.h>
  57. #include <sys/types.h>
  58. #ifdef FORCE_INJECT_TX_RING
  59. /* TX_RING uses PF_PACKET API so don't undef it here */
  60. #undef HAVE_LIBDNET
  61. #undef HAVE_PCAP_INJECT
  62. #undef HAVE_PCAP_SENDPACKET
  63. #undef HAVE_BPF
  64. #undef HAVE_LIBXDP
  65. #endif
  66. #ifdef FORCE_INJECT_PF_PACKET
  67. #undef HAVE_TX_RING
  68. #undef HAVE_LIBDNET
  69. #undef HAVE_PCAP_INJECT
  70. #undef HAVE_PCAP_SENDPACKET
  71. #undef HAVE_BPF
  72. #undef HAVE_LIBXDP
  73. #endif
  74. #ifdef FORCE_INJECT_LIBDNET
  75. #undef HAVE_TX_RING
  76. #undef HAVE_PF_PACKET
  77. #undef HAVE_PCAP_INJECT
  78. #undef HAVE_PCAP_SENDPACKET
  79. #undef HAVE_BPF
  80. #undef HAVE_LIBXDP
  81. #endif
  82. #ifdef FORCE_INJECT_BPF
  83. #undef HAVE_TX_RING
  84. #undef HAVE_LIBDNET
  85. #undef HAVE_PCAP_INJECT
  86. #undef HAVE_PCAP_SENDPACKET
  87. #undef HAVE_PF_PACKET
  88. #undef HAVE_LIBXDP
  89. #endif
  90. #ifdef FORCE_INJECT_PCAP_INJECT
  91. #undef HAVE_TX_RING
  92. #undef HAVE_LIBDNET
  93. #undef HAVE_PCAP_SENDPACKET
  94. #undef HAVE_BPF
  95. #undef HAVE_PF_PACKET
  96. #undef HAVE_LIBXDP
  97. #endif
  98. #ifdef FORCE_INJECT_PCAP_SENDPACKET
  99. #undef HAVE_TX_RING
  100. #undef HAVE_LIBDNET
  101. #undef HAVE_PCAP_INJECT
  102. #undef HAVE_BPF
  103. #undef HAVE_PF_PACKET
  104. #undef HAVE_LIBXDP
  105. #endif
  106. #ifdef FORCE_INJECT_LIBXDP
  107. #undef HAVE_TX_RING
  108. #undef HAVE_LIBDNET
  109. #undef HAVE_PF_PACKET
  110. #undef HAVE_PCAP_INJECT
  111. #undef HAVE_PCAP_SENDPACKET
  112. #undef HAVE_BPF
  113. #endif
  114. #if (defined HAVE_WINPCAP && defined HAVE_PCAP_INJECT)
  115. #undef HAVE_PCAP_INJECT /* configure returns true for some odd reason */
  116. #endif
  117. #if !defined HAVE_PCAP_INJECT && !defined HAVE_PCAP_SENDPACKET && !defined HAVE_LIBDNET && !defined HAVE_PF_PACKET && \
  118. !defined HAVE_BPF && !defined TX_RING && !defined HAVE_LIBXDP
  119. #error You need pcap_inject() or pcap_sendpacket() from libpcap, libdnet, Linux's PF_PACKET/TX_RING/AF_XDP with libxdp or *BSD's BPF
  120. #endif
  121. #ifdef HAVE_SYS_PARAM_H
  122. #include <sys/param.h>
  123. #endif
  124. #ifdef HAVE_NET_ROUTE_H
  125. #include <net/route.h>
  126. #endif
  127. #include <stdlib.h>
  128. #include <unistd.h>
  129. #ifdef HAVE_PF_PACKET
  130. #undef INJECT_METHOD
  131. /* give priority to TX_RING */
  132. #ifndef HAVE_TX_RING
  133. #define INJECT_METHOD "PF_PACKET send()"
  134. #else
  135. #define INJECT_METHOD "PF_PACKET / TX_RING"
  136. #endif
  137. #include <fcntl.h>
  138. #include <net/if.h>
  139. #include <net/if_arp.h>
  140. #include <netinet/in.h>
  141. #include <netpacket/packet.h>
  142. #include <sys/utsname.h>
  143. #ifdef HAVE_TX_RING
  144. #include "txring.h"
  145. #endif
  146. static sendpacket_t *sendpacket_open_pf(const char *, char *);
  147. static struct tcpr_ether_addr *sendpacket_get_hwaddr_pf(sendpacket_t *);
  148. static int get_iface_index(int fd, const char *device, char *);
  149. #endif /* HAVE_PF_PACKET */
  150. #ifdef HAVE_TUNTAP
  151. #ifdef HAVE_LINUX
  152. #include <linux/if_tun.h>
  153. #include <net/if.h>
  154. #elif defined(HAVE_FREEBSD)
  155. #define TUNTAP_DEVICE_PREFIX "/dev/"
  156. #endif
  157. static sendpacket_t *sendpacket_open_tuntap(const char *, char *);
  158. #endif
  159. #if defined HAVE_BPF && !defined INJECT_METHOD
  160. #undef INJECT_METHOD
  161. #define INJECT_METHOD "bpf send()"
  162. #include <net/bpf.h>
  163. #include <net/if.h>
  164. #include <net/if_dl.h> // used for get_hwaddr_bpf()
  165. #include <net/route.h>
  166. #include <sys/sysctl.h>
  167. #include <sys/uio.h>
  168. static sendpacket_t *sendpacket_open_bpf(const char *, char *) _U_;
  169. static struct tcpr_ether_addr *sendpacket_get_hwaddr_bpf(sendpacket_t *) _U_;
  170. #endif /* HAVE_BPF */
  171. #if defined HAVE_LIBDNET && !defined INJECT_METHOD
  172. #undef INJECT_METHOD
  173. #define INJECT_METHOD "libdnet eth_send()"
  174. /* need to undef these which are pulled in via defines.h, prior to importing dnet.h */
  175. #undef icmp_id
  176. #undef icmp_seq
  177. #undef icmp_data
  178. #undef icmp_mask
  179. #ifdef HAVE_DNET_H
  180. #include <dnet.h>
  181. #endif
  182. #ifdef HAVE_DUMBNET_H
  183. #include <dumbnet.h>
  184. #endif
  185. static sendpacket_t *sendpacket_open_libdnet(const char *, char *) _U_;
  186. static struct tcpr_ether_addr *sendpacket_get_hwaddr_libdnet(sendpacket_t *) _U_;
  187. #endif /* HAVE_LIBDNET */
  188. #if (defined HAVE_PCAP_INJECT || defined HAVE_PCAP_SENDPACKET) && \
  189. !(defined HAVE_PF_PACKET || defined BPF || defined HAVE_LIBDNET)
  190. static sendpacket_t *sendpacket_open_pcap(const char *, char *) _U_;
  191. static struct tcpr_ether_addr *sendpacket_get_hwaddr_pcap(sendpacket_t *) _U_;
  192. #endif /* HAVE_PCAP_INJECT || HAVE_PACKET_SENDPACKET */
  193. #if defined HAVE_PCAP_INJECT && !defined INJECT_METHOD
  194. #undef INJECT_METHOD
  195. #define INJECT_METHOD "pcap_inject()"
  196. #elif defined HAVE_PCAP_SENDPACKET && !defined INJECT_METHOD
  197. #undef INJECT_METHOD
  198. #define INJECT_METHOD "pcap_sendpacket()"
  199. #endif
  200. #ifdef HAVE_LIBXDP
  201. #include <sys/mman.h>
  202. static sendpacket_t *sendpacket_open_xsk(const char *, char *) _U_;
  203. static struct tcpr_ether_addr *sendpacket_get_hwaddr_libxdp(sendpacket_t *);
  204. #endif
  205. #if defined HAVE_LIBXDP && !defined INJECT_METHOD
  206. #undef INJECT_METHOD
  207. #define INJECT_METHOD "xsk_ring_prod_submit()"
  208. #endif
  209. static sendpacket_t *sendpacket_open_pcap_dump(const char *, char *) _U_;
  210. static void sendpacket_seterr(sendpacket_t *sp, const char *fmt, ...);
  211. static sendpacket_t *sendpacket_open_khial(const char *, char *) _U_;
  212. static struct tcpr_ether_addr *sendpacket_get_hwaddr_khial(sendpacket_t *) _U_;
  213. /**
  214. * returns number of bytes sent on success or -1 on error
  215. * Note: it is theoretically possible to get a return code >0 and < len
  216. * which for most people would be considered an error (the packet wasn't fully sent)
  217. * so you may want to test for recode != len too.
  218. *
  219. * Most socket API's have two interesting errors: ENOBUFS & EAGAIN. ENOBUFS
  220. * is usually due to the kernel buffers being full. EAGAIN happens when you
  221. * try to send traffic faster then the PHY allows.
  222. */
  223. int
  224. sendpacket(sendpacket_t *sp, const u_char *data, size_t len, struct pcap_pkthdr *pkthdr)
  225. {
  226. int retcode = 0, val;
  227. static u_char buffer[10000]; /* 10K bytes, enough for jumbo frames + pkthdr
  228. * larger than page size so made static to
  229. * prevent page misses on stack
  230. */
  231. static const size_t buffer_payload_size = sizeof(buffer) + sizeof(struct pcap_pkthdr);
  232. assert(sp);
  233. #ifndef HAVE_LIBXDP
  234. // In case of XDP packet processing we are storing data in sp->packet_processing->xdp_descs
  235. assert(data);
  236. #endif
  237. if (len == 0)
  238. return -1;
  239. TRY_SEND_AGAIN:
  240. sp->attempt++;
  241. switch (sp->handle_type) {
  242. case SP_TYPE_KHIAL:
  243. memcpy(buffer, pkthdr, sizeof(struct pcap_pkthdr));
  244. memcpy(buffer + sizeof(struct pcap_pkthdr), data, min(len, buffer_payload_size));
  245. /* tell the kernel module which direction the traffic is going */
  246. if (sp->cache_dir == TCPR_DIR_C2S) { /* aka PRIMARY */
  247. val = KHIAL_DIRECTION_RX;
  248. if (ioctl(sp->handle.fd, KHIAL_SET_DIRECTION, (void *)&val) < 0) {
  249. sendpacket_seterr(sp, "Error setting direction on %s: %s (%d)", sp->device, strerror(errno), errno);
  250. return -1;
  251. }
  252. } else if (sp->cache_dir == TCPR_DIR_S2C) {
  253. val = KHIAL_DIRECTION_TX;
  254. if (ioctl(sp->handle.fd, KHIAL_SET_DIRECTION, (void *)&val) < 0) {
  255. sendpacket_seterr(sp, "Error setting direction on %s: %s (%d)", sp->device, strerror(errno), errno);
  256. return -1;
  257. }
  258. }
  259. /* write the pkthdr + packet data all at once */
  260. retcode = (int)write(sp->handle.fd, (void *)buffer, sizeof(struct pcap_pkthdr) + len);
  261. retcode -= sizeof(struct pcap_pkthdr); /* only record packet bytes we sent, not pcap data too */
  262. if (retcode < 0 && !sp->abort) {
  263. switch (errno) {
  264. case EAGAIN:
  265. sp->retry_eagain++;
  266. goto TRY_SEND_AGAIN;
  267. case ENOBUFS:
  268. sp->retry_enobufs++;
  269. goto TRY_SEND_AGAIN;
  270. default:
  271. sendpacket_seterr(sp,
  272. "Error with %s [" COUNTER_SPEC "]: %s (errno = %d)",
  273. "khial",
  274. sp->sent + sp->failed + 1,
  275. strerror(errno),
  276. errno);
  277. }
  278. break;
  279. }
  280. break;
  281. case SP_TYPE_TUNTAP:
  282. retcode = (int)write(sp->handle.fd, (void *)data, len);
  283. break;
  284. /* Linux PF_PACKET and TX_RING */
  285. case SP_TYPE_PF_PACKET:
  286. case SP_TYPE_TX_RING:
  287. #if defined HAVE_PF_PACKET
  288. #ifdef HAVE_TX_RING
  289. retcode = (int)txring_put(sp->tx_ring, data, len);
  290. #else
  291. retcode = (int)send(sp->handle.fd, (void *)data, len, 0);
  292. #endif
  293. /* out of buffers, or hit max PHY speed, silently retry
  294. * as long as we're not told to abort
  295. */
  296. if (retcode < 0 && !sp->abort) {
  297. switch (errno) {
  298. case EAGAIN:
  299. sp->retry_eagain++;
  300. goto TRY_SEND_AGAIN;
  301. case ENOBUFS:
  302. sp->retry_enobufs++;
  303. goto TRY_SEND_AGAIN;
  304. default:
  305. sendpacket_seterr(sp,
  306. "Error with %s [" COUNTER_SPEC "]: %s (errno = %d)",
  307. INJECT_METHOD,
  308. sp->sent + sp->failed + 1,
  309. strerror(errno),
  310. errno);
  311. }
  312. }
  313. #endif /* HAVE_PF_PACKET */
  314. break;
  315. /* BPF */
  316. case SP_TYPE_BPF:
  317. #if defined HAVE_BPF
  318. retcode = write(sp->handle.fd, (void *)data, len);
  319. /* out of buffers, or hit max PHY speed, silently retry */
  320. if (retcode < 0 && !sp->abort) {
  321. switch (errno) {
  322. case EAGAIN:
  323. sp->retry_eagain++;
  324. goto TRY_SEND_AGAIN;
  325. break;
  326. case ENOBUFS:
  327. sp->retry_enobufs++;
  328. goto TRY_SEND_AGAIN;
  329. break;
  330. default:
  331. sendpacket_seterr(sp,
  332. "Error with %s [" COUNTER_SPEC "]: %s (errno = %d)",
  333. INJECT_METHOD,
  334. sp->sent + sp->failed + 1,
  335. strerror(errno),
  336. errno);
  337. }
  338. }
  339. #endif
  340. break;
  341. /* Libdnet */
  342. case SP_TYPE_LIBDNET:
  343. #if defined HAVE_LIBDNET
  344. retcode = eth_send(sp->handle.ldnet, (void *)data, (size_t)len);
  345. /* out of buffers, or hit max PHY speed, silently retry */
  346. if (retcode < 0 && !sp->abort) {
  347. switch (errno) {
  348. case EAGAIN:
  349. sp->retry_eagain++;
  350. goto TRY_SEND_AGAIN;
  351. break;
  352. case ENOBUFS:
  353. sp->retry_enobufs++;
  354. goto TRY_SEND_AGAIN;
  355. break;
  356. default:
  357. sendpacket_seterr(sp,
  358. "Error with %s [" COUNTER_SPEC "]: %s (errno = %d)",
  359. INJECT_METHOD,
  360. sp->sent + sp->failed + 1,
  361. strerror(errno),
  362. errno);
  363. }
  364. }
  365. #endif
  366. break;
  367. case SP_TYPE_LIBPCAP:
  368. #if (defined HAVE_PCAP_INJECT || defined HAVE_PCAP_SENDPACKET)
  369. #if defined HAVE_PCAP_INJECT
  370. /*
  371. * pcap methods don't seem to support ENOBUFS, so we just straight fail
  372. * is there a better way???
  373. */
  374. retcode = pcap_inject(sp->handle.pcap, (void *)data, len);
  375. #elif defined HAVE_PCAP_SENDPACKET
  376. retcode = pcap_sendpacket(sp->handle.pcap, data, (int)len);
  377. #endif
  378. /* out of buffers, or hit max PHY speed, silently retry */
  379. if (retcode < 0 && !sp->abort) {
  380. switch (errno) {
  381. case EAGAIN:
  382. sp->retry_eagain++;
  383. goto TRY_SEND_AGAIN;
  384. case ENOBUFS:
  385. sp->retry_enobufs++;
  386. goto TRY_SEND_AGAIN;
  387. default:
  388. sendpacket_seterr(sp,
  389. "Error with %s [" COUNTER_SPEC "]: %s (errno = %d)",
  390. INJECT_METHOD,
  391. sp->sent + sp->failed + 1,
  392. pcap_geterr(sp->handle.pcap),
  393. errno);
  394. }
  395. }
  396. #if defined HAVE_PCAP_SENDPACKET
  397. /*
  398. * pcap_sendpacket returns 0 on success, not the packet length!
  399. * hence, we have to fix retcode to be more standard on success
  400. */
  401. if (retcode == 0)
  402. retcode = (int)len;
  403. #endif /* HAVE_PCAP_SENDPACKET */
  404. #endif /* HAVE_PCAP_INJECT || HAVE_PCAP_SENDPACKET */
  405. break;
  406. case SP_TYPE_LIBPCAP_DUMP:
  407. pcap_dump((u_char *)sp->handle.dump.dump, pkthdr, data);
  408. retcode = len;
  409. break;
  410. case SP_TYPE_NETMAP:
  411. #ifdef HAVE_NETMAP
  412. retcode = sendpacket_send_netmap(sp, data, len);
  413. if (retcode == -1) {
  414. sendpacket_seterr(sp, "interface hung!!");
  415. } else if (retcode == -2) {
  416. /* this indicates that a retry was requested - this is not a failure */
  417. sp->retry_eagain++;
  418. retcode = 0;
  419. #ifdef HAVE_SCHED_H
  420. /* yield the CPU so other apps remain responsive */
  421. sched_yield();
  422. #endif
  423. goto TRY_SEND_AGAIN;
  424. }
  425. #endif /* HAVE_NETMAP */
  426. break;
  427. case SP_TYPE_LIBXDP:
  428. #ifdef HAVE_LIBXDP
  429. retcode = len;
  430. xsk_ring_prod__submit(&(sp->xsk_info->tx), sp->pckt_count); // submit all packets at once
  431. sp->xsk_info->ring_stats.tx_npkts += sp->pckt_count;
  432. sp->xsk_info->outstanding_tx += sp->pckt_count;
  433. while (sp->xsk_info->outstanding_tx != 0) {
  434. complete_tx_only(sp);
  435. }
  436. sp->sent += sp->pckt_count;
  437. #endif
  438. break;
  439. default:
  440. errx(-1, "Unsupported sp->handle_type = %d", sp->handle_type);
  441. } /* end case */
  442. if (retcode < 0) {
  443. sp->failed++;
  444. } else if (sp->abort) {
  445. sendpacket_seterr(sp, "User abort");
  446. } else if (retcode != (int)len) {
  447. sendpacket_seterr(sp, "Only able to write %d bytes out of %lu bytes total", retcode, len);
  448. sp->trunc_packets++;
  449. } else {
  450. #ifndef HAVE_LIBXDP
  451. sp->bytes_sent += len;
  452. sp->sent++;
  453. #else
  454. if (sp->handle_type != SP_TYPE_LIBXDP) {
  455. sp->bytes_sent += len;
  456. sp->sent++;
  457. }
  458. #endif
  459. }
  460. return retcode;
  461. }
  462. /**
  463. * Open the given network device name and returns a sendpacket_t struct
  464. * pass the error buffer (in case there's a problem) and the direction
  465. * that this interface represents
  466. */
  467. sendpacket_t *
  468. sendpacket_open(const char *device,
  469. char *errbuf,
  470. tcpr_dir_t direction,
  471. sendpacket_type_t sendpacket_type _U_,
  472. void *arg _U_)
  473. {
  474. sendpacket_t *sp;
  475. struct stat sdata;
  476. assert(device);
  477. assert(errbuf);
  478. errbuf[0] = '\0';
  479. if (sendpacket_type == SP_TYPE_LIBPCAP_DUMP) {
  480. sp = sendpacket_open_pcap_dump(device, errbuf);
  481. } else {
  482. /* khial is universal */
  483. if (stat(device, &sdata) == 0) {
  484. if (((sdata.st_mode & S_IFMT) == S_IFCHR)) {
  485. sp = sendpacket_open_khial(device, errbuf);
  486. } else {
  487. switch (sdata.st_mode & S_IFMT) {
  488. case S_IFBLK:
  489. errx(-1, "\"%s\" is a block device and is not a valid Tcpreplay device", device);
  490. case S_IFDIR:
  491. errx(-1, "\"%s\" is a directory and is not a valid Tcpreplay device", device);
  492. case S_IFIFO:
  493. errx(-1, "\"%s\" is a FIFO and is not a valid Tcpreplay device", device);
  494. case S_IFLNK:
  495. errx(-1, "\"%s\" is a symbolic link and is not a valid Tcpreplay device", device);
  496. case S_IFREG:
  497. errx(-1, "\"%s\" is a file and is not a valid Tcpreplay device", device);
  498. default:
  499. errx(-1, "\"%s\" is not a valid Tcpreplay device", device);
  500. }
  501. }
  502. #ifdef HAVE_TUNTAP
  503. } else if (strncmp(device, "tap", 3) == 0) {
  504. sp = sendpacket_open_tuntap(device, errbuf);
  505. #endif
  506. } else {
  507. #ifdef HAVE_NETMAP
  508. if (sendpacket_type == SP_TYPE_NETMAP)
  509. sp = (sendpacket_t *)sendpacket_open_netmap(device, errbuf, arg);
  510. else
  511. #endif
  512. #ifdef HAVE_LIBXDP
  513. if (sendpacket_type == SP_TYPE_LIBXDP)
  514. sp = sendpacket_open_xsk(device, errbuf);
  515. else
  516. #endif
  517. #if defined HAVE_PF_PACKET
  518. sp = sendpacket_open_pf(device, errbuf);
  519. #elif defined HAVE_BPF
  520. sp = sendpacket_open_bpf(device, errbuf);
  521. #elif defined HAVE_LIBDNET
  522. sp = sendpacket_open_libdnet(device, errbuf);
  523. #elif (defined HAVE_PCAP_INJECT || defined HAVE_PCAP_SENDPACKET)
  524. sp = sendpacket_open_pcap(device, errbuf);
  525. #else
  526. #error "No defined packet injection method for sendpacket_open()"
  527. #endif
  528. }
  529. }
  530. if (sp) {
  531. sp->open = 1;
  532. sp->cache_dir = direction;
  533. } else {
  534. errx(-1, "failed to open device %s: %s", device, errbuf);
  535. }
  536. return sp;
  537. }
  538. /**
  539. * Get packet stats for the given sendpacket_t
  540. */
  541. size_t
  542. sendpacket_getstat(sendpacket_t *sp, char *buf, size_t buf_size)
  543. {
  544. size_t offset;
  545. assert(sp);
  546. assert(buf);
  547. memset(buf, 0, buf_size);
  548. offset = snprintf(buf,
  549. buf_size,
  550. "Statistics for network device: %s\n"
  551. "\tSuccessful packets: " COUNTER_SPEC "\n"
  552. "\tFailed packets: " COUNTER_SPEC "\n"
  553. "\tTruncated packets: " COUNTER_SPEC "\n"
  554. "\tRetried packets (ENOBUFS): " COUNTER_SPEC "\n"
  555. "\tRetried packets (EAGAIN): " COUNTER_SPEC "\n",
  556. sp->device,
  557. sp->sent,
  558. sp->failed,
  559. sp->trunc_packets,
  560. sp->retry_enobufs,
  561. sp->retry_eagain);
  562. if (sp->flow_packets && offset > 0) {
  563. offset += snprintf(&buf[offset],
  564. buf_size - offset,
  565. "\tFlows total: " COUNTER_SPEC "\n"
  566. "\tFlows unique: " COUNTER_SPEC "\n"
  567. "\tFlows expired: " COUNTER_SPEC "\n"
  568. "\tFlow packets: " COUNTER_SPEC "\n"
  569. "\tNon-flow packets: " COUNTER_SPEC "\n"
  570. "\tInvalid flow packets: " COUNTER_SPEC "\n",
  571. sp->flows,
  572. sp->flows_expired,
  573. sp->flows_expired,
  574. sp->flow_packets,
  575. sp->flow_non_flow_packets,
  576. sp->flows_invalid_packets);
  577. }
  578. return offset;
  579. }
  580. /**
  581. * close the given sendpacket
  582. */
  583. void
  584. sendpacket_close(sendpacket_t *sp)
  585. {
  586. assert(sp);
  587. switch (sp->handle_type) {
  588. case SP_TYPE_KHIAL:
  589. close(sp->handle.fd);
  590. break;
  591. case SP_TYPE_BPF:
  592. #if (defined HAVE_PCAP_INJECT || defined HAVE_PCAP_SENDPACKET)
  593. close(sp->handle.fd);
  594. #endif
  595. break;
  596. case SP_TYPE_PF_PACKET:
  597. case SP_TYPE_TX_RING:
  598. #ifdef HAVE_PF_PACKET
  599. close(sp->handle.fd);
  600. #endif
  601. break;
  602. case SP_TYPE_LIBPCAP:
  603. #ifdef HAVE_LIBPCAP
  604. pcap_close(sp->handle.pcap);
  605. #endif
  606. break;
  607. case SP_TYPE_LIBPCAP_DUMP:
  608. pcap_dump_close(sp->handle.dump.dump);
  609. pcap_close(sp->handle.dump.pcap);
  610. break;
  611. case SP_TYPE_LIBDNET:
  612. #ifdef HAVE_LIBDNET
  613. eth_close(sp->handle.ldnet);
  614. #endif
  615. break;
  616. case SP_TYPE_LIBNET:
  617. err(-1, "Libnet is no longer supported!");
  618. case SP_TYPE_NETMAP:
  619. #ifdef HAVE_NETMAP
  620. sendpacket_close_netmap(sp);
  621. #endif /* HAVE_NETMAP */
  622. break;
  623. case SP_TYPE_TUNTAP:
  624. #ifdef HAVE_TUNTAP
  625. close(sp->handle.fd);
  626. #endif
  627. break;
  628. case SP_TYPE_LIBXDP:
  629. #ifdef HAVE_LIBXDP
  630. close(sp->handle.fd);
  631. safe_free(sp->xsk_info);
  632. safe_free(sp->umem_info);
  633. #endif
  634. break;
  635. case SP_TYPE_NONE:
  636. err(-1, "no injector selected!");
  637. }
  638. safe_free(sp);
  639. }
  640. /**
  641. * returns the Layer 2 address of the interface current
  642. * open. on error, return NULL
  643. */
  644. struct tcpr_ether_addr *
  645. sendpacket_get_hwaddr(sendpacket_t *sp)
  646. {
  647. struct tcpr_ether_addr *addr;
  648. assert(sp);
  649. /* if we already have our MAC address stored, just return it */
  650. if (memcmp(&sp->ether, "\x00\x00\x00\x00\x00\x00", ETHER_ADDR_LEN) != 0)
  651. return &sp->ether;
  652. if (sp->handle_type == SP_TYPE_KHIAL) {
  653. addr = sendpacket_get_hwaddr_khial(sp);
  654. } else if (sp->handle_type == SP_TYPE_LIBPCAP_DUMP) {
  655. sendpacket_seterr(sp, "Error: sendpacket_get_hwaddr() not yet supported for pcap dump");
  656. return NULL;
  657. } else {
  658. #if defined HAVE_PF_PACKET
  659. addr = sendpacket_get_hwaddr_pf(sp);
  660. #elif defined HAVE_LIBXDP
  661. addr = sendpacket_get_hwaddr_libxdp(sp);
  662. #elif defined HAVE_BPF
  663. addr = sendpacket_get_hwaddr_bpf(sp);
  664. #elif defined HAVE_LIBDNET
  665. addr = sendpacket_get_hwaddr_libdnet(sp);
  666. #elif (defined HAVE_PCAP_INJECT || defined HAVE_PCAP_SENDPACKET)
  667. addr = sendpacket_get_hwaddr_pcap(sp);
  668. #endif
  669. }
  670. return addr;
  671. }
  672. /**
  673. * returns the error string
  674. */
  675. char *
  676. sendpacket_geterr(sendpacket_t *sp)
  677. {
  678. assert(sp);
  679. return sp->errbuf;
  680. }
  681. /**
  682. * Set's the error string
  683. */
  684. static void
  685. sendpacket_seterr(sendpacket_t *sp, const char *fmt, ...)
  686. {
  687. va_list ap;
  688. assert(sp);
  689. va_start(ap, fmt);
  690. if (fmt != NULL)
  691. (void)vsnprintf(sp->errbuf, SENDPACKET_ERRBUF_SIZE, fmt, ap);
  692. va_end(ap);
  693. sp->errbuf[(SENDPACKET_ERRBUF_SIZE - 1)] = '\0'; // be safe
  694. }
  695. #if (defined HAVE_PCAP_INJECT || defined HAVE_PCAP_SENDPACKET) && \
  696. !(defined HAVE_PF_PACKET || defined BPF || defined HAVE_LIBDNET)
  697. /**
  698. * Inner sendpacket_open() method for using libpcap
  699. */
  700. static sendpacket_t *
  701. sendpacket_open_pcap(const char *device, char *errbuf)
  702. {
  703. pcap_t *pcap;
  704. sendpacket_t *sp;
  705. #ifdef BIOCSHDRCMPLT
  706. u_int spoof_eth_src = 1;
  707. int fd;
  708. #endif
  709. assert(device);
  710. assert(errbuf);
  711. dbg(1, "sendpacket: using Libpcap");
  712. /* open_pcap_live automatically fills out our errbuf for us */
  713. if ((pcap = pcap_open_live(device, 0, 0, 0, errbuf)) == NULL)
  714. return NULL;
  715. sp = (sendpacket_t *)safe_malloc(sizeof(sendpacket_t));
  716. strlcpy(sp->device, device, sizeof(sp->device));
  717. sp->handle.pcap = pcap;
  718. #ifdef BIOCSHDRCMPLT
  719. /*
  720. * Only systems using BPF on the backend need this...
  721. * other systems don't have ioctl and will get compile errors.
  722. */
  723. fd = pcap_get_selectable_fd(pcap);
  724. if (ioctl(fd, BIOCSHDRCMPLT, &spoof_eth_src) == -1)
  725. errx(-1, "Unable to enable source MAC spoof support: %s", strerror(errno));
  726. #endif
  727. sp->handle_type = SP_TYPE_LIBPCAP;
  728. return sp;
  729. }
  730. /**
  731. * Get the hardware MAC address for the given interface using libpcap
  732. */
  733. static struct tcpr_ether_addr *
  734. sendpacket_get_hwaddr_pcap(sendpacket_t *sp)
  735. {
  736. assert(sp);
  737. sendpacket_seterr(sp, "Error: sendpacket_get_hwaddr() not yet supported for pcap injection");
  738. return NULL;
  739. }
  740. #endif /* HAVE_PCAP_INJECT || HAVE_PCAP_SENDPACKET */
  741. /**
  742. * Inner sendpacket_open() method for using libpcap
  743. */
  744. static sendpacket_t *
  745. sendpacket_open_pcap_dump(const char *device, char *errbuf)
  746. {
  747. pcap_t *pcap;
  748. pcap_dumper_t* dump;
  749. sendpacket_t *sp;
  750. assert(device);
  751. assert(errbuf);
  752. dbg(1, "sendpacket: using Libpcap");
  753. pcap = pcap_open_dead(DLT_EN10MB, MAX_SNAPLEN);
  754. if ((dump = pcap_dump_open(pcap, device)) == NULL){
  755. char* err_msg = pcap_geterr(pcap);
  756. strlcpy(errbuf, err_msg, PCAP_ERRBUF_SIZE);
  757. pcap_close(pcap);
  758. return NULL;
  759. }
  760. sp = (sendpacket_t *)safe_malloc(sizeof(sendpacket_t));
  761. strlcpy(sp->device, device, sizeof(sp->device));
  762. sp->handle.dump.pcap = pcap;
  763. sp->handle.dump.dump = dump;
  764. sp->handle_type = SP_TYPE_LIBPCAP_DUMP;
  765. return sp;
  766. }
  767. #if defined HAVE_LIBDNET && !defined HAVE_PF_PACKET && !defined HAVE_BPF
  768. /**
  769. * Inner sendpacket_open() method for using libdnet
  770. */
  771. static sendpacket_t *
  772. sendpacket_open_libdnet(const char *device, char *errbuf)
  773. {
  774. eth_t *ldnet;
  775. sendpacket_t *sp;
  776. assert(device);
  777. assert(errbuf);
  778. dbg(1, "sendpacket: using Libdnet");
  779. if ((ldnet = eth_open(device)) == NULL)
  780. return NULL;
  781. sp = (sendpacket_t *)safe_malloc(sizeof(sendpacket_t));
  782. strlcpy(sp->device, device, sizeof(sp->device));
  783. sp->handle.ldnet = ldnet;
  784. sp->handle_type = SP_TYPE_LIBDNET;
  785. return sp;
  786. }
  787. /**
  788. * Get the hardware MAC address for the given interface using libdnet
  789. */
  790. static struct tcpr_ether_addr *
  791. sendpacket_get_hwaddr_libdnet(sendpacket_t *sp)
  792. {
  793. struct tcpr_ether_addr *addr = NULL;
  794. int ret;
  795. assert(sp);
  796. ret = eth_get(sp->handle.ldnet, (eth_addr_t *)addr);
  797. if (addr == NULL || ret < 0) {
  798. sendpacket_seterr(sp, "Error getting hwaddr via libdnet: %s", strerror(errno));
  799. return NULL;
  800. }
  801. memcpy(&sp->ether, addr, sizeof(struct tcpr_ether_addr));
  802. return (&sp->ether);
  803. }
  804. #endif /* HAVE_LIBDNET */
  805. #if defined HAVE_TUNTAP
  806. /**
  807. * Inner sendpacket_open() method for tuntap devices
  808. */
  809. static sendpacket_t *
  810. sendpacket_open_tuntap(const char *device, char *errbuf)
  811. {
  812. sendpacket_t *sp;
  813. struct ifreq ifr;
  814. int tapfd;
  815. assert(device);
  816. assert(errbuf);
  817. #if defined HAVE_LINUX
  818. if ((tapfd = open("/dev/net/tun", O_RDWR)) < 0) {
  819. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "Could not open /dev/net/tun control file: %s", strerror(errno));
  820. return NULL;
  821. }
  822. memset(&ifr, 0, sizeof(ifr));
  823. ifr.ifr_flags = (IFF_TAP | IFF_NO_PI);
  824. strncpy(ifr.ifr_name, device, sizeof(ifr.ifr_name) - 1);
  825. if (ioctl(tapfd, TUNSETIFF, (void *)&ifr) < 0) {
  826. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "Unable to create tuntap interface: %s", device);
  827. close(tapfd);
  828. return NULL;
  829. }
  830. #elif defined(HAVE_FREEBSD)
  831. if (*device == '/') {
  832. if ((tapfd = open(device, O_RDWR)) < 0) {
  833. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "Could not open device %s: %s", device, strerror(errno));
  834. return NULL;
  835. }
  836. } else {
  837. /* full path needed */
  838. char *path;
  839. int prefix_length = strlen(TUNTAP_DEVICE_PREFIX);
  840. if ((path = malloc(strlen(device) + prefix_length + 1)) == NULL) {
  841. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "Malloc error: %s", strerror(errno));
  842. return NULL;
  843. }
  844. snprintf(path, strlen(device) + prefix_length + 1, "%s%s", TUNTAP_DEVICE_PREFIX, device);
  845. if ((tapfd = open(path, O_RDWR)) < 0) {
  846. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "Could not open device %s: %s", path, strerror(errno));
  847. free(path);
  848. return NULL;
  849. }
  850. free(path);
  851. }
  852. #endif
  853. /* prep & return our sp handle */
  854. sp = (sendpacket_t *)safe_malloc(sizeof(sendpacket_t));
  855. strlcpy(sp->device, device, sizeof(sp->device));
  856. sp->handle.fd = tapfd;
  857. sp->handle_type = SP_TYPE_TUNTAP;
  858. return sp;
  859. }
  860. #endif
  861. #if defined HAVE_PF_PACKET
  862. /**
  863. * Inner sendpacket_open() method for using Linux's PF_PACKET or TX_RING
  864. */
  865. static sendpacket_t *
  866. sendpacket_open_pf(const char *device, char *errbuf)
  867. {
  868. int mysocket;
  869. sendpacket_t *sp;
  870. struct ifreq ifr;
  871. struct sockaddr_ll sa;
  872. int err;
  873. socklen_t errlen = sizeof(err);
  874. unsigned int UNUSED(mtu) = 1500;
  875. #ifdef SO_BROADCAST
  876. int n = 1;
  877. #endif
  878. assert(device);
  879. assert(errbuf);
  880. #if defined TX_RING
  881. dbg(1, "sendpacket: using TX_RING");
  882. #else
  883. dbg(1, "sendpacket: using PF_PACKET");
  884. #endif
  885. memset(&sa, 0, sizeof(sa));
  886. /* open our socket */
  887. if ((mysocket = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL))) < 0) {
  888. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "socket: %s", strerror(errno));
  889. return NULL;
  890. }
  891. /* get the interface id for the device */
  892. if ((sa.sll_ifindex = get_iface_index(mysocket, device, errbuf)) < 0) {
  893. close(mysocket);
  894. return NULL;
  895. }
  896. /* bind socket to our interface id */
  897. sa.sll_family = AF_PACKET;
  898. sa.sll_protocol = htons(ETH_P_ALL);
  899. if (bind(mysocket, (struct sockaddr *)&sa, sizeof(sa)) < 0) {
  900. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "bind error: %s", strerror(errno));
  901. close(mysocket);
  902. return NULL;
  903. }
  904. /* check for errors, network down, etc... */
  905. if (getsockopt(mysocket, SOL_SOCKET, SO_ERROR, &err, &errlen) < 0) {
  906. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "error opening %s: %s", device, strerror(errno));
  907. close(mysocket);
  908. return NULL;
  909. }
  910. if (err > 0) {
  911. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "error opening %s: %s", device, strerror(err));
  912. close(mysocket);
  913. return NULL;
  914. }
  915. /* get hardware type for our interface */
  916. memset(&ifr, 0, sizeof(ifr));
  917. strlcpy(ifr.ifr_name, device, sizeof(ifr.ifr_name));
  918. if (ioctl(mysocket, SIOCGIFHWADDR, &ifr) < 0) {
  919. close(mysocket);
  920. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "Error getting hardware type: %s", strerror(errno));
  921. return NULL;
  922. }
  923. /* make sure it's not loopback (PF_PACKET doesn't support it) */
  924. if (ifr.ifr_hwaddr.sa_family != ARPHRD_ETHER)
  925. warnx("Unsupported physical layer type 0x%04x on %s. Maybe it works, maybe it won't."
  926. " See tickets #123/318",
  927. ifr.ifr_hwaddr.sa_family,
  928. device);
  929. #ifdef SO_BROADCAST
  930. /*
  931. * man 7 socket
  932. *
  933. * Set or get the broadcast flag. When enabled, datagram sockets
  934. * receive packets sent to a broadcast address and they are allowed
  935. * to send packets to a broadcast address. This option has no
  936. * effect on stream-oriented sockets.
  937. */
  938. if (setsockopt(mysocket, SOL_SOCKET, SO_BROADCAST, &n, sizeof(n)) == -1) {
  939. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "SO_BROADCAST: %s", strerror(errno));
  940. close(mysocket);
  941. return NULL;
  942. }
  943. #endif /* SO_BROADCAST */
  944. /* prep & return our sp handle */
  945. sp = (sendpacket_t *)safe_malloc(sizeof(sendpacket_t));
  946. strlcpy(sp->device, device, sizeof(sp->device));
  947. sp->handle.fd = mysocket;
  948. #ifdef HAVE_TX_RING
  949. /* Look up for MTU */
  950. memset(&ifr, 0, sizeof(ifr));
  951. strlcpy(ifr.ifr_name, sp->device, sizeof(ifr.ifr_name));
  952. if (ioctl(mysocket, SIOCGIFMTU, &ifr) < 0) {
  953. close(mysocket);
  954. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "Error getting MTU: %s", strerror(errno));
  955. return NULL;
  956. }
  957. mtu = ifr.ifr_ifru.ifru_mtu;
  958. /* Init TX ring for sp->handle.fd socket */
  959. if ((sp->tx_ring = txring_init(sp->handle.fd, mtu)) == 0) {
  960. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "txring_init: %s", strerror(errno));
  961. close(mysocket);
  962. return NULL;
  963. }
  964. sp->handle_type = SP_TYPE_TX_RING;
  965. #else
  966. sp->handle_type = SP_TYPE_PF_PACKET;
  967. #endif
  968. return sp;
  969. }
  970. /**
  971. * get the interface index (necessary for sending packets w/ PF_PACKET)
  972. */
  973. static int
  974. get_iface_index(int fd, const char *device, char *errbuf)
  975. {
  976. struct ifreq ifr;
  977. memset(&ifr, 0, sizeof(ifr));
  978. strlcpy(ifr.ifr_name, (const char *)device, sizeof(ifr.ifr_name));
  979. if (ioctl(fd, SIOCGIFINDEX, &ifr) == -1) {
  980. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "ioctl: %s", strerror(errno));
  981. return (-1);
  982. }
  983. return ifr.ifr_ifindex;
  984. }
  985. /**
  986. * gets the hardware address via Linux's PF packet interface
  987. */
  988. static struct tcpr_ether_addr *
  989. sendpacket_get_hwaddr_pf(sendpacket_t *sp)
  990. {
  991. struct ifreq ifr;
  992. int fd;
  993. assert(sp);
  994. if (!sp->open) {
  995. sendpacket_seterr(sp, "Unable to get hardware address on un-opened sendpacket handle");
  996. return NULL;
  997. }
  998. /* create dummy socket for ioctl */
  999. if ((fd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
  1000. sendpacket_seterr(sp, "Unable to open dummy socket for get_hwaddr: %s", strerror(errno));
  1001. return NULL;
  1002. }
  1003. memset(&ifr, 0, sizeof(ifr));
  1004. strlcpy(ifr.ifr_name, sp->device, sizeof(ifr.ifr_name));
  1005. if (ioctl(fd, SIOCGIFHWADDR, (int8_t *)&ifr) < 0) {
  1006. close(fd);
  1007. sendpacket_seterr(sp, "Error getting hardware address: %s", strerror(errno));
  1008. return NULL;
  1009. }
  1010. memcpy(&sp->ether, &ifr.ifr_hwaddr.sa_data, ETHER_ADDR_LEN);
  1011. close(fd);
  1012. return (&sp->ether);
  1013. }
  1014. #endif /* HAVE_PF_PACKET */
  1015. #if defined HAVE_BPF
  1016. /**
  1017. * Inner sendpacket_open() method for using BSD's BPF interface
  1018. */
  1019. static sendpacket_t *
  1020. sendpacket_open_bpf(const char *device, char *errbuf)
  1021. {
  1022. sendpacket_t *sp;
  1023. char bpf_dev[16];
  1024. int dev, mysocket;
  1025. struct ifreq ifr;
  1026. struct bpf_version bv;
  1027. u_int v;
  1028. #if defined(BIOCGHDRCMPLT) && defined(BIOCSHDRCMPLT)
  1029. u_int spoof_eth_src = 1;
  1030. #endif
  1031. assert(device);
  1032. assert(errbuf);
  1033. memset(&ifr, '\0', sizeof(struct ifreq));
  1034. dbg(1, "sendpacket_open_bpf: using BPF");
  1035. /* open socket */
  1036. mysocket = -1;
  1037. for (dev = 0; dev < 512; dev++) {
  1038. memset(bpf_dev, '\0', sizeof(bpf_dev));
  1039. snprintf(bpf_dev, sizeof(bpf_dev), "/dev/bpf%d", dev);
  1040. dbgx(3, "sendpacket_open_bpf: attempting to open %s", bpf_dev);
  1041. if (!access(bpf_dev, F_OK) && (mysocket = open(bpf_dev, O_RDWR, 0)) > 0) {
  1042. dbg(3, "Success!");
  1043. break;
  1044. }
  1045. dbgx(4, "failed with error %s", strerror(errno));
  1046. }
  1047. /* error?? */
  1048. if (mysocket < 0) {
  1049. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "Unable to open /dev/bpfX: %s", strerror(errno));
  1050. errbuf[SENDPACKET_ERRBUF_SIZE - 1] = '\0';
  1051. return NULL;
  1052. }
  1053. /* get BPF version */
  1054. if (ioctl(mysocket, BIOCVERSION, (caddr_t)&bv) < 0) {
  1055. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "Unable to get bpf version: %s", strerror(errno));
  1056. return NULL;
  1057. }
  1058. if (bv.bv_major != BPF_MAJOR_VERSION || bv.bv_minor != BPF_MINOR_VERSION) {
  1059. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "Kernel's bpf version is out of date.");
  1060. return NULL;
  1061. }
  1062. /* attach to device */
  1063. strlcpy(ifr.ifr_name, device, sizeof(ifr.ifr_name));
  1064. if (ioctl(mysocket, BIOCSETIF, (caddr_t)&ifr) < 0) {
  1065. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "Unable to bind %s to %s: %s", bpf_dev, device, strerror(errno));
  1066. return NULL;
  1067. }
  1068. /* get datalink type */
  1069. if (ioctl(mysocket, BIOCGDLT, (caddr_t)&v) < 0) {
  1070. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "Unable to get datalink type: %s", strerror(errno));
  1071. return NULL;
  1072. }
  1073. /*
  1074. * NetBSD and FreeBSD BPF have an ioctl for enabling/disabling
  1075. * automatic filling of the link level source address.
  1076. */
  1077. #if defined(BIOCGHDRCMPLT) && defined(BIOCSHDRCMPLT)
  1078. if (ioctl(mysocket, BIOCSHDRCMPLT, &spoof_eth_src) == -1) {
  1079. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "Unable to enable spoofing src MAC: %s", strerror(errno));
  1080. return NULL;
  1081. }
  1082. #endif
  1083. /* allocate our sp handle, and return it */
  1084. sp = (sendpacket_t *)safe_malloc(sizeof(sendpacket_t));
  1085. strlcpy(sp->device, device, sizeof(sp->device));
  1086. sp->handle.fd = mysocket;
  1087. sp->handle_type = SP_TYPE_BPF;
  1088. return sp;
  1089. }
  1090. /**
  1091. * Get the interface hardware MAC address when using BPF
  1092. */
  1093. static struct tcpr_ether_addr *
  1094. sendpacket_get_hwaddr_bpf(sendpacket_t *sp)
  1095. {
  1096. int mib[6];
  1097. size_t len;
  1098. int8_t *buf, *next, *end;
  1099. struct if_msghdr *ifm;
  1100. struct sockaddr_dl *sdl;
  1101. assert(sp);
  1102. mib[0] = CTL_NET;
  1103. mib[1] = AF_ROUTE;
  1104. mib[2] = 0;
  1105. mib[3] = AF_LINK;
  1106. mib[4] = NET_RT_IFLIST;
  1107. mib[5] = 0;
  1108. if (sysctl(mib, 6, NULL, &len, NULL, 0) == -1) {
  1109. sendpacket_seterr(sp, "%s(): sysctl(): %s", __func__, strerror(errno));
  1110. return NULL;
  1111. }
  1112. buf = (int8_t *)safe_malloc(len);
  1113. if (sysctl(mib, 6, buf, &len, NULL, 0) == -1) {
  1114. sendpacket_seterr(sp, "%s(): sysctl(): %s", __func__, strerror(errno));
  1115. safe_free(buf);
  1116. return NULL;
  1117. }
  1118. end = buf + len;
  1119. for (next = buf; next < end; next += ifm->ifm_msglen) {
  1120. ifm = (struct if_msghdr *)next;
  1121. if (ifm->ifm_type == RTM_IFINFO) {
  1122. sdl = (struct sockaddr_dl *)(ifm + 1);
  1123. if (strncmp(&sdl->sdl_data[0], sp->device, sdl->sdl_len) == 0) {
  1124. memcpy(&sp->ether, LLADDR(sdl), ETHER_ADDR_LEN);
  1125. break;
  1126. }
  1127. }
  1128. }
  1129. safe_free(buf);
  1130. return (&sp->ether);
  1131. }
  1132. #endif /* HAVE_BPF */
  1133. /**
  1134. * Get the DLT type of the opened sendpacket
  1135. * Return -1 if we can't figure it out, else return the DLT_ value
  1136. */
  1137. int
  1138. sendpacket_get_dlt(sendpacket_t *sp)
  1139. {
  1140. int dlt = DLT_EN10MB;
  1141. switch (sp->handle_type) {
  1142. case SP_TYPE_KHIAL:
  1143. case SP_TYPE_NETMAP:
  1144. case SP_TYPE_TUNTAP:
  1145. case SP_TYPE_LIBXDP:
  1146. case SP_TYPE_LIBPCAP_DUMP:
  1147. /* always EN10MB */
  1148. return dlt;
  1149. default:;
  1150. }
  1151. #if defined HAVE_BPF
  1152. if ((ioctl(sp->handle.fd, BIOCGDLT, &dlt)) < 0) {
  1153. warnx("Unable to get DLT value for BPF device (%s): %s", sp->device, strerror(errno));
  1154. return (-1);
  1155. }
  1156. #elif defined HAVE_PF_PACKET || defined HAVE_LIBDNET
  1157. /* use libpcap to get dlt */
  1158. pcap_t *pcap;
  1159. char errbuf[PCAP_ERRBUF_SIZE];
  1160. if ((pcap = pcap_open_live(sp->device, 65535, 0, 0, errbuf)) == NULL) {
  1161. warnx("Unable to get DLT value for %s: %s", sp->device, errbuf);
  1162. return (-1);
  1163. }
  1164. dlt = pcap_datalink(pcap);
  1165. pcap_close(pcap);
  1166. #elif defined HAVE_PCAP_SENDPACKET || defined HAVE_PCAP_INJECT
  1167. dlt = pcap_datalink(sp->handle.pcap);
  1168. #endif
  1169. return dlt;
  1170. }
  1171. /**
  1172. * \brief Returns a string of the name of the injection method being used
  1173. */
  1174. const char *
  1175. sendpacket_get_method(sendpacket_t *sp)
  1176. {
  1177. if (sp == NULL) {
  1178. return INJECT_METHOD;
  1179. } else if (sp->handle_type == SP_TYPE_KHIAL) {
  1180. return "khial";
  1181. } else if (sp->handle_type == SP_TYPE_NETMAP) {
  1182. return "netmap";
  1183. } else {
  1184. return INJECT_METHOD;
  1185. }
  1186. }
  1187. /**
  1188. * Opens a character device for injecting packets directly into
  1189. * your kernel via a custom driver
  1190. */
  1191. static sendpacket_t *
  1192. sendpacket_open_khial(const char *device, char *errbuf)
  1193. {
  1194. int mysocket;
  1195. sendpacket_t *sp;
  1196. assert(device);
  1197. assert(errbuf);
  1198. if ((mysocket = open(device, O_WRONLY | O_EXCL)) < 0) {
  1199. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "error opening khial device: %s", strerror(errno));
  1200. return NULL;
  1201. }
  1202. sp = (sendpacket_t *)safe_malloc(sizeof(sendpacket_t));
  1203. strlcpy(sp->device, device, sizeof(sp->device));
  1204. sp->handle.fd = mysocket;
  1205. sp->handle_type = SP_TYPE_KHIAL;
  1206. return sp;
  1207. }
  1208. /**
  1209. * Get the hardware MAC address for the given interface using khial
  1210. */
  1211. static struct tcpr_ether_addr *
  1212. sendpacket_get_hwaddr_khial(sendpacket_t *sp)
  1213. {
  1214. assert(sp);
  1215. sendpacket_seterr(sp, "Error: sendpacket_get_hwaddr() not yet supported for character devices");
  1216. return NULL;
  1217. }
  1218. /**
  1219. * \brief Cause the currently running sendpacket() call to stop
  1220. */
  1221. void
  1222. sendpacket_abort(sendpacket_t *sp)
  1223. {
  1224. assert(sp);
  1225. sp->abort = true;
  1226. }
  1227. #ifdef HAVE_LIBXDP
  1228. static struct xsk_socket_info *
  1229. xsk_configure_socket(struct xsk_umem_info *umem, struct xsk_socket_config *cfg, int queue_id, const char *device)
  1230. {
  1231. struct xsk_socket_info *xsk;
  1232. struct xsk_ring_cons *rxr = NULL;
  1233. int ret;
  1234. xsk = (struct xsk_socket_info *)safe_malloc(sizeof(struct xsk_socket_info));
  1235. xsk->umem = umem;
  1236. ret = xsk_socket__create(&xsk->xsk, device, queue_id, umem->umem, rxr, &xsk->tx, cfg);
  1237. if (ret) {
  1238. return NULL;
  1239. }
  1240. memset(&xsk->app_stats, 0, sizeof(xsk->app_stats));
  1241. return xsk;
  1242. }
  1243. static sendpacket_t *
  1244. sendpacket_open_xsk(const char *device, char *errbuf)
  1245. {
  1246. sendpacket_t *sp;
  1247. assert(device);
  1248. assert(errbuf);
  1249. int nb_of_frames = 4096;
  1250. int frame_size = 4096;
  1251. int nb_of_completion_queue_desc = 4096;
  1252. int nb_of_fill_queue_desc = 4096;
  1253. struct xsk_umem_info *umem_info =
  1254. create_umem_area(nb_of_frames, frame_size, nb_of_completion_queue_desc, nb_of_fill_queue_desc);
  1255. if (umem_info == NULL) {
  1256. return NULL;
  1257. }
  1258. int nb_of_tx_queue_desc = 4096;
  1259. int nb_of_rx_queue_desc = 4096;
  1260. u_int32_t queue_id = 0;
  1261. struct xsk_socket_info *xsk_info =
  1262. create_xsk_socket(umem_info, nb_of_tx_queue_desc, nb_of_rx_queue_desc, device, queue_id, errbuf);
  1263. if (xsk_info == NULL) {
  1264. return NULL;
  1265. }
  1266. sp = (sendpacket_t *)safe_malloc(sizeof(sendpacket_t));
  1267. strlcpy(sp->device, device, sizeof(sp->device));
  1268. sp->handle.fd = xsk_info->xsk->fd;
  1269. sp->handle_type = SP_TYPE_LIBXDP;
  1270. sp->xsk_info = xsk_info;
  1271. sp->umem_info = umem_info;
  1272. sp->frame_size = frame_size;
  1273. sp->tx_size = nb_of_tx_queue_desc;
  1274. return sp;
  1275. }
  1276. struct xsk_umem_info *
  1277. create_umem_area(int nb_of_frames, int frame_size, int nb_of_completion_queue_descs, int nb_of_fill_queue_descs)
  1278. {
  1279. int umem_size = nb_of_frames * frame_size;
  1280. struct xsk_umem_info *umem;
  1281. void *umem_area = NULL;
  1282. struct xsk_umem_config cfg = {/* We recommend that you set the fill ring size >= HW RX ring size +
  1283. * AF_XDP RX ring size. Make sure you fill up the fill ring
  1284. * with buffers at regular intervals, and you will with this setting
  1285. * avoid allocation failures in the driver. These are usually quite
  1286. * expensive since drivers have not been written to assume that
  1287. * allocation failures are common. For regular sockets, kernel
  1288. * allocated memory is used that only runs out in OOM situations
  1289. * that should be rare.
  1290. */
  1291. .fill_size = nb_of_fill_queue_descs * 2,
  1292. .comp_size = nb_of_completion_queue_descs,
  1293. .frame_size = frame_size,
  1294. .frame_headroom = 0,
  1295. .flags = XDP_UMEM_UNALIGNED_CHUNK_FLAG};
  1296. umem = (struct xsk_umem_info *)safe_malloc(sizeof(struct xsk_umem_info));
  1297. if (posix_memalign(&umem_area,
  1298. getpagesize(), /* PAGE_SIZE aligned */
  1299. umem_size)) {
  1300. fprintf(stderr, "ERROR: Can't allocate buffer memory \"%s\"\n", strerror(errno));
  1301. exit(EXIT_FAILURE);
  1302. }
  1303. int ret = xsk_umem__create(&umem->umem, umem_area, umem_size, &umem->fq, &umem->cq, &cfg);
  1304. umem->buffer = umem_area;
  1305. if (ret != 0) {
  1306. return NULL;
  1307. }
  1308. return umem;
  1309. }
  1310. struct xsk_socket_info *
  1311. create_xsk_socket(struct xsk_umem_info *umem_info,
  1312. int nb_of_tx_queue_desc,
  1313. int nb_of_rx_queue_desc,
  1314. const char *device,
  1315. u_int32_t queue_id,
  1316. char *errbuf)
  1317. {
  1318. struct xsk_socket_info *xsk_info = (struct xsk_socket_info *)safe_malloc(sizeof(struct xsk_socket_info));
  1319. struct xsk_socket_config *socket_config = (struct xsk_socket_config *)safe_malloc(sizeof(struct xsk_socket_config));
  1320. socket_config->rx_size = nb_of_rx_queue_desc;
  1321. socket_config->tx_size = nb_of_tx_queue_desc;
  1322. socket_config->libbpf_flags = XSK_LIBBPF_FLAGS__INHIBIT_PROG_LOAD;
  1323. socket_config->bind_flags = 0; // XDP_FLAGS_SKB_MODE (1U << 1) or XDP_FLAGS_DRV_MODE (1U << 2)
  1324. xsk_info = xsk_configure_socket(umem_info, socket_config, queue_id, device);
  1325. if (xsk_info == NULL) {
  1326. snprintf(errbuf, SENDPACKET_ERRBUF_SIZE, "AF_XDP socket configuration is not successful: %s", strerror(errno));
  1327. return NULL;
  1328. }
  1329. return xsk_info;
  1330. }
  1331. /*
  1332. * gets the hardware address via Linux's PF packet interface
  1333. */
  1334. static _U_ struct tcpr_ether_addr *
  1335. sendpacket_get_hwaddr_libxdp(sendpacket_t *sp)
  1336. {
  1337. struct ifreq ifr;
  1338. int fd;
  1339. assert(sp);
  1340. if (!sp->open) {
  1341. sendpacket_seterr(sp, "Unable to get hardware address on un-opened sendpacket handle");
  1342. return NULL;
  1343. }
  1344. /* create dummy socket for ioctl */
  1345. if ((fd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
  1346. sendpacket_seterr(sp, "Unable to open dummy socket for get_hwaddr: %s", strerror(errno));
  1347. return NULL;
  1348. }
  1349. memset(&ifr, 0, sizeof(ifr));
  1350. strlcpy(ifr.ifr_name, sp->device, sizeof(ifr.ifr_name));
  1351. if (ioctl(fd, SIOCGIFHWADDR, (int8_t *)&ifr) < 0) {
  1352. close(fd);
  1353. sendpacket_seterr(sp, "Error getting hardware address: %s", strerror(errno));
  1354. return NULL;
  1355. }
  1356. memcpy(&sp->ether, &ifr.ifr_hwaddr.sa_data, ETHER_ADDR_LEN);
  1357. close(fd);
  1358. return (&sp->ether);
  1359. }
  1360. #endif /* HAVE_LIBXDP */