mirror of
https://github.com/emqx/neuron.git
synced 2026-10-29 15:09:11 +08:00
@@ -30,6 +30,7 @@ typedef enum neu_conn_type {
|
||||
NEU_CONN_TCP_CLIENT,
|
||||
NEU_CONN_UDP,
|
||||
NEU_CONN_TTY_CLIENT,
|
||||
NEU_CONN_UDP_TO,
|
||||
} neu_conn_type_e;
|
||||
|
||||
typedef enum neu_conn_tty_baud {
|
||||
@@ -103,6 +104,12 @@ typedef struct neu_conn_param {
|
||||
uint16_t timeout; // millisecond
|
||||
} udp;
|
||||
|
||||
struct {
|
||||
char * src_ip;
|
||||
uint16_t src_port;
|
||||
uint16_t timeout; // millisecond
|
||||
} udpto;
|
||||
|
||||
struct {
|
||||
char * device;
|
||||
neu_conn_tty_data_e data;
|
||||
@@ -231,6 +238,8 @@ int neu_conn_tcp_server_close_client(neu_conn_t *conn, int fd);
|
||||
* less than or equal to 0 fails.
|
||||
*/
|
||||
ssize_t neu_conn_send(neu_conn_t *conn, uint8_t *buf, ssize_t len);
|
||||
ssize_t neu_conn_udp_sendto(neu_conn_t *conn, uint8_t *buf, ssize_t len,
|
||||
void *dst);
|
||||
|
||||
/**
|
||||
* @brief Receive data via connection read.
|
||||
@@ -242,6 +251,8 @@ ssize_t neu_conn_send(neu_conn_t *conn, uint8_t *buf, ssize_t len);
|
||||
* received, less than or equal to 0 fails.
|
||||
*/
|
||||
ssize_t neu_conn_recv(neu_conn_t *conn, uint8_t *buf, ssize_t len);
|
||||
ssize_t neu_conn_udp_recvfrom(neu_conn_t *conn, uint8_t *buf, ssize_t len,
|
||||
void *src);
|
||||
|
||||
/**
|
||||
* @brief Specify the client to send data.
|
||||
|
||||
@@ -32,19 +32,15 @@ neu_conn_eth_t *neu_conn_eth_init(const char *interface, void *ctx);
|
||||
int neu_conn_eth_uninit(neu_conn_eth_t *conn);
|
||||
int neu_conn_eth_check_interface(const char *interface);
|
||||
void neu_conn_eth_get_mac(neu_conn_eth_t *conn, uint8_t *mac);
|
||||
int neu_conn_eth_size();
|
||||
|
||||
typedef void (*neu_conn_eth_msg_callback)(neu_conn_eth_t *conn, void *ctx,
|
||||
uint16_t protocol, uint16_t n_byte,
|
||||
uint8_t *bytes, uint8_t src_mac[6]);
|
||||
|
||||
int neu_conn_eth_register_lldp(neu_conn_eth_t *conn, const char *device,
|
||||
neu_conn_eth_msg_callback callback);
|
||||
int neu_conn_eth_unregister_lldp(neu_conn_eth_t *conn, const char *device);
|
||||
|
||||
typedef struct neu_conn_eth_sub neu_conn_eth_sub_t;
|
||||
|
||||
neu_conn_eth_sub_t *neu_conn_eth_register(neu_conn_eth_t *conn,
|
||||
uint16_t protocol, uint8_t mac[6],
|
||||
neu_conn_eth_sub_t *neu_conn_eth_register(neu_conn_eth_t *conn, uint8_t mac[6],
|
||||
neu_conn_eth_msg_callback callback);
|
||||
int neu_conn_eth_unregister(neu_conn_eth_t *conn, neu_conn_eth_sub_t *sub);
|
||||
|
||||
|
||||
@@ -75,6 +75,7 @@ typedef enum {
|
||||
NEU_ERR_TAG_ADDRESS_TOO_LONG = 2207,
|
||||
NEU_ERR_TAG_DESCRIPTION_TOO_LONG = 2208,
|
||||
NEU_ERR_TAG_PRECISION_INVALID = 2209,
|
||||
NEU_ERR_TAG_EXIST = 2210,
|
||||
|
||||
NEU_ERR_LIBRARY_NOT_FOUND = 2301,
|
||||
NEU_ERR_LIBRARY_INFO_INVALID = 2302,
|
||||
|
||||
@@ -312,6 +312,7 @@ ssize_t neu_conn_send(neu_conn_t *conn, uint8_t *buf, ssize_t len)
|
||||
|
||||
if (conn->is_connected) {
|
||||
switch (conn->param.type) {
|
||||
case NEU_CONN_UDP_TO:
|
||||
case NEU_CONN_TCP_SERVER:
|
||||
assert(false);
|
||||
break;
|
||||
@@ -364,6 +365,7 @@ ssize_t neu_conn_recv(neu_conn_t *conn, uint8_t *buf, ssize_t len)
|
||||
}
|
||||
|
||||
switch (conn->param.type) {
|
||||
case NEU_CONN_UDP_TO:
|
||||
case NEU_CONN_TCP_SERVER:
|
||||
zlog_fatal(conn->param.log, "neu_conn_recv cann't recv tcp server msg");
|
||||
assert(1 == 0);
|
||||
@@ -414,6 +416,115 @@ ssize_t neu_conn_recv(neu_conn_t *conn, uint8_t *buf, ssize_t len)
|
||||
|
||||
return ret;
|
||||
}
|
||||
ssize_t neu_conn_udp_sendto(neu_conn_t *conn, uint8_t *buf, ssize_t len,
|
||||
void *dst)
|
||||
{
|
||||
ssize_t ret = 0;
|
||||
|
||||
pthread_mutex_lock(&conn->mtx);
|
||||
if (conn->stop) {
|
||||
pthread_mutex_unlock(&conn->mtx);
|
||||
return ret;
|
||||
}
|
||||
|
||||
if (!conn->is_connected) {
|
||||
conn_connect(conn);
|
||||
}
|
||||
|
||||
if (conn->is_connected) {
|
||||
switch (conn->param.type) {
|
||||
case NEU_CONN_TCP_CLIENT:
|
||||
case NEU_CONN_TCP_SERVER:
|
||||
case NEU_CONN_TTY_CLIENT:
|
||||
case NEU_CONN_UDP:
|
||||
assert(false);
|
||||
break;
|
||||
case NEU_CONN_UDP_TO:
|
||||
if (conn->block) {
|
||||
ret = sendto(conn->fd, buf, len, MSG_NOSIGNAL,
|
||||
(struct sockaddr *) dst, sizeof(struct sockaddr));
|
||||
} else {
|
||||
ret = sendto(conn->fd, buf, len, MSG_NOSIGNAL | MSG_DONTWAIT,
|
||||
(struct sockaddr *) dst, sizeof(struct sockaddr));
|
||||
}
|
||||
break;
|
||||
}
|
||||
if (ret != len) {
|
||||
zlog_error(
|
||||
conn->param.log,
|
||||
"conn udp fd: %d, sendto (%s:%d) buf len: %zd, ret: %zd, "
|
||||
"errno: %s(%d)",
|
||||
conn->fd, inet_ntoa(((struct sockaddr_in *) dst)->sin_addr),
|
||||
htons(((struct sockaddr_in *) dst)->sin_port), len, ret,
|
||||
strerror(errno), errno);
|
||||
}
|
||||
|
||||
if (ret == -1 && errno != EAGAIN) {
|
||||
conn_disconnect(conn);
|
||||
}
|
||||
|
||||
if (ret > 0 && conn->callback_trigger == false) {
|
||||
conn->connected(conn->data, conn->fd);
|
||||
conn->callback_trigger = true;
|
||||
}
|
||||
}
|
||||
|
||||
if (ret > 0) {
|
||||
conn->state.send_bytes += ret;
|
||||
}
|
||||
|
||||
pthread_mutex_unlock(&conn->mtx);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
ssize_t neu_conn_udp_recvfrom(neu_conn_t *conn, uint8_t *buf, ssize_t len,
|
||||
void *src)
|
||||
{
|
||||
ssize_t ret = 0;
|
||||
socklen_t addr_len = sizeof(struct sockaddr_in);
|
||||
|
||||
pthread_mutex_lock(&conn->mtx);
|
||||
if (conn->stop) {
|
||||
pthread_mutex_unlock(&conn->mtx);
|
||||
return ret;
|
||||
}
|
||||
|
||||
switch (conn->param.type) {
|
||||
case NEU_CONN_TCP_SERVER:
|
||||
case NEU_CONN_TCP_CLIENT:
|
||||
case NEU_CONN_TTY_CLIENT:
|
||||
case NEU_CONN_UDP:
|
||||
assert(1 == 0);
|
||||
break;
|
||||
case NEU_CONN_UDP_TO:
|
||||
ret =
|
||||
recvfrom(conn->fd, buf, len, 0, (struct sockaddr *) src, &addr_len);
|
||||
break;
|
||||
}
|
||||
if (ret <= 0) {
|
||||
zlog_error(conn->param.log,
|
||||
"conn udp fd: %d, recv buf len %zd, ret: %zd, errno: %s(%d)",
|
||||
conn->fd, len, ret, strerror(errno), errno);
|
||||
}
|
||||
|
||||
if (errno == EPIPE || ret == -1 || ret == 0) {
|
||||
conn_disconnect(conn);
|
||||
}
|
||||
|
||||
if (ret > 0 && conn->callback_trigger == false) {
|
||||
conn->connected(conn->data, conn->fd);
|
||||
conn->callback_trigger = true;
|
||||
}
|
||||
|
||||
pthread_mutex_unlock(&conn->mtx);
|
||||
|
||||
if (ret > 0) {
|
||||
conn->state.recv_bytes += ret;
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
void neu_conn_connect(neu_conn_t *conn)
|
||||
{
|
||||
@@ -453,6 +564,9 @@ static void conn_free_param(neu_conn_t *conn)
|
||||
free(conn->param.params.udp.src_ip);
|
||||
free(conn->param.params.udp.dst_ip);
|
||||
break;
|
||||
case NEU_CONN_UDP_TO:
|
||||
free(conn->param.params.udpto.src_ip);
|
||||
break;
|
||||
case NEU_CONN_TTY_CLIENT:
|
||||
free(conn->param.params.tty_client.device);
|
||||
break;
|
||||
@@ -499,6 +613,12 @@ static void conn_init_param(neu_conn_t *conn, neu_conn_param_t *param)
|
||||
conn->param.params.udp.timeout = param->params.udp.timeout;
|
||||
conn->block = conn->param.params.udp.timeout > 0;
|
||||
break;
|
||||
case NEU_CONN_UDP_TO:
|
||||
conn->param.params.udpto.src_ip = strdup(param->params.udpto.src_ip);
|
||||
conn->param.params.udpto.src_port = param->params.udpto.src_port;
|
||||
conn->param.params.udpto.timeout = param->params.udpto.timeout;
|
||||
conn->block = conn->param.params.udpto.timeout > 0;
|
||||
break;
|
||||
case NEU_CONN_TTY_CLIENT:
|
||||
conn->param.params.tty_client.device =
|
||||
strdup(param->params.tty_client.device);
|
||||
@@ -683,6 +803,44 @@ static void conn_connect(neu_conn_t *conn)
|
||||
}
|
||||
break;
|
||||
}
|
||||
case NEU_CONN_UDP_TO: {
|
||||
if (conn->block) {
|
||||
struct timeval tv = {
|
||||
.tv_sec = conn->param.params.udpto.timeout / 1000,
|
||||
.tv_usec = (conn->param.params.udpto.timeout % 1000) * 1000,
|
||||
};
|
||||
|
||||
fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
||||
setsockopt(fd, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));
|
||||
setsockopt(fd, SOL_SOCKET, SO_SNDTIMEO, &tv, sizeof(tv));
|
||||
} else {
|
||||
fd = socket(AF_INET, SOCK_DGRAM | SOCK_NONBLOCK, IPPROTO_UDP);
|
||||
}
|
||||
int so_broadcast = 1;
|
||||
setsockopt(fd, SOL_SOCKET, SO_BROADCAST, &so_broadcast,
|
||||
sizeof(so_broadcast));
|
||||
struct sockaddr_in local = {
|
||||
.sin_family = AF_INET,
|
||||
.sin_port = htons(conn->param.params.udpto.src_port),
|
||||
.sin_addr.s_addr = inet_addr(conn->param.params.udpto.src_ip),
|
||||
};
|
||||
|
||||
ret = bind(fd, (struct sockaddr *) &local, sizeof(struct sockaddr_in));
|
||||
if (ret != 0) {
|
||||
close(fd);
|
||||
zlog_error(conn->param.log, "bind %s:%d error: %s(%d)",
|
||||
conn->param.params.udpto.src_ip,
|
||||
conn->param.params.udpto.src_port, strerror(errno),
|
||||
errno);
|
||||
conn->is_connected = false;
|
||||
return;
|
||||
}
|
||||
|
||||
conn->is_connected = true;
|
||||
conn->fd = fd;
|
||||
|
||||
break;
|
||||
}
|
||||
case NEU_CONN_TTY_CLIENT: {
|
||||
struct termios tty_opt = { 0 };
|
||||
|
||||
@@ -854,6 +1012,7 @@ static void conn_disconnect(neu_conn_t *conn)
|
||||
break;
|
||||
case NEU_CONN_TCP_CLIENT:
|
||||
case NEU_CONN_UDP:
|
||||
case NEU_CONN_UDP_TO:
|
||||
case NEU_CONN_TTY_CLIENT:
|
||||
if (conn->fd > 0) {
|
||||
close(conn->fd);
|
||||
|
||||
+101
-198
@@ -33,6 +33,7 @@
|
||||
|
||||
#include "connection/neu_connection_eth.h"
|
||||
#include "event/event.h"
|
||||
#include "type.h"
|
||||
#include "utils/log.h"
|
||||
#include "utils/utextend.h"
|
||||
|
||||
@@ -40,13 +41,12 @@
|
||||
#define ETH_P_LLDP 0x88CC
|
||||
#endif
|
||||
|
||||
typedef struct neu_conn_eth_sub {
|
||||
uint16_t protocol;
|
||||
char mac[ETH_ALEN];
|
||||
} callback_elem_key_t;
|
||||
struct neu_conn_eth_sub {
|
||||
uint8_t mac[6];
|
||||
};
|
||||
|
||||
typedef struct {
|
||||
callback_elem_key_t key;
|
||||
char mac[20];
|
||||
|
||||
neu_conn_eth_msg_callback callback;
|
||||
|
||||
@@ -54,8 +54,6 @@ typedef struct {
|
||||
} callback_elem_t;
|
||||
|
||||
typedef struct {
|
||||
char device[NEU_NODE_NAME_LEN];
|
||||
|
||||
neu_conn_eth_msg_callback callback;
|
||||
|
||||
UT_hash_handle hh;
|
||||
@@ -65,20 +63,16 @@ typedef struct interface_conn {
|
||||
char * interface;
|
||||
neu_events_t *events;
|
||||
uint8_t count;
|
||||
uint8_t buf[1500];
|
||||
|
||||
int lldp_fd;
|
||||
int profinet_fd;
|
||||
int vlan_fd;
|
||||
uint8_t mac[ETH_ALEN];
|
||||
|
||||
neu_event_io_t *profinet_event;
|
||||
neu_event_io_t *lldp_event;
|
||||
neu_event_io_t *vlan_event;
|
||||
|
||||
lldp_callback_elem_t *lldp_callbacks;
|
||||
callback_elem_t * callbacks;
|
||||
pthread_mutex_t mtx;
|
||||
callback_elem_t *callbacks;
|
||||
pthread_mutex_t mtx;
|
||||
} interface_conn_t;
|
||||
|
||||
static pthread_mutex_t mtx = { 0 };
|
||||
@@ -92,13 +86,25 @@ struct neu_conn_eth {
|
||||
};
|
||||
|
||||
static int get_mac(neu_conn_eth_t *conn, const char *interface);
|
||||
static int init_socket(const char *interface, uint16_t protocol);
|
||||
static int init_socket(const char *interface, uint16_t protocol,
|
||||
uint8_t mac[6]);
|
||||
static int uninit_socket(int fd);
|
||||
|
||||
static int lldp_msg_cb(enum neu_event_io_type type, int fd, void *usr_data);
|
||||
static int eth_msg_cb(enum neu_event_io_type type, int fd, void *usr_data);
|
||||
static int eth_msg_cb(enum neu_event_io_type type, int fd, void *usr_data);
|
||||
static uint8_t pf_dcp_broadcast[ETH_ALEN] = {
|
||||
0x01, 0x0e, 0xcf, 0x00, 0x00, 0x01
|
||||
};
|
||||
|
||||
static char *lldp_parse_port_id(uint16_t n_byte, uint8_t *bytes);
|
||||
int neu_conn_eth_size()
|
||||
{
|
||||
int ret = 0;
|
||||
|
||||
for (int i = 0; i < interface_max; i++) {
|
||||
ret += in_conns[i].count;
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
neu_conn_eth_t *neu_conn_eth_init(const char *interface, void *ctx)
|
||||
{
|
||||
@@ -127,18 +133,20 @@ neu_conn_eth_t *neu_conn_eth_init(const char *interface, void *ctx)
|
||||
if (in_conns[i].interface == NULL) {
|
||||
neu_event_io_param_t param = { 0 };
|
||||
|
||||
pthread_mutex_init(&in_conns[i].mtx, NULL);
|
||||
conn_eth->ic = &in_conns[i];
|
||||
|
||||
in_conns[i].interface = strdup(interface);
|
||||
in_conns[i].events = neu_event_new();
|
||||
in_conns[i].count += 1;
|
||||
|
||||
in_conns[i].profinet_fd = init_socket(interface, 0x8892);
|
||||
in_conns[i].lldp_fd = init_socket(interface, ETH_P_LLDP);
|
||||
in_conns[i].vlan_fd = init_socket(interface, 0x8100);
|
||||
|
||||
pthread_mutex_init(&in_conns[i].mtx, NULL);
|
||||
conn_eth->ic = &in_conns[i];
|
||||
get_mac(conn_eth, interface);
|
||||
|
||||
in_conns[i].profinet_fd =
|
||||
init_socket(interface, 0x8892, in_conns[i].mac);
|
||||
// in_conns[i].vlan_fd =
|
||||
// init_socket(interface, 0x8100, in_conns[i].mac);
|
||||
|
||||
param.fd = in_conns[i].profinet_fd;
|
||||
param.usr_data = (void *) conn_eth;
|
||||
param.cb = eth_msg_cb;
|
||||
@@ -146,17 +154,11 @@ neu_conn_eth_t *neu_conn_eth_init(const char *interface, void *ctx)
|
||||
in_conns[i].profinet_event =
|
||||
neu_event_add_io(in_conns[i].events, param);
|
||||
|
||||
param.fd = in_conns[i].lldp_fd;
|
||||
param.cb = lldp_msg_cb;
|
||||
// param.fd = in_conns[i].vlan_fd;
|
||||
// param.cb = eth_msg_cb;
|
||||
|
||||
in_conns[i].lldp_event =
|
||||
neu_event_add_io(in_conns[i].events, param);
|
||||
|
||||
param.fd = in_conns[i].vlan_fd;
|
||||
param.cb = eth_msg_cb;
|
||||
|
||||
in_conns[i].vlan_event =
|
||||
neu_event_add_io(in_conns[i].events, param);
|
||||
// in_conns[i].vlan_event =
|
||||
// neu_event_add_io(in_conns[i].events, param);
|
||||
|
||||
break;
|
||||
}
|
||||
@@ -175,16 +177,15 @@ int neu_conn_eth_uninit(neu_conn_eth_t *conn)
|
||||
conn->ic->count -= 1;
|
||||
if (conn->ic->count == 0) {
|
||||
neu_event_del_io(conn->ic->events, conn->ic->profinet_event);
|
||||
neu_event_del_io(conn->ic->events, conn->ic->lldp_event);
|
||||
neu_event_del_io(conn->ic->events, conn->ic->vlan_event);
|
||||
// neu_event_del_io(conn->ic->events, conn->ic->vlan_event);
|
||||
|
||||
neu_event_close(conn->ic->events);
|
||||
|
||||
uninit_socket(conn->ic->profinet_fd);
|
||||
uninit_socket(conn->ic->lldp_fd);
|
||||
uninit_socket(conn->ic->vlan_fd);
|
||||
// uninit_socket(conn->ic->vlan_fd);
|
||||
|
||||
free(conn->ic->interface);
|
||||
conn->ic->interface = NULL;
|
||||
pthread_mutex_destroy(&conn->ic->mtx);
|
||||
}
|
||||
pthread_mutex_unlock(&mtx);
|
||||
@@ -219,33 +220,29 @@ int neu_conn_eth_check_interface(const char *interface)
|
||||
return ret;
|
||||
}
|
||||
|
||||
neu_conn_eth_sub_t *neu_conn_eth_register(neu_conn_eth_t *conn,
|
||||
uint16_t protocol, uint8_t mac[6],
|
||||
neu_conn_eth_sub_t *neu_conn_eth_register(neu_conn_eth_t *conn, uint8_t xmac[6],
|
||||
neu_conn_eth_msg_callback callback)
|
||||
{
|
||||
callback_elem_key_t key = { 0 };
|
||||
neu_conn_eth_sub_t *sub = NULL;
|
||||
callback_elem_t * elem = NULL;
|
||||
char tmp_mac[20] = { 0 };
|
||||
neu_conn_eth_sub_t *sub = NULL;
|
||||
callback_elem_t * elem = NULL;
|
||||
|
||||
key.protocol = protocol;
|
||||
memcpy(key.mac, mac, ETH_ALEN);
|
||||
snprintf(tmp_mac, sizeof(tmp_mac), "%X:%X:%X-%X:%X:%X", xmac[0], xmac[1],
|
||||
xmac[2], xmac[3], xmac[4], xmac[5]);
|
||||
|
||||
pthread_mutex_lock(&conn->ic->mtx);
|
||||
|
||||
HASH_FIND(hh, conn->ic->callbacks, &key, sizeof(callback_elem_key_t), elem);
|
||||
HASH_FIND_STR(conn->ic->callbacks, tmp_mac, elem);
|
||||
if (elem == NULL) {
|
||||
elem = calloc(1, sizeof(callback_elem_t));
|
||||
sub = calloc(1, sizeof(neu_conn_eth_sub_t));
|
||||
|
||||
elem->key.protocol = protocol;
|
||||
memcpy(key.mac, mac, ETH_ALEN);
|
||||
elem->callback = callback;
|
||||
memcpy(elem->mac, tmp_mac, sizeof(elem->mac));
|
||||
|
||||
memcpy(sub->mac, mac, ETH_ALEN);
|
||||
sub->protocol = protocol;
|
||||
memcpy(sub->mac, xmac, ETH_ALEN);
|
||||
|
||||
HASH_ADD(hh, conn->ic->callbacks, key, sizeof(callback_elem_key_t),
|
||||
elem);
|
||||
HASH_ADD_STR(conn->ic->callbacks, mac, elem);
|
||||
}
|
||||
|
||||
pthread_mutex_unlock(&conn->ic->mtx);
|
||||
@@ -255,15 +252,15 @@ neu_conn_eth_sub_t *neu_conn_eth_register(neu_conn_eth_t *conn,
|
||||
|
||||
int neu_conn_eth_unregister(neu_conn_eth_t *conn, neu_conn_eth_sub_t *sub)
|
||||
{
|
||||
callback_elem_key_t key = { 0 };
|
||||
callback_elem_t * elem = NULL;
|
||||
char tmp_mac[20] = { 0 };
|
||||
callback_elem_t *elem = NULL;
|
||||
|
||||
key.protocol = sub->protocol;
|
||||
memcpy(key.mac, sub->mac, ETH_ALEN);
|
||||
snprintf(tmp_mac, sizeof(tmp_mac), "%X:%X:%X-%X:%X:%X", sub->mac[0],
|
||||
sub->mac[1], sub->mac[2], sub->mac[3], sub->mac[4], sub->mac[5]);
|
||||
|
||||
pthread_mutex_lock(&conn->ic->mtx);
|
||||
|
||||
HASH_FIND(hh, conn->ic->callbacks, &key, sizeof(callback_elem_key_t), elem);
|
||||
HASH_FIND_STR(conn->ic->callbacks, tmp_mac, elem);
|
||||
if (elem != NULL) {
|
||||
HASH_DEL(conn->ic->callbacks, elem);
|
||||
free(elem);
|
||||
@@ -276,46 +273,6 @@ int neu_conn_eth_unregister(neu_conn_eth_t *conn, neu_conn_eth_sub_t *sub)
|
||||
return 0;
|
||||
}
|
||||
|
||||
int neu_conn_eth_register_lldp(neu_conn_eth_t *conn, const char *device,
|
||||
neu_conn_eth_msg_callback callback)
|
||||
{
|
||||
lldp_callback_elem_t *elem = NULL;
|
||||
int ret = 0;
|
||||
|
||||
pthread_mutex_lock(&conn->ic->mtx);
|
||||
|
||||
HASH_FIND_STR(conn->ic->lldp_callbacks, device, elem);
|
||||
if (elem != NULL) {
|
||||
ret = -1;
|
||||
} else {
|
||||
elem = calloc(1, sizeof(lldp_callback_elem_t));
|
||||
|
||||
elem->callback = callback;
|
||||
strcpy(elem->device, device);
|
||||
|
||||
HASH_ADD_STR(conn->ic->lldp_callbacks, device, elem);
|
||||
}
|
||||
|
||||
pthread_mutex_unlock(&conn->ic->mtx);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
int neu_conn_eth_unregister_lldp(neu_conn_eth_t *conn, const char *device)
|
||||
{
|
||||
lldp_callback_elem_t *elem = NULL;
|
||||
|
||||
pthread_mutex_lock(&conn->ic->mtx);
|
||||
HASH_FIND_STR(conn->ic->lldp_callbacks, device, elem);
|
||||
if (elem != NULL) {
|
||||
HASH_DEL(conn->ic->lldp_callbacks, elem);
|
||||
free(elem);
|
||||
}
|
||||
pthread_mutex_unlock(&conn->ic->mtx);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int neu_conn_eth_send(neu_conn_eth_t *conn, uint16_t protocol, uint16_t n_byte,
|
||||
uint8_t *bytes)
|
||||
{
|
||||
@@ -323,12 +280,10 @@ int neu_conn_eth_send(neu_conn_eth_t *conn, uint16_t protocol, uint16_t n_byte,
|
||||
int ret = 0;
|
||||
|
||||
switch (protocol) {
|
||||
case ETH_P_LLDP:
|
||||
case 0x8892:
|
||||
fd = conn->ic->profinet_fd;
|
||||
break;
|
||||
case ETH_P_LLDP:
|
||||
fd = conn->ic->lldp_fd;
|
||||
break;
|
||||
case 0x8100:
|
||||
fd = conn->ic->vlan_fd;
|
||||
break;
|
||||
@@ -357,20 +312,22 @@ static int get_mac(neu_conn_eth_t *conn, const char *interface)
|
||||
return ret;
|
||||
}
|
||||
|
||||
static int init_socket(const char *interface, uint16_t protocol)
|
||||
static int init_socket(const char *interface, uint16_t protocol, uint8_t mac[6])
|
||||
{
|
||||
struct ifreq ifr = { 0 };
|
||||
struct sockaddr_ll sll = { 0 };
|
||||
struct packet_mreq mreq = { 0 };
|
||||
const uint8_t pn_mcast_addr[ETH_ALEN] = {
|
||||
0x01, 0x0e, 0xcf, 0x00, 0x00, 0x00
|
||||
};
|
||||
int fd = socket(PF_PACKET, SOCK_RAW, htons(protocol));
|
||||
int ret = 0;
|
||||
int ret = 1;
|
||||
|
||||
assert(fd > 0);
|
||||
setsockopt(fd, SOL_SOCKET, SO_DONTROUTE, &ret, sizeof(ret));
|
||||
|
||||
(void) mac;
|
||||
(void) pn_mcast_addr;
|
||||
|
||||
strcpy(ifr.ifr_name, interface);
|
||||
if (ioctl(fd, SIOCGIFINDEX, &ifr) != 0) {
|
||||
close(fd);
|
||||
@@ -380,25 +337,12 @@ static int init_socket(const char *interface, uint16_t protocol)
|
||||
sll.sll_family = AF_PACKET;
|
||||
sll.sll_ifindex = ifr.ifr_ifindex;
|
||||
sll.sll_protocol = htons(protocol);
|
||||
|
||||
mreq.mr_ifindex = ifr.ifr_ifindex;
|
||||
mreq.mr_type = PACKET_HOST | PACKET_MR_MULTICAST;
|
||||
mreq.mr_alen = ETH_ALEN;
|
||||
memcpy(mreq.mr_address, pn_mcast_addr, ETH_ALEN);
|
||||
|
||||
ifr.ifr_flags = 0;
|
||||
ioctl(fd, SIOCGIFFLAGS, &ifr);
|
||||
|
||||
bind(fd, (struct sockaddr *) &sll, sizeof(sll));
|
||||
|
||||
if (protocol == 0x8892) {
|
||||
ret = setsockopt(fd, SOL_PACKET, PACKET_ADD_MEMBERSHIP, &mreq,
|
||||
sizeof(mreq));
|
||||
assert(ret == 0);
|
||||
}
|
||||
ret = bind(fd, (struct sockaddr *) &sll, sizeof(sll));
|
||||
assert(ret == 0);
|
||||
|
||||
return fd;
|
||||
}
|
||||
|
||||
static int uninit_socket(int fd)
|
||||
{
|
||||
close(fd);
|
||||
@@ -412,27 +356,53 @@ static int eth_msg_cb(enum neu_event_io_type type, int fd, void *usr_data)
|
||||
|
||||
switch (type) {
|
||||
case NEU_EVENT_IO_READ: {
|
||||
callback_elem_key_t key = { 0 };
|
||||
callback_elem_t * elem = NULL;
|
||||
|
||||
memset(conn->ic->buf, 0, sizeof(conn->ic->buf));
|
||||
int ret = recv(fd, conn->ic->buf, sizeof(conn->ic->buf), 0);
|
||||
callback_elem_t *elem = NULL;
|
||||
uint8_t buf[1500] = { 0 };
|
||||
int ret = recv(fd, buf, 1500, 0);
|
||||
|
||||
if (ret < ETH_HLEN) {
|
||||
nlog_warn("eth recv length < %d msg", ETH_HLEN);
|
||||
return 0;
|
||||
}
|
||||
|
||||
struct ethhdr *ehdr = (struct ethhdr *) conn->ic->buf;
|
||||
struct ethhdr *ehdr = (struct ethhdr *) buf;
|
||||
ehdr->h_proto = ntohs(ehdr->h_proto);
|
||||
|
||||
key.protocol = ntohs(ehdr->h_proto);
|
||||
memcpy(key.mac, ehdr->h_source, ETH_ALEN);
|
||||
if (ehdr->h_proto != 0x8100 && ehdr->h_proto != 0x8892) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
HASH_FIND(hh, conn->ic->callbacks, &key, sizeof(key), elem);
|
||||
if (memcmp(ehdr->h_dest, pf_dcp_broadcast, ETH_ALEN) == 0) {
|
||||
callback_elem_t *tmp = NULL;
|
||||
|
||||
if (elem != NULL) {
|
||||
elem->callback(conn, conn->ctx, key.protocol, ret - ETH_HLEN,
|
||||
conn->ic->buf + ETH_HLEN, ehdr->h_source);
|
||||
HASH_ITER(hh, conn->ic->callbacks, elem, tmp)
|
||||
{
|
||||
elem->callback(conn, conn->ctx, ehdr->h_proto, ret - ETH_HLEN,
|
||||
buf + ETH_HLEN, ehdr->h_source);
|
||||
}
|
||||
} else {
|
||||
if (memcmp(ehdr->h_dest, conn->ic->mac, ETH_ALEN) != 0) {
|
||||
return 0;
|
||||
}
|
||||
char tmp_mac[20] = { 0 };
|
||||
|
||||
snprintf(tmp_mac, sizeof(tmp_mac), "%X:%X:%X-%X:%X:%X",
|
||||
ehdr->h_source[0], ehdr->h_source[1], ehdr->h_source[2],
|
||||
ehdr->h_source[3], ehdr->h_source[4], ehdr->h_source[5]);
|
||||
|
||||
HASH_FIND_STR(conn->ic->callbacks, tmp_mac, elem);
|
||||
if (elem != NULL) {
|
||||
elem->callback(conn, conn->ctx, ehdr->h_proto, ret - ETH_HLEN,
|
||||
buf + ETH_HLEN, ehdr->h_source);
|
||||
} else {
|
||||
memset(tmp_mac, 0, sizeof(tmp_mac));
|
||||
snprintf(tmp_mac, sizeof(tmp_mac), "0:0:0-0:0:0");
|
||||
HASH_FIND_STR(conn->ic->callbacks, tmp_mac, elem);
|
||||
if (elem != NULL) {
|
||||
elem->callback(conn, conn->ctx, ehdr->h_proto,
|
||||
ret - ETH_HLEN, buf + ETH_HLEN,
|
||||
ehdr->h_source);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
break;
|
||||
@@ -446,70 +416,3 @@ static int eth_msg_cb(enum neu_event_io_type type, int fd, void *usr_data)
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int lldp_msg_cb(enum neu_event_io_type type, int fd, void *usr_data)
|
||||
{
|
||||
neu_conn_eth_t *conn = (neu_conn_eth_t *) usr_data;
|
||||
int ret = 0;
|
||||
char * port_id = NULL;
|
||||
|
||||
switch (type) {
|
||||
case NEU_EVENT_IO_READ:
|
||||
memset(conn->ic->buf, 0, sizeof(conn->ic->buf));
|
||||
ret = recv(fd, conn->ic->buf, sizeof(conn->ic->buf), 0);
|
||||
port_id = lldp_parse_port_id(ret - sizeof(struct ethhdr),
|
||||
conn->ic->buf + sizeof(struct ethhdr));
|
||||
|
||||
if (port_id != NULL) {
|
||||
lldp_callback_elem_t *elem = NULL;
|
||||
struct ethhdr * ehdr = (struct ethhdr *) conn->ic->buf;
|
||||
|
||||
HASH_FIND_STR(conn->ic->callbacks, port_id, elem);
|
||||
if (elem != NULL) {
|
||||
elem->callback(
|
||||
conn, conn->ctx, ETH_P_LLDP, ret - sizeof(struct ethhdr),
|
||||
conn->ic->buf + sizeof(struct ethhdr), ehdr->h_source);
|
||||
}
|
||||
}
|
||||
|
||||
break;
|
||||
case NEU_EVENT_IO_CLOSED:
|
||||
case NEU_EVENT_IO_HUP:
|
||||
nlog_warn("eth conn lldp error type: %d, fd: %d(%s)", type, fd,
|
||||
conn->ic->interface);
|
||||
break;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
struct lldp_tvl {
|
||||
uint16_t type : 7;
|
||||
uint16_t length : 9;
|
||||
} __attribute__((packed));
|
||||
|
||||
#define END_OF_LLDPDU_TLV_TYPE 0x00
|
||||
#define CHASSIS_ID_TLV_TYPE 0x01
|
||||
#define PORT_ID_TLV_TYPE 0x02
|
||||
#define TIME_TO_LIVE_TLV_TYPE 0x03
|
||||
|
||||
static char *lldp_parse_port_id(uint16_t n_byte, uint8_t *bytes)
|
||||
{
|
||||
struct lldp_tvl *tvl = (struct lldp_tvl *) bytes;
|
||||
|
||||
if (n_byte < sizeof(struct lldp_tvl) + tvl->length) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if (tvl->type == PORT_ID_TLV_TYPE) {
|
||||
static __thread char port_id[NEU_NODE_NAME_LEN] = { 0 };
|
||||
uint8_t * data = (uint8_t *) &tvl[1];
|
||||
|
||||
strncpy(port_id, (char *) data + 1, tvl->length - 1);
|
||||
return port_id;
|
||||
} else {
|
||||
return lldp_parse_port_id(
|
||||
n_byte - sizeof(struct lldp_tvl) - tvl->length,
|
||||
bytes + sizeof(struct lldp_tvl) + tvl->length);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user