mirror of
https://github.com/espressif/esp-idf.git
synced 2024-10-05 20:47:46 -04:00
pcap: refactor and move to common components
This commit is contained in:
parent
03a89f1c39
commit
f04f078825
2
examples/common_components/pcap/CMakeLists.txt
Normal file
2
examples/common_components/pcap/CMakeLists.txt
Normal file
@ -0,0 +1,2 @@
|
|||||||
|
idf_component_register(SRCS "src/pcap.c"
|
||||||
|
INCLUDE_DIRS "include")
|
5
examples/common_components/pcap/component.mk
Normal file
5
examples/common_components/pcap/component.mk
Normal file
@ -0,0 +1,5 @@
|
|||||||
|
#
|
||||||
|
# Component Makefile
|
||||||
|
#
|
||||||
|
COMPONENT_ADD_INCLUDEDIRS := include
|
||||||
|
COMPONENT_SRCDIRS := src
|
128
examples/common_components/pcap/include/pcap.h
Normal file
128
examples/common_components/pcap/include/pcap.h
Normal file
@ -0,0 +1,128 @@
|
|||||||
|
/*
|
||||||
|
* SPDX-FileCopyrightText: 2015-2021 Espressif Systems (Shanghai) CO LTD
|
||||||
|
*
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <stdio.h>
|
||||||
|
#include "esp_err.h"
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
extern "C" {
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#define PCAP_DEFAULT_VERSION_MAJOR 0x02 /*!< Major Version */
|
||||||
|
#define PCAP_DEFAULT_VERSION_MINOR 0x04 /*!< Minor Version */
|
||||||
|
#define PCAP_DEFAULT_TIME_ZONE_GMT 0x00 /*!< Time Zone */
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Type of pcap file handle
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
typedef struct pcap_file_t *pcap_file_handle_t;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Link layer Type Definition, used for Pcap reader to decode payload
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
typedef enum {
|
||||||
|
PCAP_LINK_TYPE_LOOPBACK = 0, /*!< Loopback devices, except for later OpenBSD */
|
||||||
|
PCAP_LINK_TYPE_ETHERNET = 1, /*!< Ethernet, and Linux loopback devices */
|
||||||
|
PCAP_LINK_TYPE_TOKEN_RING = 6, /*!< 802.5 Token Ring */
|
||||||
|
PCAP_LINK_TYPE_ARCNET = 7, /*!< ARCnet */
|
||||||
|
PCAP_LINK_TYPE_SLIP = 8, /*!< SLIP */
|
||||||
|
PCAP_LINK_TYPE_PPP = 9, /*!< PPP */
|
||||||
|
PCAP_LINK_TYPE_FDDI = 10, /*!< FDDI */
|
||||||
|
PCAP_LINK_TYPE_ATM = 100, /*!< LLC/SNAP encapsulated ATM */
|
||||||
|
PCAP_LINK_TYPE_RAW_IP = 101, /*!< Raw IP, without link */
|
||||||
|
PCAP_LINK_TYPE_BSD_SLIP = 102, /*!< BSD/OS SLIP */
|
||||||
|
PCAP_LINK_TYPE_BSD_PPP = 103, /*!< BSD/OS PPP */
|
||||||
|
PCAP_LINK_TYPE_CISCO_HDLC = 104, /*!< Cisco HDLC */
|
||||||
|
PCAP_LINK_TYPE_802_11 = 105, /*!< 802.11 */
|
||||||
|
PCAP_LINK_TYPE_BSD_LOOPBACK = 108, /*!< OpenBSD loopback devices(with AF_value in network byte order) */
|
||||||
|
PCAP_LINK_TYPE_LOCAL_TALK = 114 /*!< LocalTalk */
|
||||||
|
} pcap_link_type_t;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Pcap configuration Type Definition
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
typedef struct {
|
||||||
|
FILE *fp; /*!< Pointer to a standard file handle */
|
||||||
|
unsigned int major_version; /*!< Pcap version: major */
|
||||||
|
unsigned int minor_version; /*!< Pcap version: minor */
|
||||||
|
unsigned int time_zone; /*!< Pcap timezone code */
|
||||||
|
struct {
|
||||||
|
unsigned int little_endian: 1; /*!< Whether the pcap file is recored in little endian format */
|
||||||
|
} flags;
|
||||||
|
} pcap_config_t;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Create a new pcap session, and returns pcap file handle
|
||||||
|
*
|
||||||
|
* @note This function won't create the low level FILE* object, the user should take care of the creation of the File Stream.
|
||||||
|
*
|
||||||
|
* @param[in] config pcap file configuration
|
||||||
|
* @param[out] ret_pcap Returned pcap file handle
|
||||||
|
* @return
|
||||||
|
* - ESP_OK: Create pcap file successfully
|
||||||
|
* - ESP_ERR_INVALID_ARG: Create pcap file failed because of invalid argument
|
||||||
|
* - ESP_ERR_NO_MEM: Create pcap file failed because out of memory
|
||||||
|
* - ESP_FAIL: Create pcap file failed
|
||||||
|
*/
|
||||||
|
esp_err_t pcap_new_session(const pcap_config_t *config, pcap_file_handle_t *ret_pcap);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Delete the pcap session, and close the File Stream
|
||||||
|
*
|
||||||
|
* @param[in] pcap pcap file handle created by `pcap_new_session()`
|
||||||
|
* @return
|
||||||
|
* - ESP_OK: Delete pcap session successfully
|
||||||
|
* - ESP_ERR_INVALID_ARG: Delete pcap session failed because of invalid argument
|
||||||
|
* - ESP_FAIL: Delete pcap session failed
|
||||||
|
*/
|
||||||
|
esp_err_t pcap_del_session(pcap_file_handle_t pcap);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Write pcap file header
|
||||||
|
*
|
||||||
|
* @param[in] pcap pcap file handle created by `pcap_new_session()`
|
||||||
|
* @param[in] link_type Network link layer type
|
||||||
|
* @return
|
||||||
|
* - ESP_OK: Write pcap file header successfully
|
||||||
|
* - ESP_ERR_INVALID_ARG: Write pcap file header failed because of invalid argument
|
||||||
|
* - ESP_FAIL: Write pcap file header failed
|
||||||
|
*/
|
||||||
|
esp_err_t pcap_write_header(pcap_file_handle_t pcap, pcap_link_type_t link_type);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Capture one packet into pcap file
|
||||||
|
*
|
||||||
|
* @param[in] pcap pcap file handle created by `pcap_new_session()`
|
||||||
|
* @param[in] payload pointer of the captured data buffer
|
||||||
|
* @param[in] length length of captured data buffer
|
||||||
|
* @param[in] seconds second of capture time
|
||||||
|
* @param[in] microseconds microsecond of capture time
|
||||||
|
* @return
|
||||||
|
* - ESP_OK: Write network packet into pcap file successfully
|
||||||
|
* - ESP_ERR_INVALID_ARG: Write network packet into pcap file failed because of invalid argument
|
||||||
|
* - ESP_FAIL: Write network packet into pcap file failed
|
||||||
|
*/
|
||||||
|
esp_err_t pcap_capture_packet(pcap_file_handle_t pcap, void *payload, uint32_t length, uint32_t seconds, uint32_t microseconds);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Print the summary of pcap file into stream
|
||||||
|
*
|
||||||
|
* @param[in] pcap pcap file handle created by `pcap_new_session()`
|
||||||
|
* @param[in] print_file the file stream to save the summary
|
||||||
|
* @return
|
||||||
|
* - ESP_OK: Print pcap file summary successfully
|
||||||
|
* - ESP_ERR_INVALID_ARG: Print pcap file summary failed because of invalid argument
|
||||||
|
* - ESP_FAIL: Print pcap file summary failed
|
||||||
|
*/
|
||||||
|
esp_err_t pcap_print_summary(pcap_file_handle_t pcap, FILE *print_file);
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
}
|
||||||
|
#endif
|
224
examples/common_components/pcap/src/pcap.c
Normal file
224
examples/common_components/pcap/src/pcap.c
Normal file
@ -0,0 +1,224 @@
|
|||||||
|
/*
|
||||||
|
* SPDX-FileCopyrightText: 2015-2021 Espressif Systems (Shanghai) CO LTD
|
||||||
|
*
|
||||||
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include "esp_log.h"
|
||||||
|
#include "esp_check.h"
|
||||||
|
#include "pcap.h"
|
||||||
|
|
||||||
|
static const char *TAG = "pcap";
|
||||||
|
|
||||||
|
#define PCAP_MAGIC_BIG_ENDIAN 0xA1B2C3D4 /*!< Big-Endian */
|
||||||
|
#define PCAP_MAGIC_LITTLE_ENDIAN 0xD4C3B2A1 /*!< Little-Endian */
|
||||||
|
|
||||||
|
typedef struct pcap_file_t pcap_file_t;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Pcap File Header
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
typedef struct {
|
||||||
|
uint32_t magic; /*!< Magic Number */
|
||||||
|
uint16_t major; /*!< Major Version */
|
||||||
|
uint16_t minor; /*!< Minor Version */
|
||||||
|
uint32_t zone; /*!< Time Zone Offset */
|
||||||
|
uint32_t sigfigs; /*!< Timestamp Accuracy */
|
||||||
|
uint32_t snaplen; /*!< Max Length to Capture */
|
||||||
|
uint32_t link_type; /*!< Link Layer Type */
|
||||||
|
} pcap_file_header_t;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Pcap Packet Header
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
typedef struct {
|
||||||
|
uint32_t seconds; /*!< Number of seconds since January 1st, 1970, 00:00:00 GMT */
|
||||||
|
uint32_t microseconds; /*!< Number of microseconds when the packet was captured (offset from seconds) */
|
||||||
|
uint32_t capture_length; /*!< Number of bytes of captured data, no longer than packet_length */
|
||||||
|
uint32_t packet_length; /*!< Actual length of current packet */
|
||||||
|
} pcap_packet_header_t;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Pcap Runtime Handle
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
struct pcap_file_t {
|
||||||
|
FILE *file; /*!< File handle */
|
||||||
|
pcap_link_type_t link_type; /*!< Pcap Link Type */
|
||||||
|
unsigned int major_version; /*!< Pcap version: major */
|
||||||
|
unsigned int minor_version; /*!< Pcap version: minor */
|
||||||
|
unsigned int time_zone; /*!< Pcap timezone code */
|
||||||
|
uint32_t endian_magic; /*!< Magic value related to endian format */
|
||||||
|
};
|
||||||
|
|
||||||
|
esp_err_t pcap_new_session(const pcap_config_t *config, pcap_file_handle_t *ret_pcap)
|
||||||
|
{
|
||||||
|
esp_err_t ret = ESP_OK;
|
||||||
|
pcap_file_t *pcap = NULL;
|
||||||
|
ESP_GOTO_ON_FALSE(config && ret_pcap, ESP_ERR_INVALID_ARG, err, TAG, "invalid argument");
|
||||||
|
ESP_GOTO_ON_FALSE(config->fp, ESP_ERR_INVALID_ARG, err, TAG, "pcap file handle can't be NULL");
|
||||||
|
pcap = calloc(1, sizeof(pcap_file_t));
|
||||||
|
ESP_GOTO_ON_FALSE(pcap, ESP_ERR_NO_MEM, err, TAG, "no mem for pcap file object");
|
||||||
|
pcap->file = config->fp;
|
||||||
|
pcap->major_version = config->major_version;
|
||||||
|
pcap->minor_version = config->minor_version;
|
||||||
|
pcap->endian_magic = config->flags.little_endian ? PCAP_MAGIC_LITTLE_ENDIAN : PCAP_MAGIC_BIG_ENDIAN;
|
||||||
|
pcap->time_zone = config->time_zone;
|
||||||
|
*ret_pcap = pcap;
|
||||||
|
return ret;
|
||||||
|
err:
|
||||||
|
if (pcap) {
|
||||||
|
free(pcap);
|
||||||
|
}
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
esp_err_t pcap_del_session(pcap_file_handle_t pcap)
|
||||||
|
{
|
||||||
|
ESP_RETURN_ON_FALSE(pcap, ESP_ERR_INVALID_ARG, TAG, "invalid argument");
|
||||||
|
if (pcap->file) {
|
||||||
|
fclose(pcap->file);
|
||||||
|
pcap->file = NULL;
|
||||||
|
}
|
||||||
|
free(pcap);
|
||||||
|
return ESP_OK;
|
||||||
|
}
|
||||||
|
|
||||||
|
esp_err_t pcap_write_header(pcap_file_handle_t pcap, pcap_link_type_t link_type)
|
||||||
|
{
|
||||||
|
ESP_RETURN_ON_FALSE(pcap, ESP_ERR_INVALID_ARG, TAG, "invalid argument");
|
||||||
|
/* Write Pcap File header */
|
||||||
|
pcap_file_header_t header = {
|
||||||
|
.magic = pcap->endian_magic,
|
||||||
|
.major = pcap->major_version,
|
||||||
|
.minor = pcap->minor_version,
|
||||||
|
.zone = pcap->time_zone,
|
||||||
|
.sigfigs = 0,
|
||||||
|
.snaplen = 0x40000,
|
||||||
|
.link_type = link_type,
|
||||||
|
};
|
||||||
|
size_t real_write = fwrite(&header, sizeof(header), 1, pcap->file);
|
||||||
|
ESP_RETURN_ON_FALSE(real_write == 1, ESP_FAIL, TAG, "write pcap file header failed");
|
||||||
|
/* Save the link type to pcap file object */
|
||||||
|
pcap->link_type = link_type;
|
||||||
|
/* Flush content in the buffer into device */
|
||||||
|
fflush(pcap->file);
|
||||||
|
return ESP_OK;
|
||||||
|
}
|
||||||
|
|
||||||
|
esp_err_t pcap_capture_packet(pcap_file_handle_t pcap, void *payload, uint32_t length, uint32_t seconds, uint32_t microseconds)
|
||||||
|
{
|
||||||
|
ESP_RETURN_ON_FALSE(pcap && payload, ESP_ERR_INVALID_ARG, TAG, "invalid argumnet");
|
||||||
|
size_t real_write = 0;
|
||||||
|
pcap_packet_header_t header = {
|
||||||
|
.seconds = seconds,
|
||||||
|
.microseconds = microseconds,
|
||||||
|
.capture_length = length,
|
||||||
|
.packet_length = length
|
||||||
|
};
|
||||||
|
real_write = fwrite(&header, sizeof(header), 1, pcap->file);
|
||||||
|
ESP_RETURN_ON_FALSE(real_write == 1, ESP_FAIL, TAG, "write packet header failed");
|
||||||
|
real_write = fwrite(payload, sizeof(uint8_t), length, pcap->file);
|
||||||
|
ESP_RETURN_ON_FALSE(real_write == length, ESP_FAIL, TAG, "write packet payload failed");
|
||||||
|
/* Flush content in the buffer into device */
|
||||||
|
fflush(pcap->file);
|
||||||
|
return ESP_OK;
|
||||||
|
}
|
||||||
|
|
||||||
|
esp_err_t pcap_print_summary(pcap_file_handle_t pcap, FILE *print_file)
|
||||||
|
{
|
||||||
|
esp_err_t ret = ESP_OK;
|
||||||
|
long size = 0;
|
||||||
|
char *packet_payload = NULL;
|
||||||
|
ESP_RETURN_ON_FALSE(pcap && print_file, ESP_ERR_INVALID_ARG, TAG, "invalid argument");
|
||||||
|
// get file size
|
||||||
|
fseek(pcap->file, 0L, SEEK_END);
|
||||||
|
size = ftell(pcap->file);
|
||||||
|
fseek(pcap->file, 0L, SEEK_SET);
|
||||||
|
// file empty is allowed, so return ESP_OK
|
||||||
|
ESP_RETURN_ON_FALSE(size, ESP_OK, TAG, "pcap file is empty");
|
||||||
|
// packet index (by bytes)
|
||||||
|
uint32_t index = 0;
|
||||||
|
pcap_file_header_t file_header;
|
||||||
|
size_t real_read = fread(&file_header, sizeof(pcap_file_header_t), 1, pcap->file);
|
||||||
|
ESP_RETURN_ON_FALSE(real_read == 1, ESP_FAIL, TAG, "read pcap file header failed");
|
||||||
|
index += sizeof(pcap_file_header_t);
|
||||||
|
//print pcap header information
|
||||||
|
fprintf(print_file, "------------------------------------------------------------------------\n");
|
||||||
|
fprintf(print_file, "Pcap packet Head:\n");
|
||||||
|
fprintf(print_file, "------------------------------------------------------------------------\n");
|
||||||
|
fprintf(print_file, "Magic Number: %x\n", file_header.magic);
|
||||||
|
fprintf(print_file, "Major Version: %d\n", file_header.major);
|
||||||
|
fprintf(print_file, "Minor Version: %d\n", file_header.minor);
|
||||||
|
fprintf(print_file, "SnapLen: %d\n", file_header.snaplen);
|
||||||
|
fprintf(print_file, "LinkType: %d\n", file_header.link_type);
|
||||||
|
fprintf(print_file, "------------------------------------------------------------------------\n");
|
||||||
|
uint32_t packet_num = 0;
|
||||||
|
pcap_packet_header_t packet_header;
|
||||||
|
while (index < size) {
|
||||||
|
real_read = fread(&packet_header, sizeof(pcap_packet_header_t), 1, pcap->file);
|
||||||
|
ESP_GOTO_ON_FALSE(real_read == 1, ESP_FAIL, err, TAG, "read pcap packet header failed");
|
||||||
|
// print packet header information
|
||||||
|
fprintf(print_file, "Packet %d:\n", packet_num);
|
||||||
|
fprintf(print_file, "Timestamp (Seconds): %d\n", packet_header.seconds);
|
||||||
|
fprintf(print_file, "Timestamp (Microseconds): %d\n", packet_header.microseconds);
|
||||||
|
fprintf(print_file, "Capture Length: %d\n", packet_header.capture_length);
|
||||||
|
fprintf(print_file, "Packet Length: %d\n", packet_header.packet_length);
|
||||||
|
size_t payload_length = packet_header.capture_length;
|
||||||
|
packet_payload = malloc(payload_length);
|
||||||
|
ESP_GOTO_ON_FALSE(packet_payload, ESP_ERR_NO_MEM, err, TAG, "no mem to save packet payload");
|
||||||
|
real_read = fread(packet_payload, payload_length, 1, pcap->file);
|
||||||
|
ESP_GOTO_ON_FALSE(real_read == 1, ESP_FAIL, err, TAG, "read payload error");
|
||||||
|
// print packet information
|
||||||
|
// currently only print info for 802.11
|
||||||
|
if (file_header.link_type == PCAP_LINK_TYPE_802_11) {
|
||||||
|
fprintf(print_file, "Packet Type: %2x\n", (packet_payload[0] >> 4) & 0x03);
|
||||||
|
fprintf(print_file, "Packet Subtype: %2x\n", packet_payload[0] & 0x0F);
|
||||||
|
fprintf(print_file, "Destination: ");
|
||||||
|
for (int j = 0; j < 5; j++) {
|
||||||
|
fprintf(print_file, "%2x ", packet_payload[4 + j]);
|
||||||
|
}
|
||||||
|
fprintf(print_file, "%2x\n", packet_payload[9]);
|
||||||
|
fprintf(print_file, "Source: ");
|
||||||
|
for (int j = 0; j < 5; j++) {
|
||||||
|
fprintf(print_file, "%2x ", packet_payload[10 + j]);
|
||||||
|
}
|
||||||
|
fprintf(print_file, "%2x\n", packet_payload[15]);
|
||||||
|
fprintf(print_file, "------------------------------------------------------------------------\n");
|
||||||
|
} else if (file_header.link_type == PCAP_LINK_TYPE_ETHERNET){
|
||||||
|
fprintf(print_file, "Destination: ");
|
||||||
|
for (int j = 0; j < 5; j++) {
|
||||||
|
fprintf(print_file, "%2x ", packet_payload[j]);
|
||||||
|
}
|
||||||
|
fprintf(print_file, "%2x\n", packet_payload[5]);
|
||||||
|
fprintf(print_file, "Source: ");
|
||||||
|
for (int j = 0; j < 5; j++) {
|
||||||
|
fprintf(print_file, "%2x ", packet_payload[6 + j]);
|
||||||
|
}
|
||||||
|
fprintf(print_file, "%2x\n", packet_payload[11]);
|
||||||
|
fprintf(print_file, "Type: 0x%x\n", packet_payload[13] | (packet_payload[12] << 8));
|
||||||
|
fprintf(print_file, "------------------------------------------------------------------------\n");
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
fprintf(print_file, "Unknown link type:%d\n", file_header.link_type);
|
||||||
|
fprintf(print_file, "------------------------------------------------------------------------\n");
|
||||||
|
}
|
||||||
|
free(packet_payload);
|
||||||
|
packet_payload = NULL;
|
||||||
|
index += packet_header.capture_length + sizeof(pcap_packet_header_t);
|
||||||
|
packet_num ++;
|
||||||
|
}
|
||||||
|
fprintf(print_file, "Pcap packet Number: %d\n", packet_num);
|
||||||
|
fprintf(print_file, "------------------------------------------------------------------------\n");
|
||||||
|
return ret;
|
||||||
|
err:
|
||||||
|
if (packet_payload) {
|
||||||
|
free(packet_payload);
|
||||||
|
}
|
||||||
|
return ret;
|
||||||
|
}
|
@ -1,2 +0,0 @@
|
|||||||
idf_component_register(SRCS "pcap.c"
|
|
||||||
INCLUDE_DIRS .)
|
|
@ -1,11 +0,0 @@
|
|||||||
#
|
|
||||||
# Component Makefile
|
|
||||||
#
|
|
||||||
# This Makefile should, at the very least, just include $(SDK_PATH)/Makefile. By default,
|
|
||||||
# this will take the sources in the src/ directory, compile them and link them into
|
|
||||||
# lib(subdirectory_name).a in the build directory. This behaviour is entirely configurable,
|
|
||||||
# please read the SDK documents if you need to do this.
|
|
||||||
#
|
|
||||||
|
|
||||||
#include $(IDF_PATH)/make/component_common.mk
|
|
||||||
COMPONENT_ADD_INCLUDEDIRS := .
|
|
@ -1,141 +0,0 @@
|
|||||||
// Copyright 2015-2019 Espressif Systems (Shanghai) PTE LTD
|
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
// you may not use this file except in compliance with the License.
|
|
||||||
// You may obtain a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
// See the License for the specific language governing permissions and
|
|
||||||
// limitations under the License.
|
|
||||||
#include <stdio.h>
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include <string.h>
|
|
||||||
#include <sys/unistd.h>
|
|
||||||
#include <sys/fcntl.h>
|
|
||||||
#include "esp_types.h"
|
|
||||||
#include "esp_err.h"
|
|
||||||
#include "esp_log.h"
|
|
||||||
#include "pcap.h"
|
|
||||||
|
|
||||||
static const char *PCAP_TAG = "pcap";
|
|
||||||
|
|
||||||
#define PCAP_CHECK(a, str, goto_tag, ...) \
|
|
||||||
do \
|
|
||||||
{ \
|
|
||||||
if (!(a)) \
|
|
||||||
{ \
|
|
||||||
ESP_LOGE(PCAP_TAG, "%s(%d): " str, __FUNCTION__, __LINE__, ##__VA_ARGS__); \
|
|
||||||
goto goto_tag; \
|
|
||||||
} \
|
|
||||||
} while (0)
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief Pcap File Header
|
|
||||||
*
|
|
||||||
*/
|
|
||||||
typedef struct {
|
|
||||||
uint32_t magic; /*!< Magic Number */
|
|
||||||
uint16_t major; /*!< Major Version */
|
|
||||||
uint16_t minor; /*!< Minor Version */
|
|
||||||
uint32_t zone; /*!< Time Zone Offset */
|
|
||||||
uint32_t sigfigs; /*!< Timestamp Accuracy */
|
|
||||||
uint32_t snaplen; /*!< Max Length to Capture */
|
|
||||||
uint32_t link_type; /*!< Link Layer Type */
|
|
||||||
} pcap_file_header_t;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief Pcap Packet Header
|
|
||||||
*
|
|
||||||
*/
|
|
||||||
typedef struct {
|
|
||||||
uint32_t seconds; /*!< Number of seconds since January 1st, 1970, 00:00:00 GMT */
|
|
||||||
uint32_t microseconds; /*!< Number of microseconds when the packet was captured (offset from seconds) */
|
|
||||||
uint32_t capture_length; /*!< Number of bytes of captured data, no longer than packet_length */
|
|
||||||
uint32_t packet_length; /*!< Actual length of current packet */
|
|
||||||
} pcap_packet_header_t;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief Pcap Runtime Handle
|
|
||||||
*
|
|
||||||
*/
|
|
||||||
typedef struct {
|
|
||||||
FILE *file; /*!< File handle */
|
|
||||||
} pcap_runtime_t;
|
|
||||||
|
|
||||||
esp_err_t pcap_capture_packet(pcap_handle_t handle, void *payload, uint32_t length, uint32_t seconds, uint32_t microseconds)
|
|
||||||
{
|
|
||||||
PCAP_CHECK(handle, "pcap handle is NULL", err);
|
|
||||||
pcap_runtime_t *pcap_rt = (pcap_runtime_t *)handle;
|
|
||||||
PCAP_CHECK(pcap_rt->file, "pcap file is NULL", err);
|
|
||||||
size_t real_write = 0;
|
|
||||||
pcap_packet_header_t header = {
|
|
||||||
.seconds = seconds,
|
|
||||||
.microseconds = microseconds,
|
|
||||||
.capture_length = length,
|
|
||||||
.packet_length = length
|
|
||||||
};
|
|
||||||
real_write = fwrite(&header, sizeof(header), 1, pcap_rt->file);
|
|
||||||
PCAP_CHECK(real_write == 1, "write packet header error", err);
|
|
||||||
real_write = fwrite(payload, sizeof(uint8_t), length, pcap_rt->file);
|
|
||||||
PCAP_CHECK(real_write == length, "write packet payload error", err);
|
|
||||||
/* Flush content in the buffer into device */
|
|
||||||
fflush(pcap_rt->file);
|
|
||||||
return ESP_OK;
|
|
||||||
err:
|
|
||||||
return ESP_FAIL;
|
|
||||||
}
|
|
||||||
|
|
||||||
esp_err_t pcap_deinit(pcap_handle_t handle)
|
|
||||||
{
|
|
||||||
PCAP_CHECK(handle, "pcap handle is NULL", err);
|
|
||||||
pcap_runtime_t *pcap_rt = (pcap_runtime_t *)handle;
|
|
||||||
|
|
||||||
if (pcap_rt->file != NULL) {
|
|
||||||
fclose(pcap_rt->file);
|
|
||||||
pcap_rt->file = NULL;
|
|
||||||
}
|
|
||||||
free(pcap_rt);
|
|
||||||
ESP_LOGD(PCAP_TAG, "pcap deinit OK");
|
|
||||||
return ESP_OK;
|
|
||||||
err:
|
|
||||||
ESP_LOGW(PCAP_TAG, "pcap deinit failed");
|
|
||||||
return ESP_FAIL;
|
|
||||||
}
|
|
||||||
|
|
||||||
esp_err_t pcap_init(pcap_config_t *config, pcap_handle_t *handle)
|
|
||||||
{
|
|
||||||
PCAP_CHECK(config, "config is NULL", err);
|
|
||||||
PCAP_CHECK(handle, "pcap handle is NULL", err);
|
|
||||||
pcap_runtime_t *pcap_rt = calloc(1, sizeof(pcap_runtime_t));
|
|
||||||
PCAP_CHECK(pcap_rt, "calloc pcap runtime failed", err);
|
|
||||||
pcap_rt->file = config->fp;
|
|
||||||
/* Write Pcap File header */
|
|
||||||
pcap_file_header_t header = {
|
|
||||||
.magic = PCAP_MAGIC_BIG_ENDIAN,
|
|
||||||
.major = PCAP_VERSION_MAJOR,
|
|
||||||
.minor = PCAP_VERSION_MINOR,
|
|
||||||
.zone = PCAP_TIME_ZONE_GMT,
|
|
||||||
.sigfigs = 0,
|
|
||||||
.snaplen = 0x40000,
|
|
||||||
.link_type = config->link_type
|
|
||||||
};
|
|
||||||
size_t real_write = fwrite(&header, sizeof(header), 1, pcap_rt->file);
|
|
||||||
PCAP_CHECK(real_write == 1, "write pcap file header failed", err_write);
|
|
||||||
/* Flush content in the buffer into device */
|
|
||||||
fflush(pcap_rt->file);
|
|
||||||
*handle = (pcap_handle_t)pcap_rt;
|
|
||||||
ESP_LOGD(PCAP_TAG, "pcap init OK");
|
|
||||||
return ESP_OK;
|
|
||||||
/* Error Handling */
|
|
||||||
err_write:
|
|
||||||
fclose(pcap_rt->file);
|
|
||||||
pcap_rt->file = NULL;
|
|
||||||
free(pcap_rt);
|
|
||||||
err:
|
|
||||||
ESP_LOGW(PCAP_TAG, "pcap init failed");
|
|
||||||
return ESP_FAIL;
|
|
||||||
}
|
|
@ -1,105 +0,0 @@
|
|||||||
// Copyright 2015-2019 Espressif Systems (Shanghai) PTE LTD
|
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
// you may not use this file except in compliance with the License.
|
|
||||||
// You may obtain a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
// See the License for the specific language governing permissions and
|
|
||||||
// limitations under the License.
|
|
||||||
#pragma once
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
extern "C" {
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#include <stdio.h>
|
|
||||||
#include "esp_err.h"
|
|
||||||
|
|
||||||
#define PCAP_MAGIC_BIG_ENDIAN 0xA1B2C3D4 /*!< Big-Endian */
|
|
||||||
#define PCAP_MAGIC_LITTLE_ENDIAN 0xD4C3B2A1 /*!< Little-Endian */
|
|
||||||
|
|
||||||
#define PCAP_VERSION_MAJOR 0x02 /*!< Major Version */
|
|
||||||
#define PCAP_VERSION_MINOR 0x04 /*!< Minor Version */
|
|
||||||
|
|
||||||
#define PCAP_TIME_ZONE_GMT 0x00 /*!< Time Zone */
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief Link layer Type Definition, used for Pcap reader to decode payload
|
|
||||||
*
|
|
||||||
*/
|
|
||||||
typedef enum {
|
|
||||||
PCAP_LINK_TYPE_LOOPBACK = 0, /*!< Loopback devices, except for later OpenBSD */
|
|
||||||
PCAP_LINK_TYPE_ETHERNET = 1, /*!< Ethernet, and Linux loopback devices */
|
|
||||||
PCAP_LINK_TYPE_TOKEN_RING = 6, /*!< 802.5 Token Ring */
|
|
||||||
PCAP_LINK_TYPE_ARCNET = 7, /*!< ARCnet */
|
|
||||||
PCAP_LINK_TYPE_SLIP = 8, /*!< SLIP */
|
|
||||||
PCAP_LINK_TYPE_PPP = 9, /*!< PPP */
|
|
||||||
PCAP_LINK_TYPE_FDDI = 10, /*!< FDDI */
|
|
||||||
PCAP_LINK_TYPE_ATM = 100, /*!< LLC/SNAP encapsulated ATM */
|
|
||||||
PCAP_LINK_TYPE_RAW_IP = 101, /*!< Raw IP, without link */
|
|
||||||
PCAP_LINK_TYPE_BSD_SLIP = 102, /*!< BSD/OS SLIP */
|
|
||||||
PCAP_LINK_TYPE_BSD_PPP = 103, /*!< BSD/OS PPP */
|
|
||||||
PCAP_LINK_TYPE_CISCO_HDLC = 104, /*!< Cisco HDLC */
|
|
||||||
PCAP_LINK_TYPE_802_11 = 105, /*!< 802.11 */
|
|
||||||
PCAP_LINK_TYPE_BSD_LOOPBACK = 108, /*!< OpenBSD loopback devices(with AF_value in network byte order) */
|
|
||||||
PCAP_LINK_TYPE_LOCAL_TALK = 114 /*!< LocalTalk */
|
|
||||||
} pcap_link_type_t;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief Pcap configuration Type Definition
|
|
||||||
*
|
|
||||||
*/
|
|
||||||
typedef struct {
|
|
||||||
FILE *fp; /* Pointer to a standard file handle */
|
|
||||||
pcap_link_type_t link_type; /* Pcap Link Type */
|
|
||||||
} pcap_config_t;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief Pcap Handle Type Definition
|
|
||||||
*
|
|
||||||
*/
|
|
||||||
typedef void *pcap_handle_t;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief Initialize a pcap session
|
|
||||||
*
|
|
||||||
* @param config configuration of creating pcap object
|
|
||||||
* @param handle pcap handle
|
|
||||||
* @return esp_err_t
|
|
||||||
* - ESP_OK on success
|
|
||||||
* - ESP_FAIL on error
|
|
||||||
*/
|
|
||||||
esp_err_t pcap_init(pcap_config_t *config, pcap_handle_t *handle);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief De-initialize a pcap session
|
|
||||||
*
|
|
||||||
* @param handle pcap handle
|
|
||||||
* @return esp_err_t
|
|
||||||
* - ESP_OK on success
|
|
||||||
* - ESP_FAIL on error
|
|
||||||
*/
|
|
||||||
esp_err_t pcap_deinit(pcap_handle_t handle);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief Capture one packet into pcap file
|
|
||||||
*
|
|
||||||
* @param handle pcap handle
|
|
||||||
* @param payload pointer of the captured data
|
|
||||||
* @param length length of captured data
|
|
||||||
* @param seconds second of capture time
|
|
||||||
* @param microseconds microsecond of capture time
|
|
||||||
* @return esp_err_t
|
|
||||||
* - ESP_OK on success
|
|
||||||
* - ESP_FAIL on error
|
|
||||||
*/
|
|
||||||
esp_err_t pcap_capture_packet(pcap_handle_t handle, void *payload, uint32_t length, uint32_t seconds, uint32_t microseconds);
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
}
|
|
||||||
#endif
|
|
Loading…
x
Reference in New Issue
Block a user