1use std::net::{Ipv4Addr, Ipv6Addr};
7
8#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
12pub struct MacAddress(pub [u8; 6]);
13
14impl MacAddress {
15 pub const fn new(bytes: [u8; 6]) -> Self {
17 Self(bytes)
18 }
19
20 pub const fn as_bytes(&self) -> &[u8; 6] {
22 &self.0
23 }
24
25 pub fn is_broadcast(&self) -> bool {
27 self.0 == [0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF]
28 }
29
30 pub fn is_multicast(&self) -> bool {
32 self.0[0] & 0x01 != 0
33 }
34
35 pub fn is_unicast(&self) -> bool {
37 !self.is_multicast()
38 }
39}
40
41impl std::fmt::Display for MacAddress {
42 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
43 write!(
44 f,
45 "{:02x}:{:02x}:{:02x}:{:02x}:{:02x}:{:02x}",
46 self.0[0], self.0[1], self.0[2], self.0[3], self.0[4], self.0[5]
47 )
48 }
49}
50
51impl From<[u8; 6]> for MacAddress {
52 fn from(bytes: [u8; 6]) -> Self {
53 Self(bytes)
54 }
55}
56
57impl From<MacAddress> for [u8; 6] {
58 fn from(mac: MacAddress) -> Self {
59 mac.0
60 }
61}
62
63#[derive(Debug, Clone, Copy, PartialEq, Eq)]
65pub enum DatagramVersion {
66 Version5 = 5,
67}
68
69#[derive(Debug, Clone, PartialEq, Eq)]
71pub enum Address {
72 Unknown,
73 IPv4(Ipv4Addr),
74 IPv6(Ipv6Addr),
75}
76
77#[derive(Debug, Clone, Copy, PartialEq, Eq)]
80pub struct DataFormat(pub u32);
81
82impl DataFormat {
83 pub fn new(enterprise: u32, format: u32) -> Self {
84 Self((enterprise << 12) | (format & 0xFFF))
85 }
86
87 pub fn enterprise(&self) -> u32 {
88 self.0 >> 12
89 }
90
91 pub fn format(&self) -> u32 {
92 self.0 & 0xFFF
93 }
94}
95
96#[derive(Debug, Clone, Copy, PartialEq, Eq)]
98pub struct DataSource(pub u32);
99
100impl DataSource {
101 pub fn new(source_type: u8, index: u32) -> Self {
102 Self(((source_type as u32) << 24) | (index & 0xFFFFFF))
103 }
104
105 pub fn source_type(&self) -> u8 {
106 (self.0 >> 24) as u8
107 }
108
109 pub fn index(&self) -> u32 {
110 self.0 & 0xFFFFFF
111 }
112}
113
114#[derive(Debug, Clone, Copy, PartialEq, Eq)]
116pub struct DataSourceExpanded {
117 pub source_id_type: u32,
118 pub source_id_index: u32,
119}
120
121#[derive(Debug, Clone, Copy, PartialEq, Eq)]
123pub struct Interface(pub u32);
124
125impl Interface {
126 pub fn format(&self) -> u8 {
127 (self.0 >> 30) as u8
128 }
129
130 pub fn value(&self) -> u32 {
131 self.0 & 0x3FFFFFFF
132 }
133
134 pub fn is_single(&self) -> bool {
135 self.format() == 0
136 }
137
138 pub fn is_discarded(&self) -> bool {
139 self.format() == 1
140 }
141
142 pub fn is_multiple(&self) -> bool {
143 self.format() == 2
144 }
145}
146
147#[derive(Debug, Clone, Copy, PartialEq, Eq)]
149pub struct InterfaceExpanded {
150 pub format: u32,
151 pub value: u32,
152}
153
154#[derive(Debug, Clone, PartialEq, Eq)]
156pub enum FlowData {
157 SampledHeader(crate::models::record_flows::SampledHeader),
159 SampledEthernet(crate::models::record_flows::SampledEthernet),
161 SampledIpv4(crate::models::record_flows::SampledIpv4),
163 SampledIpv6(crate::models::record_flows::SampledIpv6),
165 ExtendedSwitch(crate::models::record_flows::ExtendedSwitch),
167 ExtendedRouter(crate::models::record_flows::ExtendedRouter),
169 ExtendedGateway(crate::models::record_flows::ExtendedGateway),
171 ExtendedUser(crate::models::record_flows::ExtendedUser),
173 ExtendedUrl(crate::models::record_flows::ExtendedUrl),
176 ExtendedMpls(crate::models::record_flows::ExtendedMpls),
178 ExtendedNat(crate::models::record_flows::ExtendedNat),
180 ExtendedMplsTunnel(crate::models::record_flows::ExtendedMplsTunnel),
182 ExtendedMplsVc(crate::models::record_flows::ExtendedMplsVc),
184 ExtendedMplsFec(crate::models::record_flows::ExtendedMplsFec),
186 ExtendedMplsLvpFec(crate::models::record_flows::ExtendedMplsLvpFec),
188 ExtendedVlanTunnel(crate::models::record_flows::ExtendedVlanTunnel),
190 Extended80211Payload(crate::models::record_flows::Extended80211Payload),
192 Extended80211Rx(crate::models::record_flows::Extended80211Rx),
194 Extended80211Tx(crate::models::record_flows::Extended80211Tx),
196 Extended80211Aggregation(crate::models::record_flows::Extended80211Aggregation),
198 ExtendedOpenFlowV1(crate::models::record_flows::ExtendedOpenFlowV1),
200 ExtendedL2TunnelEgress(crate::models::record_flows::ExtendedL2TunnelEgress),
202 ExtendedL2TunnelIngress(crate::models::record_flows::ExtendedL2TunnelIngress),
204 ExtendedIpv4TunnelEgress(crate::models::record_flows::ExtendedIpv4TunnelEgress),
206 ExtendedIpv4TunnelIngress(crate::models::record_flows::ExtendedIpv4TunnelIngress),
208 ExtendedIpv6TunnelEgress(crate::models::record_flows::ExtendedIpv6TunnelEgress),
210 ExtendedIpv6TunnelIngress(crate::models::record_flows::ExtendedIpv6TunnelIngress),
212 ExtendedDecapsulateEgress(crate::models::record_flows::ExtendedDecapsulateEgress),
214 ExtendedDecapsulateIngress(crate::models::record_flows::ExtendedDecapsulateIngress),
216 ExtendedVniEgress(crate::models::record_flows::ExtendedVniEgress),
218 ExtendedVniIngress(crate::models::record_flows::ExtendedVniIngress),
220 ExtendedSocketIpv4(crate::models::record_flows::ExtendedSocketIpv4),
222 ExtendedSocketIpv6(crate::models::record_flows::ExtendedSocketIpv6),
224 AppOperation(crate::models::record_flows::AppOperation),
226 AppParentContext(crate::models::record_flows::AppParentContext),
228 AppInitiator(crate::models::record_flows::AppInitiator),
230 AppTarget(crate::models::record_flows::AppTarget),
232 Unknown { format: DataFormat, data: Vec<u8> },
234}
235
236#[derive(Debug, Clone, PartialEq, Eq)]
238pub struct FlowRecord {
239 pub flow_format: DataFormat,
240 pub flow_data: FlowData,
241}
242
243#[derive(Debug, Clone, PartialEq)]
245pub enum CounterData {
246 GenericInterface(crate::models::record_counters::GenericInterfaceCounters),
248 EthernetInterface(crate::models::record_counters::EthernetInterfaceCounters),
250 TokenRing(crate::models::record_counters::TokenRingCounters),
252 Vg100Interface(crate::models::record_counters::Vg100InterfaceCounters),
254 Vlan(crate::models::record_counters::VlanCounters),
256 Ieee80211(crate::models::record_counters::Ieee80211Counters),
258 Processor(crate::models::record_counters::ProcessorCounters),
260 RadioUtilization(crate::models::record_counters::RadioUtilization),
262 HostDescription(crate::models::record_counters::HostDescription),
264 HostAdapters(crate::models::record_counters::HostAdapters),
266 HostParent(crate::models::record_counters::HostParent),
268 HostCpu(crate::models::record_counters::HostCpu),
270 HostMemory(crate::models::record_counters::HostMemory),
272 HostDiskIo(crate::models::record_counters::HostDiskIo),
274 HostNetIo(crate::models::record_counters::HostNetIo),
276 VirtualNode(crate::models::record_counters::VirtualNode),
278 VirtualCpu(crate::models::record_counters::VirtualCpu),
280 VirtualMemory(crate::models::record_counters::VirtualMemory),
282 VirtualDiskIo(crate::models::record_counters::VirtualDiskIo),
284 VirtualNetIo(crate::models::record_counters::VirtualNetIo),
286 OpenFlowPort(crate::models::record_counters::OpenFlowPort),
288 OpenFlowPortName(crate::models::record_counters::OpenFlowPortName),
290 AppOperations(crate::models::record_counters::AppOperations),
292 AppResources(crate::models::record_counters::AppResources),
294 AppWorkers(crate::models::record_counters::AppWorkers),
296 Unknown { format: DataFormat, data: Vec<u8> },
298}
299
300#[derive(Debug, Clone, PartialEq)]
302pub struct CounterRecord {
303 pub counter_format: DataFormat,
304 pub counter_data: CounterData,
305}
306
307#[derive(Debug, Clone, PartialEq, Eq)]
309pub struct FlowSample {
310 pub sequence_number: u32,
311 pub source_id: DataSource,
312 pub sampling_rate: u32,
313 pub sample_pool: u32,
314 pub drops: u32,
315 pub input: Interface,
316 pub output: Interface,
317 pub flow_records: Vec<FlowRecord>,
318}
319
320#[derive(Debug, Clone, PartialEq)]
322pub struct CountersSample {
323 pub sequence_number: u32,
324 pub source_id: DataSource,
325 pub counters: Vec<CounterRecord>,
326}
327
328#[derive(Debug, Clone, PartialEq, Eq)]
330pub struct FlowSampleExpanded {
331 pub sequence_number: u32,
332 pub source_id: DataSourceExpanded,
333 pub sampling_rate: u32,
334 pub sample_pool: u32,
335 pub drops: u32,
336 pub input: InterfaceExpanded,
337 pub output: InterfaceExpanded,
338 pub flow_records: Vec<FlowRecord>,
339}
340
341#[derive(Debug, Clone, PartialEq)]
343pub struct CountersSampleExpanded {
344 pub sequence_number: u32,
345 pub source_id: DataSourceExpanded,
346 pub counters: Vec<CounterRecord>,
347}
348
349#[derive(Debug, Clone, PartialEq)]
351pub enum SampleData {
352 FlowSample(FlowSample),
353 CountersSample(CountersSample),
354 FlowSampleExpanded(FlowSampleExpanded),
355 CountersSampleExpanded(CountersSampleExpanded),
356 Unknown { format: DataFormat, data: Vec<u8> },
357}
358
359#[derive(Debug, Clone, PartialEq)]
361pub struct SampleRecord {
362 pub sample_type: DataFormat,
363 pub sample_data: SampleData,
364}
365
366#[derive(Debug, Clone, PartialEq)]
368pub struct SFlowDatagram {
369 pub version: DatagramVersion,
370 pub agent_address: Address,
371 pub sub_agent_id: u32,
372 pub sequence_number: u32,
373 pub uptime: u32,
374 pub samples: Vec<SampleRecord>,
375}
376
377impl SFlowDatagram {
378 pub fn new(
380 agent_address: Address,
381 sub_agent_id: u32,
382 sequence_number: u32,
383 uptime: u32,
384 ) -> Self {
385 Self {
386 version: DatagramVersion::Version5,
387 agent_address,
388 sub_agent_id,
389 sequence_number,
390 uptime,
391 samples: Vec::new(),
392 }
393 }
394}