BadVPN – Rev 1
?pathlinks?
/**
* @file addr.h
* @author Ambroz Bizjak <ambrop7@gmail.com>
*
* @section LICENSE
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. Neither the name of the author nor the
* names of its contributors may be used to endorse or promote products
* derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
* @section DESCRIPTION
*
* AddrProto, a protocol for encoding network addresses.
*
* AddrProto is built with BProto, the protocol and code generator for building
* custom message protocols. The BProto specification file is addr.bproto.
*/
#ifndef BADVPN_PROTOCOL_ADDR_H
#define BADVPN_PROTOCOL_ADDR_H
#include <stdint.h>
#include <string.h>
#include <misc/debug.h>
#include <system/BAddr.h>
#include <base/BLog.h>
#include <generated/bproto_addr.h>
#include <generated/blog_channel_addr.h>
#define ADDR_TYPE_IPV4 1
#define ADDR_TYPE_IPV6 2
#define ADDR_SIZE_IPV4 (addr_SIZEtype + addr_SIZEip_port + addr_SIZEipv4_addr)
#define ADDR_SIZE_IPV6 (addr_SIZEtype + addr_SIZEip_port + addr_SIZEipv6_addr)
/**
* Determines if the given address is supported by AddrProto.
* Depends only on the type of the address.
*
* @param addr address to check. Must be recognized according to {@link BAddr_IsRecognized}.
* @return 1 if supported, 0 if not
*/
static int addr_supported (BAddr addr);
/**
* Determines the size of the given address when encoded by AddrProto.
* Depends only on the type of the address.
*
* @param addr address to check. Must be supported according to {@link addr_supported}.
* @return encoded size
*/
static int addr_size (BAddr addr);
/**
* Encodes an address according to AddrProto.
*
* @param out output buffer. Must have at least addr_size(addr) space.
* @param addr address to encode. Must be supported according to {@link addr_supported}.
*/
static void addr_write (uint8_t *out, BAddr addr);
/**
* Decodes an address according to AddrProto.
*
* @param data input buffer containing the address to decode
* @param data_len size of input. Must be >=0.
* @param out_addr the decoded address will be stored here on success
* @return 1 on success, 0 on failure
*/
static int addr_read (uint8_t *data, int data_len, BAddr *out_addr) WARN_UNUSED;
int addr_supported (BAddr addr)
{
BAddr_Assert(&addr);
switch (addr.type) {
case BADDR_TYPE_IPV4:
case BADDR_TYPE_IPV6:
return 1;
default:
return 0;
}
}
int addr_size (BAddr addr)
{
ASSERT(addr_supported(addr))
switch (addr.type) {
case BADDR_TYPE_IPV4:
return ADDR_SIZE_IPV4;
case BADDR_TYPE_IPV6:
return ADDR_SIZE_IPV6;
default:
ASSERT(0)
return 0;
}
}
void addr_write (uint8_t *out, BAddr addr)
{
ASSERT(addr_supported(addr))
addrWriter writer;
addrWriter_Init(&writer, out);
switch (addr.type) {
case BADDR_TYPE_IPV4: {
addrWriter_Addtype(&writer, ADDR_TYPE_IPV4);
uint8_t *out_port = addrWriter_Addip_port(&writer);
memcpy(out_port, &addr.ipv4.port, sizeof(addr.ipv4.port));
uint8_t *out_addr = addrWriter_Addipv4_addr(&writer);
memcpy(out_addr, &addr.ipv4.ip, sizeof(addr.ipv4.ip));
} break;
case BADDR_TYPE_IPV6: {
addrWriter_Addtype(&writer, ADDR_TYPE_IPV6);
uint8_t *out_port = addrWriter_Addip_port(&writer);
memcpy(out_port, &addr.ipv6.port, sizeof(addr.ipv6.port));
uint8_t *out_addr = addrWriter_Addipv6_addr(&writer);
memcpy(out_addr, addr.ipv6.ip, sizeof(addr.ipv6.ip));
} break;
default:
ASSERT(0);
}
int len = addrWriter_Finish(&writer);
B_USE(len)
ASSERT(len == addr_size(addr))
}
int addr_read (uint8_t *data, int data_len, BAddr *out_addr)
{
ASSERT(data_len >= 0)
addrParser parser;
if (!addrParser_Init(&parser, data, data_len)) {
BLog(BLOG_ERROR, "failed to parse addr");
return 0;
}
uint8_t type = 0; // to remove warning
addrParser_Gettype(&parser, &type);
switch (type) {
case ADDR_TYPE_IPV4: {
uint8_t *port_data;
if (!addrParser_Getip_port(&parser, &port_data)) {
BLog(BLOG_ERROR, "port missing for IPv4 address");
return 0;
}
uint8_t *addr_data;
if (!addrParser_Getipv4_addr(&parser, &addr_data)) {
BLog(BLOG_ERROR, "address missing for IPv4 address");
return 0;
}
uint16_t port;
uint32_t addr;
memcpy(&port, port_data, sizeof(port));
memcpy(&addr, addr_data, sizeof(addr));
BAddr_InitIPv4(out_addr, addr, port);
} break;
case ADDR_TYPE_IPV6: {
uint8_t *port_data;
if (!addrParser_Getip_port(&parser, &port_data)) {
BLog(BLOG_ERROR, "port missing for IPv6 address");
return 0;
}
uint8_t *addr_data;
if (!addrParser_Getipv6_addr(&parser, &addr_data)) {
BLog(BLOG_ERROR, "address missing for IPv6 address");
return 0;
}
uint16_t port;
memcpy(&port, port_data, sizeof(port));
BAddr_InitIPv6(out_addr, addr_data, port);
} break;
default:
BLog(BLOG_ERROR, "unknown address type %d", (int)type);
return 0;
}
return 1;
}
#endif