Bochs/bochs/misc/bxhub.cc
2017-03-08 22:09:50 +00:00

516 lines
15 KiB
C++

/////////////////////////////////////////////////////////////////////////
// $Id$
/////////////////////////////////////////////////////////////////////////
/*
* Copyright (C) 2003 by Mariusz Matuszek [NOmrmmSPAM @ users.sourceforge.net]
* Copyright (C) 2017 The Bochs Project
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library 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
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/
// bxhub.c: a simple, two-port software 'ethernet hub' for use with
// eth_socket Bochs ethernet pktmover.
// Extensions by Volker Ruppert (2017):
// - Windows support added.
// - Integrated 'vnet' server features (ARP, ICMP-echo, DHCP and TFTP).
// - Command line options added for base UDP port and 'vnet' server features.
// - Support for connects from up to 6 Bochs sessions.
// - Support for connecting from other machines.
#include "config.h"
extern "C" {
#ifdef WIN32
#include <winsock2.h>
#else
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#define closesocket(s) close(s)
typedef int SOCKET;
#endif
#include <signal.h>
};
#ifndef BXHUB
#define BXHUB
#endif
#ifdef WIN32
#define MSG_NOSIGNAL 0
#define MSG_DONTWAIT 0
#endif
#include "misc/bxcompat.h"
#include "osdep.h"
#include "iodev/network/netmod.h"
#define BXHUB_MAX_CLIENTS 6
typedef struct {
SOCKET so;
struct sockaddr_in sin, sout;
bx_bool init;
dhcp_cfg_t dhcp;
Bit8u *reply_buffer;
int pending_reply_size;
} hub_client_t;
const Bit8u default_host_macaddr[6] = {0xb0, 0xc4, 0x20, 0x00, 0x00, 0x0f};
const Bit8u default_host_ipv4addr[4] = {10, 0, 2, 2};
const Bit8u broadcast_ipv4addr[3][4] =
{
{ 0, 0, 0, 0},
{255,255,255,255},
{ 10, 0, 2,255},
};
static Bit8u default_guest_ipv4addr[4] = {10, 0, 2, 15};
static Bit16u port_base = 40000;
static char tftp_root[BX_PATHNAME_LEN];
static Bit8u host_macaddr[6];
static int client_max;
static hub_client_t hclient[BXHUB_MAX_CLIENTS];
bx_bool handle_ipv4(hub_client_t *client, Bit8u *buf, unsigned len)
{
unsigned total_len;
unsigned fragment_flags;
unsigned fragment_offset;
unsigned l3header_len;
const Bit8u *l4pkt;
unsigned l4pkt_len;
unsigned udp_src_port;
unsigned udp_dst_port;
unsigned udp_reply_size = 0;
unsigned icmptype;
unsigned icmpcode;
Bit8u *replybuf = client->reply_buffer;
dhcp_cfg_t *dhcpc = &client->dhcp;
// guest-to-host IPv4
if (len < (14U+20U)) {
return 0;
}
ip_header_t *iphdr = (ip_header_t *)((Bit8u *)buf +
sizeof(ethernet_header_t));
if (iphdr->version != 4) {
return 0;
}
l3header_len = (iphdr->header_len << 2);
if (l3header_len != 20) {
return 0;
}
if (ip_checksum((Bit8u*)iphdr, l3header_len) != (Bit16u)0xffff) {
return 0;
}
total_len = ntohs(iphdr->total_len);
if (memcmp(&iphdr->dst_addr, dhcpc->host_ipv4addr, 4) &&
memcmp(&iphdr->dst_addr, broadcast_ipv4addr[0],4) &&
memcmp(&iphdr->dst_addr, broadcast_ipv4addr[1],4) &&
memcmp(&iphdr->dst_addr, broadcast_ipv4addr[2],4))
{
return 0;
}
fragment_flags = ntohs(iphdr->frag_offs) >> 13;
fragment_offset = (ntohs(iphdr->frag_offs) & 0x1fff) << 3;
if ((fragment_flags & 0x1) || (fragment_offset != 0)) {
return 0;
}
l4pkt = &buf[14 + l3header_len];
l4pkt_len = total_len - l3header_len;
if (iphdr->protocol == 0x11) {
// guest-to-host UDP IPv4
if (l4pkt_len < 8) return 0;
udp_header_t *udphdr = (udp_header_t *)l4pkt;
udp_src_port = ntohs(udphdr->src_port);
udp_dst_port = ntohs(udphdr->dst_port);
switch (udp_dst_port) {
case 67: // BOOTP
udp_reply_size = vnet_process_dhcp(NULL, &l4pkt[8], l4pkt_len-8,
&replybuf[42], dhcpc);
break;
case 69: // TFTP
if (strlen(tftp_root) > 0) {
udp_reply_size = vnet_process_tftp(NULL, &l4pkt[8], l4pkt_len-8,
udp_src_port, &replybuf[42], tftp_root);
}
break;
default:
break;
}
if (udp_reply_size > 0) {
// host-to-guest UDP IPv4: pseudo-header
replybuf[22] = 0;
replybuf[23] = 0x11; // UDP
put_net2(&replybuf[24], 8U+udp_reply_size);
memcpy(&replybuf[26], dhcpc->host_ipv4addr, 4);
memcpy(&replybuf[30], dhcpc->guest_ipv4addr, 4);
// udp header
put_net2(&replybuf[34], udp_dst_port);
put_net2(&replybuf[36], udp_src_port);
put_net2(&replybuf[38], 8U+udp_reply_size);
put_net2(&replybuf[40], 0);
put_net2(&replybuf[40],
ip_checksum(&replybuf[22], 12U+8U+udp_reply_size) ^ (Bit16u)0xffff);
// ip header
memset(&replybuf[14], 0, 20);
replybuf[14] = 0x45;
replybuf[15] = 0x00;
put_net2(&replybuf[16], 20U+8U+udp_reply_size);
put_net2(&replybuf[18], 1);
replybuf[20] = 0x00;
replybuf[21] = 0x00;
replybuf[22] = 0x07; // TTL
replybuf[23] = 0x11; // UDP
client->pending_reply_size = udp_reply_size + 42;
}
} else if (iphdr->protocol == 0x01) {
// guest-to-host ICMP
if (l4pkt_len < 8) return 0;
icmptype = l4pkt[0];
icmpcode = l4pkt[1];
if (ip_checksum(l4pkt, l4pkt_len) != (Bit16u)0xffff) {
return 0;
}
switch (icmptype) {
case 0x08: // ECHO
if (icmpcode == 0) {
if (vnet_process_icmp_echo(&buf[14], l3header_len, l4pkt, l4pkt_len,
replybuf)) {
client->pending_reply_size = 14U+l3header_len+l4pkt_len;
}
}
break;
default:
break;
}
}
if (client->pending_reply_size > 0) {
// host-to-guest IPv4
replybuf[14] = (replybuf[14] & 0x0f) | 0x40;
l3header_len = ((unsigned)(replybuf[14] & 0x0f) << 2);
memcpy(&replybuf[26], dhcpc->host_ipv4addr, 4);
memcpy(&replybuf[30], dhcpc->guest_ipv4addr, 4);
put_net2(&replybuf[24], 0);
put_net2(&replybuf[24], ip_checksum(&replybuf[14], l3header_len) ^ (Bit16u)0xffff);
vnet_prepare_reply(replybuf, ETHERNET_TYPE_IPV4, dhcpc);
// don't forward to other client
return 1;
} else {
return 0;
}
}
bx_bool handle_arp(hub_client_t *client, Bit8u *buf, unsigned len)
{
dhcp_cfg_t *dhcpc = &client->dhcp;
arp_header_t *arphdr = (arp_header_t *)((Bit8u *)buf +
sizeof(ethernet_header_t));
if ((ntohs(arphdr->hw_addr_space) != 0x0001) ||
(ntohs(arphdr->proto_addr_space) != 0x0800) ||
(arphdr->hw_addr_len != ETHERNET_MAC_ADDR_LEN) ||
(arphdr->proto_addr_len != 4)) {
return 0;
}
switch(ntohs(arphdr->opcode)) {
case ARP_OPCODE_REQUEST:
if (vnet_process_arp_request(buf, client->reply_buffer, dhcpc)) {
client->pending_reply_size = MIN_RX_PACKET_LEN;
vnet_prepare_reply(client->reply_buffer, ETHERNET_TYPE_ARP, dhcpc);
return 1;
}
break;
case ARP_OPCODE_REPLY:
case ARP_OPCODE_REV_REQUEST:
case ARP_OPCODE_REV_REPLY:
default:
break;
}
return 0;
}
int handle_packet(hub_client_t *client, Bit8u *buf, unsigned len)
{
ethernet_header_t *ethhdr = (ethernet_header_t *)buf;
dhcp_cfg_t *dhcpc = &client->dhcp;
int ret = 1;
if (!client->init) {
if (memcmp(ethhdr->src_mac_addr, host_macaddr, 6) == 0) {
client->init = -1;
} else {
client->sout.sin_addr.s_addr = client->sin.sin_addr.s_addr;
memcpy(dhcpc->host_macaddr, host_macaddr, ETHERNET_MAC_ADDR_LEN);
memcpy(dhcpc->guest_macaddr, ethhdr->src_mac_addr, ETHERNET_MAC_ADDR_LEN);
memcpy(dhcpc->host_ipv4addr, &default_host_ipv4addr[0], 4);
memcpy(dhcpc->guest_ipv4addr, &broadcast_ipv4addr[1][0], 4);
memcpy(dhcpc->default_guest_ipv4addr, default_guest_ipv4addr, 4);
default_guest_ipv4addr[3]++;
client->reply_buffer = new Bit8u[BX_PACKET_BUFSIZE];
client->init = 1;
}
}
if (client->pending_reply_size > 0)
return 0;
switch (ntohs(ethhdr->type)) {
case ETHERNET_TYPE_IPV4:
ret = handle_ipv4(client, buf, len);
break;
case ETHERNET_TYPE_ARP:
ret = handle_arp(client, buf, len);
break;
default:
break;
}
return ret;
}
void send_packet(hub_client_t *client, Bit8u *buf, unsigned len)
{
sendto(client->so, (char*)buf, len, (MSG_NOSIGNAL|MSG_DONTWAIT),
(struct sockaddr*) &client->sout, sizeof(client->sout));
}
void broadcast_packet(int clientid, Bit8u *buf, unsigned len)
{
handle_packet(&hclient[clientid], buf, len);
for (int i = 0; i < client_max; i++) {
if (i != clientid) {
send_packet(&hclient[i], buf, len);
}
}
}
bx_bool find_client(Bit8u *dst_mac_addr, int *clientid)
{
*clientid = -1;
for (int i = 0; i < client_max; i++) {
if (memcmp(dst_mac_addr, hclient[i].dhcp.guest_macaddr, ETHERNET_MAC_ADDR_LEN) == 0) {
*clientid = i;
}
}
return (*clientid >= 0);
}
void print_usage()
{
fprintf(stderr,
"Usage: bxhub [options]\n\n"
"Supported options:\n"
" -ports=... number of virtual ethernet ports (2 - 4)\n"
" -base=... base UDP port (bxhub uses 2 ports per Bochs session)\n"
" -mac=... host MAC address (default is b0:c4:20:00:00:03)\n"
" -tftp=... enable TFTP support using specified directory\n"
" --help display this help and exit\n\n");
}
int parse_cmdline(int argc, char *argv[])
{
int arg = 1;
int ret = 1;
int n;
int tmp[6];
client_max = 2;
port_base = 40000;
tftp_root[0] = 0;
memcpy(host_macaddr, default_host_macaddr, ETHERNET_MAC_ADDR_LEN);
while ((arg < argc) && (ret == 1)) {
// parse next arg
if (!strcmp("--help", argv[arg]) || !strncmp("/?", argv[arg], 2)) {
print_usage();
ret = 0;
}
else if (!strncmp("-ports=", argv[arg], 7)) {
n = atoi(&argv[arg][7]);
if ((n > 1) && (n < BXHUB_MAX_CLIENTS)) {
client_max = n;
} else {
printf("Number of virtual ethernet ports out of range\n\n");
ret = 0;
}
}
else if (!strncmp("-base=", argv[arg], 6)) {
n = atoi(&argv[arg][6]);
if ((n >= 1000) && (n <= 65530)) {
port_base = n;
} else {
printf("UDP base port number out of range\n\n");
ret = 0;
}
}
else if (!strncmp("-tftp=", argv[arg], 6)) {
strcpy(tftp_root, &argv[arg][6]);
}
else if (!strncmp("-mac=", argv[arg], 5)) {
n = sscanf(&argv[arg][5], "%x:%x:%x:%x:%x:%x",
&tmp[0],&tmp[1],&tmp[2],&tmp[3],&tmp[4],&tmp[5]);
if (n != 6) {
printf("Host MAC address malformed.\n\n");
ret = 0;
} else {
for (n=0; n<6; n++) host_macaddr[n] = (Bit8u)tmp[n];
}
}
else if (argv[arg][0] == '-') {
printf("Unknown option: %s\n\n", argv[arg]);
ret = 0;
} else {
printf("Ignoring extra parameter: %s\n\n", argv[arg]);
}
arg++;
}
return ret;
}
void CDECL intHandler(int sig)
{
if (sig == SIGINT) {
for (int i = 0; i < client_max; i++) {
if (hclient[i].init)
delete [] hclient[i].reply_buffer;
closesocket(hclient[i].so);
}
#ifdef WIN32
WSACleanup();
#endif
}
exit(0);
}
int CDECL main(int argc, char **argv)
{
int c, i, n;
socklen_t slen;
fd_set rfds;
Bit8u buf[BX_PACKET_BUFSIZE];
ethernet_header_t *ethhdr = (ethernet_header_t *)buf;
if (!parse_cmdline(argc, argv))
exit(0);
#ifdef WIN32
WORD wVersionRequested;
WSADATA wsaData;
int err;
wVersionRequested = MAKEWORD(2, 0);
err = WSAStartup(wVersionRequested, &wsaData);
if (err != 0) {
fprintf(stderr, "WSAStartup failed\n");
return 0;
}
#endif
signal(SIGINT, intHandler);
for (i = 0; i < client_max; i++) {
memset(&hclient[i], 0, sizeof(hub_client_t));
/* create sockets */
if ((hclient[i].so = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) < 0) {
perror("bxhub - cannot create socket");
exit(1);
}
/* fill addres structures */
hclient[i].sin.sin_family = AF_INET;
hclient[i].sin.sin_port = htons(port_base + (i * 2) + 1);
hclient[i].sin.sin_addr.s_addr = htonl(INADDR_ANY);
hclient[i].sout.sin_family = AF_INET;
hclient[i].sout.sin_port = htons(port_base + (i * 2));
hclient[i].sout.sin_addr.s_addr = htonl(INADDR_ANY);
/* configure (bind) sockets */
if (bind(hclient[i].so, (struct sockaddr *) &hclient[i].sin, sizeof(hclient[i].sin)) < 0) {
perror("bxhub - cannot bind socket");
exit(2);
}
printf("RX port #%d in use: %d\n", i + 1, ntohs(hclient[i].sin.sin_port));
}
printf("Host MAC address: %02x:%02x:%02x:%02x:%02x:%02x\n",
host_macaddr[0], host_macaddr[1], host_macaddr[2],
host_macaddr[3], host_macaddr[4], host_macaddr[5]
);
if (strlen(tftp_root) > 0) {
printf("TFTP using root directory '%s'\n", tftp_root);
} else {
printf("TFTP support disabled\n");
}
printf("Press CTRL+C to quit bxhub\n");
while (1) {
/* wait for input */
FD_ZERO(&rfds);
for (i = 0; i < client_max; i++) {
FD_SET(hclient[i].so, &rfds);
}
n = select(hclient[client_max-1].so+1, &rfds, NULL, NULL, NULL);
/* data is available somewhere */
for (i = 0; i < client_max; i++) {
// check input
if (FD_ISSET(hclient[i].so, &rfds)) {
slen = sizeof(hclient[i].sin);
n = recvfrom(hclient[i].so, (char*)buf, sizeof(buf), 0,
(struct sockaddr*) &hclient[i].sin, &slen);
if (n > 0) {
if (memcmp(ethhdr->dst_mac_addr, broadcast_macaddr, ETHERNET_MAC_ADDR_LEN) == 0) {
broadcast_packet(i, buf, n);
} else if (memcmp(ethhdr->dst_mac_addr, host_macaddr, ETHERNET_MAC_ADDR_LEN) == 0) {
handle_packet(&hclient[i], buf, n);
} else if (find_client(ethhdr->dst_mac_addr, &c)) {
send_packet(&hclient[c], buf, n);
}
}
}
// send reply from builtin service
if (hclient[i].pending_reply_size > 0) {
send_packet(&hclient[i], hclient[i].reply_buffer, hclient[i].pending_reply_size);
hclient[i].pending_reply_size = 0;
}
// check MAC address of new client
if (hclient[i].init != 0) {
if (hclient[i].init < 0) {
fprintf(stderr, "bxhub - wrong MAC address configuration\n");
break;
}
}
}
}
return 0;
}