3rd-party/libtins
1
0
Fork 0
mirror of https://github.com/mfontanini/libtins synced 2026-01-25 11:41:35 +01:00
Code Issues Releases Wiki Activity

Added recv mechanism on PacketSender.

Browse Source
This commit is contained in:
Matias Fontanini
2011-08-15 12:39:46 -03:00
parent c803e25db4
commit 5c412208f9
7 changed files with 201 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
src/packetsender.cpp
View File

@@ -24,21 +24,23 @@
#include <sys/select.h>
#include <arpa/inet.h>
#include <unistd.h>
#include </usr/include/linux/if_ether.h>
#include </usr/include/linux/if_packet.h>
#include <linux/if_ether.h>
#include <linux/if_packet.h>
#include <netdb.h>
#endif
#include <assert.h>
#include <iostream>
#include <iostream> //borrar
#include <errno.h>
#include <string.h>
#include "packetsender.h"
#include "utils.h" //borrar
const int Tins::PacketSender::INVALID_RAW_SOCKET = -10;
Tins::PacketSender::PacketSender() : _sockets(SOCKETS_END, INVALID_RAW_SOCKET) {
_types[IP_SOCKET] = IPPROTO_RAW;
_types[ICMP_SOCKET] = IPPROTO_ICMP;
}
bool Tins::PacketSender::open_l2_socket() {
@@ -48,18 +50,21 @@ bool Tins::PacketSender::open_l2_socket() {
return false;
}
bool Tins::PacketSender::open_l3_socket() {
if(_sockets[IP_SOCKET] != INVALID_RAW_SOCKET)
bool Tins::PacketSender::open_l3_socket(SocketType type) {
int socktype = find_type(type);
if(socktype == -1)
return false;
if(_sockets[type] != INVALID_RAW_SOCKET)
return true;
int sockfd;
sockfd = socket(AF_INET, SOCK_RAW, IPPROTO_RAW);
sockfd = socket(AF_INET, SOCK_RAW, socktype);
if (sockfd < 0)
return false;
const int on = 1;
setsockopt(sockfd, IPPROTO_IP,IP_HDRINCL,(const void *)&on,sizeof(on));
setsockopt(sockfd, IPPROTO_IP, IP_HDRINCL,(const void *)&on,sizeof(on));
_sockets[IP_SOCKET] = sockfd;
_sockets[type] = sockfd;
return true;
}
@@ -75,6 +80,12 @@ bool Tins::PacketSender::send(PDU *pdu) {
return pdu->send(this);
}
Tins::PDU *Tins::PacketSender::send_recv(PDU *pdu) {
if(!pdu->send(this))
return 0;
return pdu->recv_response(this);
}
bool Tins::PacketSender::send_l2(PDU *pdu) {
/* To be implemented */
@@ -82,18 +93,42 @@ bool Tins::PacketSender::send_l2(PDU *pdu) {
return false;
}
bool Tins::PacketSender::send_l3(PDU *pdu, const struct sockaddr* link_addr, uint32_t len_link_addr) {
Tins::PDU *Tins::PacketSender::recv_l3(PDU *pdu, struct sockaddr *link_addr, uint32_t len_link_addr, SocketType type) {
if(!open_l3_socket(type))
return 0;
uint8_t buffer[2048];
int sock = _sockets[type];
bool done = false;
socklen_t addrlen = len_link_addr;
while(!done) {
ssize_t size = recvfrom(sock, buffer, 2048, 0, link_addr, &addrlen);
if(size == -1)
return 0;
if(pdu->matches_response(buffer, size))
return pdu->clone_packet(buffer, size);
}
return 0;
}
bool Tins::PacketSender::send_l3(PDU *pdu, struct sockaddr* link_addr, uint32_t len_link_addr, SocketType type) {
bool ret_val = true;
if(!open_l3_socket())
if(!open_l3_socket(type))
ret_val = false;
if (ret_val) {
uint32_t sz;
int sock = _sockets[IP_SOCKET];
int sock = _sockets[type];
uint8_t *buffer = pdu->serialize(sz);
ret_val = (sendto(sock, buffer, sz, 0, link_addr, len_link_addr) != -1);
delete[] buffer;
}
return ret_val;
}
int Tins::PacketSender::find_type(SocketType type) {
SocketTypeMap::iterator it = _types.find(type);
if(it == _types.end())
return -1;
else
return it->second;
}