#include <sys/types.h>
#include <netinet/in_systm.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <netinet/udp.h>
#include "alias_local.h"
struct cu_header {
u_int16_t dest_family;
u_int16_t dest_port;
u_int32_t dest_addr;
int16_t family;
u_int16_t port;
u_int32_t addr;
u_int32_t seq;
u_int16_t msg;
u_int16_t data_type;
u_int16_t packet_len;
};
struct oc_header {
u_int16_t client_count;
u_int32_t seq_no;
char user_name[20];
char reserved[4];
};
struct client_info {
u_int32_t address;
char reserved[8];
};
void
AliasHandleCUSeeMeOut(struct ip *pip, struct alias_link *link)
{
struct udphdr *ud;
ud = (struct udphdr *)((char *)pip + (pip->ip_hl << 2));
if(ud->uh_ulen >= sizeof(struct cu_header)) {
struct cu_header *cu;
struct alias_link *cu_link;
cu = (struct cu_header *)(ud + 1);
if (cu->addr)
cu->addr = (u_int32_t)GetAliasAddress(link).s_addr;
cu_link = FindUdpTcpOut(pip->ip_src, GetDestAddress(link),
ud->uh_dport, 0, IPPROTO_UDP);
#ifndef NO_FW_PUNCH
if (cu_link)
PunchFWHole(cu_link);
#endif
}
}
void
AliasHandleCUSeeMeIn(struct ip *pip, struct in_addr original_addr)
{
struct in_addr alias_addr;
struct udphdr *ud;
struct cu_header *cu;
struct oc_header *oc;
struct client_info *ci;
char *end;
int i;
alias_addr.s_addr = pip->ip_dst.s_addr;
ud = (struct udphdr *)((char *)pip + (pip->ip_hl << 2));
cu = (struct cu_header *)(ud + 1);
oc = (struct oc_header *)(cu + 1);
ci = (struct client_info *)(oc + 1);
end = (char *)cu + ud->uh_ulen;
if ((char *)oc <= end) {
if(cu->dest_addr)
cu->dest_addr = (u_int32_t)original_addr.s_addr;
if(ntohs(cu->data_type) == 101)
for(i = 0; (char *)(ci + 1) <= end && i < oc->client_count; i++, ci++)
if(ci->address == (u_int32_t)alias_addr.s_addr) {
ci->address = (u_int32_t)original_addr.s_addr;
break;
}
}
}