47 lines
1.4 KiB
C
47 lines
1.4 KiB
C
#include "ether.h"
|
|
#include "ipv4.h"
|
|
#include "socket.h"
|
|
#include "udp.h"
|
|
#include "string.h"
|
|
#include "memory.h"
|
|
#include "pvec.h"
|
|
|
|
void udp_process_packet(struct ether_t *ether, unsigned int dest, unsigned int src, struct udp_header_t *packet, unsigned int len) {
|
|
|
|
struct socket_t *sock = socket_find(htons(packet->destination_port), htons(packet->source_port), htonl(src), 0, 0);
|
|
|
|
if (!sock || sock->socket_type != 17) {
|
|
return;
|
|
}
|
|
|
|
struct udp_data_t *data = (struct udp_data_t *)malloc(sizeof(struct udp_data_t));
|
|
if (!data) {
|
|
return;
|
|
}
|
|
|
|
int ulen = packet->length - sizeof(struct udp_header_t);
|
|
|
|
data->data = (unsigned char *)malloc(ulen);
|
|
if (!data->data) {
|
|
free(data);
|
|
return;
|
|
}
|
|
memcpy(data->data, packet->payload, ulen);
|
|
data->from = src;
|
|
data->len = ulen;
|
|
|
|
ptr_vector_append(&sock->packets, data);
|
|
}
|
|
|
|
void udp_send(struct ether_t *ether, struct socket_t *sock, unsigned char *packet, unsigned int len) {
|
|
struct udp_header_t *header = (struct udp_header_t *)malloc(len + sizeof(struct udp_header_t));
|
|
|
|
header->source_port = htons(sock->port_recv);
|
|
header->destination_port = htons(sock->port_dest);
|
|
header->length = htons((unsigned short)len + sizeof(struct udp_header_t));
|
|
header->checksum = 0;
|
|
|
|
memcpy(header->payload, packet, len);
|
|
ipv4_send(ether, 0x11, htonl(sock->addr), (char *)header, len + sizeof(struct udp_header_t));
|
|
}
|