2006-04-16 21:37:00 +02:00
|
|
|
/***************************************************************************
|
|
|
|
* Copyright (C) 04/2006 by Olaf Rempel *
|
|
|
|
* razzor@kopf-tisch.de *
|
|
|
|
* *
|
|
|
|
* This program is free software; you can redistribute it and/or modify *
|
|
|
|
* it under the terms of the GNU General Public License as published by *
|
|
|
|
* the Free Software Foundation; either version 2 of the License, or *
|
|
|
|
* (at your option) any later version. *
|
|
|
|
* *
|
|
|
|
* This program is distributed in the hope that it will be useful, *
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
|
|
* GNU General Public License for more details. *
|
|
|
|
* *
|
|
|
|
* You should have received a copy of the GNU General Public License *
|
|
|
|
* along with this program; if not, write to the *
|
|
|
|
* Free Software Foundation, Inc., *
|
|
|
|
* 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
|
|
|
|
***************************************************************************/
|
2006-02-02 16:55:44 +01:00
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <unistd.h>
|
2006-03-07 20:30:11 +01:00
|
|
|
|
|
|
|
// needed for string.h / memmem()
|
|
|
|
#ifndef __USE_GNU
|
|
|
|
#define __USE_GNU
|
|
|
|
#endif
|
|
|
|
|
2006-02-02 16:55:44 +01:00
|
|
|
#include <string.h>
|
|
|
|
#include <sys/types.h>
|
|
|
|
#include <sys/socket.h>
|
|
|
|
#include <sys/ioctl.h>
|
|
|
|
#include <net/if.h>
|
|
|
|
#include <netinet/in.h>
|
|
|
|
#include <arpa/inet.h>
|
2006-03-06 20:13:26 +01:00
|
|
|
#include <ctype.h>
|
2006-02-02 16:55:44 +01:00
|
|
|
|
|
|
|
#include "netpkt.h"
|
2006-04-16 21:02:41 +02:00
|
|
|
#include "logging.h"
|
2006-02-02 16:55:44 +01:00
|
|
|
|
2006-03-05 02:28:19 +01:00
|
|
|
NetPkt::NetPkt(const char* data, int size)
|
2006-03-06 20:13:26 +01:00
|
|
|
: data(data), size(size), alloc(0)
|
2006-02-02 16:55:44 +01:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2006-03-06 20:13:26 +01:00
|
|
|
NetPkt::NetPkt(int alloc)
|
|
|
|
: size(0), alloc(alloc)
|
2006-02-02 16:55:44 +01:00
|
|
|
{
|
2006-03-06 20:13:26 +01:00
|
|
|
data = new char[alloc];
|
2006-02-02 16:55:44 +01:00
|
|
|
}
|
|
|
|
|
2006-03-05 02:28:19 +01:00
|
|
|
NetPkt::~NetPkt()
|
2006-02-02 16:55:44 +01:00
|
|
|
{
|
2006-03-06 20:13:26 +01:00
|
|
|
if (alloc > 0)
|
2006-04-16 21:28:27 +02:00
|
|
|
delete [] data;
|
2006-02-02 16:55:44 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
int NetPkt::show(char* buf, int size)
|
|
|
|
{
|
2006-03-05 02:28:19 +01:00
|
|
|
return snprintf(buf, size, "(%s:%d) (%d/%d) bytes",
|
2006-02-02 16:55:44 +01:00
|
|
|
inet_ntoa(this->addr.sin_addr),
|
|
|
|
ntohs(this->addr.sin_port),
|
2006-03-06 20:13:26 +01:00
|
|
|
this->size, this->alloc);
|
2006-02-20 12:31:34 +01:00
|
|
|
}
|
|
|
|
|
2006-03-06 20:13:26 +01:00
|
|
|
bool NetPkt::compare(unsigned int offset, const char* buf, unsigned int len)
|
2006-02-20 12:31:34 +01:00
|
|
|
{
|
2006-03-06 20:13:26 +01:00
|
|
|
if (offset + len >= this->size)
|
2006-02-20 12:31:34 +01:00
|
|
|
return false;
|
|
|
|
|
|
|
|
return (memcmp(this->data + offset, buf, len) == 0);
|
|
|
|
}
|
2006-02-20 21:58:59 +01:00
|
|
|
|
2006-03-06 20:13:26 +01:00
|
|
|
int NetPkt::find(unsigned int offset, const char *buf, unsigned int len)
|
2006-02-20 21:58:59 +01:00
|
|
|
{
|
2006-03-06 20:13:26 +01:00
|
|
|
if (offset >= this->size)
|
2006-02-20 21:58:59 +01:00
|
|
|
return -1;
|
|
|
|
|
2006-03-06 20:13:26 +01:00
|
|
|
void* found = memmem(this->data + offset, this->size, buf, len);
|
2006-03-05 02:28:19 +01:00
|
|
|
|
2006-02-20 21:58:59 +01:00
|
|
|
return (found == NULL) ? -1 : ((char*)found - this->data);
|
|
|
|
}
|
2006-03-05 02:28:19 +01:00
|
|
|
|
|
|
|
void NetPkt::setAddress(struct sockaddr_in *tmp)
|
|
|
|
{
|
|
|
|
memcpy(&addr, tmp, sizeof(addr));
|
|
|
|
}
|
|
|
|
|
|
|
|
struct sockaddr_in * NetPkt::getAddress()
|
|
|
|
{
|
|
|
|
return &addr;
|
|
|
|
}
|
2006-03-06 20:13:26 +01:00
|
|
|
|
|
|
|
int NetPkt::parse_int(unsigned int offset, int *val)
|
|
|
|
{
|
|
|
|
const char *max = this->data + this->size;
|
|
|
|
const char *c = this->data + offset;
|
|
|
|
|
|
|
|
/* untere grenze abtesten */
|
|
|
|
if (this->data > c || c > max)
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
*val = 0;
|
|
|
|
|
|
|
|
/* ziffern einlesen */
|
|
|
|
while (isdigit(*c) && c < max)
|
|
|
|
*val = (*val * 10) + (*c++ - 0x30);
|
|
|
|
|
|
|
|
return (c - (this->data + offset));
|
|
|
|
}
|
|
|
|
|
|
|
|
int NetPkt::parse_ip(unsigned int offset, struct in_addr *ip)
|
|
|
|
{
|
|
|
|
int i, tmp, count, pos = offset;
|
|
|
|
ip->s_addr = 0;
|
|
|
|
|
|
|
|
for (i = 0; i < 4; i++) {
|
|
|
|
count = this->parse_int(pos, &tmp);
|
|
|
|
pos += count;
|
|
|
|
if (count == 0 || tmp < 0 || tmp > 255)
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
ip->s_addr = ip->s_addr>>8 | tmp<<24;
|
|
|
|
|
|
|
|
if (i != 3 && this->data[pos++] != '.')
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
return pos - offset;
|
|
|
|
}
|
|
|
|
|
|
|
|
int NetPkt::getPort()
|
|
|
|
{
|
|
|
|
return ntohs(addr.sin_port);
|
|
|
|
}
|
|
|
|
|
|
|
|
char* NetPkt::showfull()
|
|
|
|
{
|
|
|
|
unsigned int pos = 0, i = 0, j;
|
|
|
|
char *buf = new char[this->size * 4 + 64];
|
|
|
|
|
|
|
|
while (pos < this->size) {
|
|
|
|
i += sprintf(buf + i, "%04X: ", pos);
|
|
|
|
for (j = 0; j < 16; j++) {
|
|
|
|
if (pos + j < this->size)
|
|
|
|
i += sprintf(buf + i, "%02X", (unsigned char)this->data[pos + j]);
|
|
|
|
else
|
|
|
|
i += sprintf(buf + i, " ");
|
|
|
|
|
|
|
|
if (j % 2)
|
|
|
|
buf[i++] = ' ';
|
|
|
|
}
|
|
|
|
|
|
|
|
for (j = 0; j < 16; j++) {
|
|
|
|
if (pos + j < this->size) {
|
|
|
|
unsigned char val = this->data[pos + j];
|
|
|
|
if (val >= 0x20 && val < 0x80)
|
|
|
|
buf[i++] = val;
|
|
|
|
else
|
|
|
|
buf[i++] = '.';
|
|
|
|
} else {
|
|
|
|
buf[i++] = ' ';
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pos += 16;
|
|
|
|
buf[i++] = '\r';
|
|
|
|
buf[i++] = '\n';
|
|
|
|
}
|
|
|
|
buf[i] = 0;
|
|
|
|
return buf;
|
|
|
|
}
|
|
|
|
|
|
|
|
int NetPkt::getSize()
|
|
|
|
{
|
|
|
|
return size;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool NetPkt::sameAddress(NetPkt* pkt)
|
|
|
|
{
|
|
|
|
return (this->addr.sin_addr.s_addr == pkt->addr.sin_addr.s_addr) &&
|
|
|
|
(this->addr.sin_port == pkt->addr.sin_port);
|
|
|
|
}
|
|
|
|
|
2006-04-15 19:55:07 +02:00
|
|
|
void NetPkt::merge(NetPkt* pkt)
|
2006-03-06 20:13:26 +01:00
|
|
|
{
|
|
|
|
unsigned int new_alloc = size + pkt->size;
|
|
|
|
|
|
|
|
char* new_data = new char[new_alloc];
|
|
|
|
memcpy(new_data, data, size);
|
|
|
|
|
|
|
|
if (alloc)
|
2006-04-16 21:28:27 +02:00
|
|
|
delete [] data;
|
2006-03-06 20:13:26 +01:00
|
|
|
|
|
|
|
data = new_data;
|
|
|
|
alloc = new_alloc;
|
|
|
|
|
|
|
|
memcpy((void*)(data + size), pkt->data, pkt->size);
|
|
|
|
size += pkt->size;
|
|
|
|
}
|
2006-04-15 19:55:07 +02:00
|
|
|
|
|
|
|
bool NetPkt::append(const char* buf, unsigned int len)
|
|
|
|
{
|
|
|
|
if (alloc < size + len)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
memcpy((void*)(data + size), buf, len);
|
|
|
|
size += len;
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|