1 // SPDX-License-Identifier: GPL-2.0
2 /*
3 * Copyright 2019 Ramon Fried <rfried.dev@gmail.com>
4 */
5
6 #include <common.h>
7 #include <net.h>
8 #include <net/pcap.h>
9 #include <time.h>
10 #include <asm/io.h>
11
12 #define LINKTYPE_ETHERNET 1
13
14 static bool initialized;
15 static bool running;
16 static bool buffer_full;
17 static void *buf;
18 static unsigned int max_size;
19 static unsigned int pos;
20
21 static unsigned long incoming_count;
22 static unsigned long outgoing_count;
23
24 struct pcap_header {
25 u32 magic;
26 u16 version_major;
27 u16 version_minor;
28 s32 thiszone;
29 u32 sigfigs;
30 u32 snaplen;
31 u32 network;
32 };
33
34 struct pcap_packet_header {
35 u32 ts_sec;
36 u32 ts_usec;
37 u32 incl_len;
38 u32 orig_len;
39 };
40
41 static struct pcap_header file_header = {
42 .magic = 0xa1b2c3d4,
43 .version_major = 2,
44 .version_minor = 4,
45 .snaplen = 65535,
46 .network = LINKTYPE_ETHERNET,
47 };
48
pcap_init(phys_addr_t paddr,unsigned long size)49 int pcap_init(phys_addr_t paddr, unsigned long size)
50 {
51 buf = map_physmem(paddr, size, 0);
52 if (!buf) {
53 printf("Failed mapping PCAP memory\n");
54 return -ENOMEM;
55 }
56
57 printf("PCAP capture initialized: addr: 0x%lx max length: %lu\n",
58 (unsigned long)buf, size);
59
60 memcpy(buf, &file_header, sizeof(file_header));
61 pos = sizeof(file_header);
62 max_size = size;
63 initialized = true;
64 running = false;
65 buffer_full = false;
66 incoming_count = 0;
67 outgoing_count = 0;
68 return 0;
69 }
70
pcap_start_stop(bool start)71 int pcap_start_stop(bool start)
72 {
73 if (!initialized) {
74 printf("error: pcap was not initialized\n");
75 return -ENODEV;
76 }
77
78 running = start;
79
80 return 0;
81 }
82
pcap_clear(void)83 int pcap_clear(void)
84 {
85 if (!initialized) {
86 printf("error: pcap was not initialized\n");
87 return -ENODEV;
88 }
89
90 pos = sizeof(file_header);
91 incoming_count = 0;
92 outgoing_count = 0;
93 buffer_full = false;
94
95 printf("pcap capture cleared\n");
96 return 0;
97 }
98
pcap_post(const void * packet,size_t len,bool outgoing)99 int pcap_post(const void *packet, size_t len, bool outgoing)
100 {
101 struct pcap_packet_header header;
102 u64 cur_time = timer_get_us();
103
104 if (!initialized || !running || !buf)
105 return -ENODEV;
106
107 if (buffer_full)
108 return -ENOMEM;
109
110 if ((pos + len + sizeof(header)) >= max_size) {
111 buffer_full = true;
112 printf("\n!!! Buffer is full, consider increasing buffer size !!!\n");
113 return -ENOMEM;
114 }
115
116 header.ts_sec = cur_time / 1000000;
117 header.ts_usec = cur_time % 1000000;
118 header.incl_len = len;
119 header.orig_len = len;
120
121 memcpy(buf + pos, &header, sizeof(header));
122 pos += sizeof(header);
123 memcpy(buf + pos, packet, len);
124 pos += len;
125
126 if (outgoing)
127 outgoing_count++;
128 else
129 incoming_count++;
130
131 env_set_hex("pcapsize", pos);
132
133 return 0;
134 }
135
pcap_print_status(void)136 int pcap_print_status(void)
137 {
138 if (!initialized) {
139 printf("pcap was not initialized\n");
140 return -ENODEV;
141 }
142 printf("PCAP status:\n");
143 printf("\tInitialized addr: 0x%lx\tmax length: %u\n",
144 (unsigned long)buf, max_size);
145 printf("\tStatus: %s.\t file size: %u\n", running ? "Active" : "Idle",
146 pos);
147 printf("\tIncoming packets: %lu Outgoing packets: %lu\n",
148 incoming_count, outgoing_count);
149
150 return 0;
151 }
152
pcap_active(void)153 bool pcap_active(void)
154 {
155 return running;
156 }
157