1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
|
/**
* @file packet.c
*
* @brief UDP-Packet, contains data, sender and receiver.
*
*/
/*
* Copyright (C) 2005 Jan Hutter, Martin Willi
* Hochschule fuer Technik Rapperswil
*
* 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. See <http://www.fsf.org/copyleft/gpl.txt>.
*
* 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.
*/
#include "packet.h"
#include "utils/allocator.h"
/**
* Private data of an packet_t object
*/
typedef struct private_packet_s private_packet_t;
struct private_packet_s {
/**
* Public part of a packet_t object
*/
packet_t public;
/* private functions */
/**
* @brief helper function to set address used by set_dest & set_source.
*
* @param this calling object_t
* @param family address family
* @param saddr source address
* @param address address as string
* @return
* - SUCCESS if successfuly
* - NOT_SUPPORTED if family is not supported
*/
status_t (*set_addr) (private_packet_t *this, int family, struct sockaddr *saddr, char *address, u_int16_t port);
};
/**
* Implements packet_t's destroy function.
* See #packet_s.destroy for description.
*/
static status_t destroy(private_packet_t *this)
{
if (this->public.data.ptr != NULL)
{
allocator_free(this->public.data.ptr);
}
allocator_free(this);
return SUCCESS;
}
/**
* Implements packet_t's clone function.
* See #packet_s.clone for description.
*/
static status_t clone (private_packet_t *packet, packet_t **clone)
{
*clone = packet_create(packet->public.family);
if ((*clone) == NULL)
{
return OUT_OF_RES;
}
(*clone)->sockaddr_len = packet->public.sockaddr_len;
(*clone)->source = packet->public.source;
(*clone)->destination = packet->public.destination;
/* only clone existing chunks :-) */
if (packet->public.data.ptr != NULL)
{
(*clone)->data.ptr = allocator_clone_bytes(packet->public.data.ptr,packet->public.data.len);
if ((*clone)->data.ptr == NULL)
{
(*clone)->destroy((*clone));
return OUT_OF_RES;
}
(*clone)->data.len = packet->public.data.len;
}
return SUCCESS;
}
/**
* Implements private_packet_t's set_addr function.
* See #private_packet_t.set_addr for description.
*/
static status_t set_addr(int family, struct sockaddr *saddr, char *address, u_int16_t port)
{
switch (family)
{
/* IPv4 */
case AF_INET:
{
struct sockaddr_in *sin = (struct sockaddr_in*)saddr;
sin->sin_family = AF_INET;
sin->sin_addr.s_addr = inet_addr("127.0.0.1");
sin->sin_port = htons(port);
return SUCCESS;;
}
}
return NOT_SUPPORTED;
}
/**
* Implements packet_t's set_destination function.
* See #packet_t.set_destination for description.
*/
static status_t set_destination(packet_t *this, char *address, u_int16_t port)
{
struct sockaddr *saddr = &(this->destination);
return set_addr(this->family, saddr, address, port);
}
/**
* Implements packet_t's set_source function.
* See #packet_t.set_source for description.
*/
static status_t set_source(packet_t *this, char *address, u_int16_t port)
{
struct sockaddr *saddr = &(this->source);
return set_addr(this->family, saddr, address, port);
}
/*
* Documented in header
*/
packet_t *packet_create(int family)
{
private_packet_t *this = allocator_alloc_thing(private_packet_t);
this->public.destroy = (status_t(*) (packet_t *)) destroy;
this->public.set_destination = set_destination;
this->public.set_source = set_source;
this->public.clone = (status_t(*) (packet_t *,packet_t**))clone;
this->public.family = family;
switch (family)
{
case AF_INET:
this->public.sockaddr_len = sizeof(struct sockaddr_in);
break;
default: /* not supported */
allocator_free(this);
return NULL;
}
this->public.data.len = 0;
this->public.data.ptr = NULL;
return &(this->public);
}
|
