mirrors/Bochs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
600228148d
Bochs/bochs/iodev/network/netutil.cc
Volker Ruppert fbc8921b0a Some changes on the "vnet" networking module. 
- Added new config file options "host", "dns" and "dhcpstart" (similar to slirp).
- modified bx_vnet_pktmover_c class to be compatible with other modules.
- On TFTP RRQ check file presence directly (fixes iPXE boot).
2020-08-29 17:45:42 +00:00

2721 lines
79 KiB
C++
Raw Blame History

/////////////////////////////////////////////////////////////////////////
// $Id$
/////////////////////////////////////////////////////////////////////////
//
// Copyright (C) 2004-2020 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
//
// netutil.cc - shared code for eth_vnet.cc and bxhub.cc
#define BX_PLUGGABLE
#ifdef BXHUB
#include "config.h"
#define WIN32_LEAN_AND_MEAN
#include "misc/bxcompat.h"
#else
#include "iodev.h"
#endif
#if BX_NETWORKING
#include "netmod.h"
#include "netutil.h"
#if !defined(WIN32) || defined(__CYGWIN__)
#include <arpa/inet.h> /* ntohs, htons */
#include <dirent.h> /* opendir, readdir */
#include <locale.h> /* setlocale */
#else
#include <winsock2.h>
#endif
typedef struct ftp_session {
Bit8u state;
bx_bool anonymous;
Bit16u pasv_port;
Bit16u client_cmd_port;
Bit16u client_data_port;
bx_bool ascii_mode;
int data_xfer_fd;
unsigned data_xfer_size;
unsigned data_xfer_pos;
unsigned cmdcode;
char *rel_path;
char *last_fname;
char dirlist_tmp[16];
struct ftp_session *next;
} ftp_session_t;
Bit16u ip_checksum(const Bit8u *buf, unsigned buf_len)
{
Bit32u sum = 0;
unsigned n;
for (n = 0; n < buf_len; n++) {
if (n & 1) {
sum += (Bit32u)(*buf++);
} else {
sum += (Bit32u)(*buf++) << 8;
}
}
while (sum > 0xffff) {
sum = (sum >> 16) + (sum & 0xffff);
}
return (Bit16u)sum;
}
// VNET server definitions
#ifdef BXHUB
#include <stdarg.h>
#include "osdep.h"
#else
#define LOG_THIS netdev->
#endif
#define ETHERNET_TYPE_IPV4 0x0800
#define ETHERNET_TYPE_ARP 0x0806
#define ETHERNET_TYPE_IPV6 0x86dd
#define ARP_OPCODE_REQUEST 1
#define ARP_OPCODE_REPLY 2
#define ARP_OPCODE_REV_REQUEST 3
#define ARP_OPCODE_REV_REPLY 4
#define ICMP_ECHO_PACKET_MAX 128
#define INET_PORT_FTPDATA 20
#define INET_PORT_FTP 21
#define INET_PORT_TIME 37
#define INET_PORT_NAME 42
#define INET_PORT_DOMAIN 53
#define INET_PORT_BOOTP_SERVER 67
#define INET_PORT_BOOTP_CLIENT 68
#define INET_PORT_HTTP 80
#define INET_PORT_NTP 123
// DHCP server
#define BOOTREQUEST 1
#define BOOTREPLY 2
#define BOOTPOPT_PADDING 0
#define BOOTPOPT_END 255
#define BOOTPOPT_SUBNETMASK 1
#define BOOTPOPT_TIMEOFFSET 2
#define BOOTPOPT_ROUTER_OPTION 3
#define BOOTPOPT_DOMAIN_NAMESERVER 6
#define BOOTPOPT_HOST_NAME 12
#define BOOTPOPT_DOMAIN_NAME 15
#define BOOTPOPT_MAX_DATAGRAM_SIZE 22
#define BOOTPOPT_DEFAULT_IP_TTL 23
#define BOOTPOPT_BROADCAST_ADDRESS 28
#define BOOTPOPT_ARPCACHE_TIMEOUT 35
#define BOOTPOPT_DEFAULT_TCP_TTL 37
#define BOOTPOPT_NTP_SERVER 42
#define BOOTPOPT_NETBIOS_NAMESERVER 44
#define BOOTPOPT_X_FONTSERVER 48
#define BOOTPOPT_REQUESTED_IP_ADDRESS 50
#define BOOTPOPT_IP_ADDRESS_LEASE_TIME 51
#define BOOTPOPT_OPTION_OVRLOAD 52
#define BOOTPOPT_DHCP_MESSAGETYPE 53
#define BOOTPOPT_SERVER_IDENTIFIER 54
#define BOOTPOPT_PARAMETER_REQUEST_LIST 55
#define BOOTPOPT_MAX_DHCP_MESSAGE_SIZE 57
#define BOOTPOPT_RENEWAL_TIME 58
#define BOOTPOPT_REBINDING_TIME 59
#define BOOTPOPT_CLASS_IDENTIFIER 60
#define BOOTPOPT_CLIENT_IDENTIFIER 61
#define DHCPDISCOVER 1
#define DHCPOFFER 2
#define DHCPREQUEST 3
#define DHCPDECLINE 4
#define DHCPACK 5
#define DHCPNAK 6
#define DHCPRELEASE 7
#define DHCPINFORM 8
#define DEFAULT_LEASE_TIME 28800
// TFTP server support by EaseWay <easeway@123.com>
#define INET_PORT_TFTP_SERVER 69
#define TFTP_RRQ 1
#define TFTP_WRQ 2
#define TFTP_DATA 3
#define TFTP_ACK 4
#define TFTP_ERROR 5
#define TFTP_OPTACK 6
#define TFTP_OPTION_OCTET 0x1
#define TFTP_OPTION_BLKSIZE 0x2
#define TFTP_OPTION_TSIZE 0x4
#define TFTP_OPTION_TIMEOUT 0x8
#define TFTP_DEFAULT_BLKSIZE 512
#define TFTP_DEFAULT_TIMEOUT 5
#define TFTP_BUFFER_SIZE 1432
static const Bit8u mcast_ipv6_mac_prefix[2] = {0x33,0x33};
static Bit8u broadcast_ipv4addr[3][4] =
{
{ 0, 0, 0, 0},
{255,255,255,255},
{192,168, 10,255}
};
static const Bit8u subnetmask_ipv4addr[4] = {0xff,0xff,0xff,0x00};
// VNET server code
vnet_server_c::vnet_server_c()
{
#ifdef BXHUB
logfd = stderr;
#endif
l4data_used = 0;
tcpfn_used = 0;
for (Bit8u c = 0; c < VNET_MAX_CLIENTS; c++) {
client[c].init = 0;
}
packet_counter = 0;
packets = NULL;
}
vnet_server_c::~vnet_server_c()
{
for (Bit8u c = 0; c < VNET_MAX_CLIENTS; c++) {
if (client[c].init) {
delete [] client[c].hostname;
}
}
#ifdef BXHUB
if (logfd != stderr) {
fclose(logfd);
}
logfd = stderr;
#endif
}
void vnet_server_c::init(bx_devmodel_c *_netdev, dhcp_cfg_t *dhcpc, const char *tftp_rootdir)
{
netdev = _netdev;
dhcp = dhcpc;
memcpy(broadcast_ipv4addr[2], dhcp->srv_ipv4addr[VNET_SRV], 3);
tftp_root = tftp_rootdir;
register_layer4_handler(0x11, INET_PORT_BOOTP_SERVER, udpipv4_dhcp_handler);
register_layer4_handler(0x11, INET_PORT_DOMAIN, udpipv4_dns_handler);
if (strlen(tftp_root) > 0) {
register_layer4_handler(0x11, INET_PORT_TFTP_SERVER, udpipv4_tftp_handler);
register_tcp_handler(INET_PORT_FTP, tcpipv4_ftp_handler);
srand((unsigned)time(NULL)); // for random FTP data port
}
}
void vnet_server_c::init_client(Bit8u clientid, const Bit8u *macaddr, char *hostname)
{
if (clientid < VNET_MAX_CLIENTS) {
client[clientid].macaddr = macaddr;
memcpy(client[clientid].default_ipv4addr, dhcp->client_base_ipv4addr, 4);
client[clientid].default_ipv4addr[3] += clientid;
memset(client[clientid].ipv4addr, 0, 4);
client[clientid].hostname = new char[256];
if (hostname != NULL) {
strcpy(client[clientid].hostname, hostname);
} else {
client[clientid].hostname[0] = 0;
}
client[clientid].init = 1;
}
}
#ifdef BXHUB
void vnet_server_c::init_log(const char *logfn)
{
logfd = fopen(logfn, "w");
}
void vnet_server_c::bx_printf(const char *fmt, ...)
{
va_list ap;
char msg[128];
va_start(ap, fmt);
vsnprintf(msg, sizeof(msg), fmt, ap);
fprintf(logfd, "%s\n", msg);
va_end(ap);
}
#endif
bx_bool vnet_server_c::find_client(const Bit8u *mac_addr, Bit8u *clientid)
{
for (Bit8u c = 0; c < VNET_MAX_CLIENTS; c++) {
if (client[c].init) {
if (memcmp(mac_addr, client[c].macaddr, ETHERNET_MAC_ADDR_LEN) == 0) {
*clientid = c;
break;
}
}
}
return (*clientid < VNET_MAX_CLIENTS);
}
void vnet_server_c::handle_packet(const Bit8u *buf, unsigned len)
{
ethernet_header_t *ethhdr = (ethernet_header_t *)buf;
Bit8u clientid = 0xff;
if (len >= 14) {
if (!find_client(ethhdr->src_mac_addr, &clientid))
return;
if (!memcmp(&ethhdr->dst_mac_addr, dhcp->host_macaddr, 6) ||
!memcmp(&ethhdr->dst_mac_addr, broadcast_macaddr, 6) ||
!memcmp(&ethhdr->dst_mac_addr, mcast_ipv6_mac_prefix, 2)) {
switch (ntohs(ethhdr->type)) {
case ETHERNET_TYPE_IPV4:
process_ipv4(clientid, buf, len);
break;
case ETHERNET_TYPE_ARP:
process_arp(clientid, buf, len);
break;
case ETHERNET_TYPE_IPV6:
BX_ERROR(("IPv6 packet not supported yet"));
break;
default: // unknown packet type.
break;
}
}
}
}
unsigned vnet_server_c::get_packet(Bit8u *buf)
{
unsigned len = 0;
packet_item_t *tmp;
if (packets != NULL) {
len = packets->len;
memcpy(buf, packets->buffer, len);
tmp = packets->next;
delete [] packets->buffer;
delete packets;
packets = tmp;
}
return len;
}
void vnet_server_c::host_to_guest(Bit8u clientid, Bit8u *buf, unsigned len, unsigned l3type)
{
packet_item_t *pitem, *tmp;
if (len < 14) {
BX_ERROR(("host_to_guest: io_len < 14!"));
return;
}
if (len < MIN_RX_PACKET_LEN) {
len = MIN_RX_PACKET_LEN;
}
ethernet_header_t *ethrhdr = (ethernet_header_t *)buf;
if (clientid != 0xff) {
memcpy(ethrhdr->dst_mac_addr, client[clientid].macaddr, ETHERNET_MAC_ADDR_LEN);
} else {
memcpy(ethrhdr->dst_mac_addr, broadcast_macaddr, ETHERNET_MAC_ADDR_LEN);
}
memcpy(ethrhdr->src_mac_addr, dhcp->host_macaddr, ETHERNET_MAC_ADDR_LEN);
ethrhdr->type = htons(l3type);
pitem = new packet_item_t;
pitem->buffer = new Bit8u[len];
memcpy(pitem->buffer, buf, len);
pitem->len = len;
pitem->next = NULL;
if (packets == NULL) {
packets = pitem;
} else {
tmp = packets;
while (tmp->next != NULL) {
tmp = tmp->next;
}
tmp->next = pitem;
}
}
/////////////////////////////////////////////////////////////////////////
// ARP
/////////////////////////////////////////////////////////////////////////
void vnet_server_c::process_arp(Bit8u clientid, const Bit8u *buf, unsigned len)
{
Bit8u replybuf[MIN_RX_PACKET_LEN];
if (len < 22) return;
if (len < (unsigned)(22+buf[18]*2+buf[19]*2)) return;
arp_header_t *arphdr = (arp_header_t *)((Bit8u *)buf +
sizeof(ethernet_header_t));
arp_header_t *arprhdr = (arp_header_t *)(replybuf + 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)) {
BX_ERROR(("Unhandled ARP message hw: 0x%04x (%d) proto: 0x%04x (%d)",
ntohs(arphdr->hw_addr_space), arphdr->hw_addr_len,
ntohs(arphdr->proto_addr_space), arphdr->proto_addr_len));
return;
}
switch (ntohs(arphdr->opcode)) {
case ARP_OPCODE_REQUEST:
if (!memcmp(&buf[22], client[clientid].macaddr, 6)) {
memcpy(client[clientid].ipv4addr, &buf[28], 4);
if (!memcmp(&buf[38], dhcp->srv_ipv4addr[VNET_SRV], 4) ||
!memcmp(&buf[38], dhcp->srv_ipv4addr[VNET_DNS], 4) ||
!memcmp(&buf[38], dhcp->srv_ipv4addr[VNET_MISC], 4)) {
memset(replybuf, 0, MIN_RX_PACKET_LEN);
memcpy(arprhdr, &buf[14], 6);
put_net2((Bit8u*)&arprhdr->opcode, ARP_OPCODE_REPLY);
memcpy((Bit8u *)arprhdr+8, dhcp->host_macaddr, ETHERNET_MAC_ADDR_LEN);
memcpy((Bit8u *)arprhdr+14, &buf[38], 4);
memcpy((Bit8u *)arprhdr+18, client[clientid].macaddr, ETHERNET_MAC_ADDR_LEN);
memcpy((Bit8u *)arprhdr+24, client[clientid].ipv4addr, 4);
host_to_guest(clientid, replybuf, MIN_RX_PACKET_LEN, ETHERNET_TYPE_ARP);
}
}
break;
case ARP_OPCODE_REPLY:
BX_ERROR(("unexpected ARP REPLY"));
break;
case ARP_OPCODE_REV_REQUEST:
BX_ERROR(("RARP is not implemented"));
break;
case ARP_OPCODE_REV_REPLY:
BX_ERROR(("unexpected RARP REPLY"));
break;
default:
BX_ERROR(("arp: unknown ARP opcode 0x%04x", ntohs(arphdr->opcode)));
break;
}
}
/////////////////////////////////////////////////////////////////////////
// IPv4
/////////////////////////////////////////////////////////////////////////
void vnet_server_c::process_ipv4(Bit8u clientid, const Bit8u *buf, unsigned len)
{
unsigned total_len;
unsigned fragment_flags;
unsigned fragment_offset;
unsigned l3header_len;
const Bit8u *l4pkt;
unsigned l4pkt_len;
Bit8u srv_id = VNET_SRV;
if (len < (14U+20U)) {
BX_ERROR(("ip packet - too small packet"));
return;
}
ip_header_t *iphdr = (ip_header_t *)((Bit8u *)buf +
sizeof(ethernet_header_t));
if (iphdr->version != 4) {
BX_ERROR(("ipv%u packet - not implemented", iphdr->version));
return;
}
l3header_len = (iphdr->header_len << 2);
if (l3header_len != 20) {
BX_ERROR(("ip: option header is not implemented"));
return;
}
if (ip_checksum((Bit8u*)iphdr, l3header_len) != (Bit16u)0xffff) {
BX_ERROR(("ip: invalid checksum"));
return;
}
total_len = ntohs(iphdr->total_len);
if (!memcmp(&iphdr->dst_addr, dhcp->srv_ipv4addr[VNET_DNS], 4)) {
srv_id = VNET_DNS;
} else if (!memcmp(&iphdr->dst_addr, dhcp->srv_ipv4addr[VNET_MISC], 4)) {
srv_id = VNET_MISC;
}
if (memcmp(&iphdr->dst_addr, dhcp->srv_ipv4addr[VNET_SRV], 4) &&
(srv_id != VNET_DNS) && (srv_id != VNET_MISC) &&
memcmp(&iphdr->dst_addr, broadcast_ipv4addr[0], 4) &&
memcmp(&iphdr->dst_addr, broadcast_ipv4addr[1], 4) &&
memcmp(&iphdr->dst_addr, broadcast_ipv4addr[2], 4))
{
BX_ERROR(("target IP address %u.%u.%u.%u is unknown",
(unsigned)buf[14+16],(unsigned)buf[14+17],
(unsigned)buf[14+18],(unsigned)buf[14+19]));
return;
}
fragment_flags = ntohs(iphdr->frag_offs) >> 13;
fragment_offset = (ntohs(iphdr->frag_offs) & 0x1fff) << 3;
if ((fragment_flags & 0x1) || (fragment_offset != 0)) {
BX_ERROR(("ignore fragmented packet!"));
return;
}
l4pkt = &buf[14 + l3header_len];
l4pkt_len = total_len - l3header_len;
switch (iphdr->protocol) {
case 0x01: // ICMP
process_icmpipv4(clientid, srv_id, &buf[14], l3header_len, l4pkt, l4pkt_len);
break;
case 0x06: // TCP
process_tcpipv4(clientid, srv_id, &buf[14], l3header_len, l4pkt, l4pkt_len);
break;
case 0x11: // UDP
process_udpipv4(clientid, srv_id, &buf[14], l3header_len, l4pkt, l4pkt_len);
break;
default:
BX_ERROR(("unknown IP protocol %02x", iphdr->protocol));
break;
}
}
void vnet_server_c::host_to_guest_ipv4(Bit8u clientid, Bit8u srv_id, Bit8u *buf, unsigned len)
{
unsigned l3header_len;
ip_header_t *iphdr = (ip_header_t *)((Bit8u *)buf +
sizeof(ethernet_header_t));
iphdr->version = 4;
l3header_len = (iphdr->header_len << 2);
iphdr->id = htons(packet_counter);
packet_counter++;
memcpy(iphdr->src_addr, dhcp->srv_ipv4addr[srv_id], 4);
memcpy(iphdr->dst_addr, client[clientid].ipv4addr, 4);
iphdr->checksum = 0;
iphdr->checksum = htons(ip_checksum((Bit8u*)iphdr, l3header_len) ^ (Bit16u)0xffff);
host_to_guest(clientid, buf, len, ETHERNET_TYPE_IPV4);
}
/////////////////////////////////////////////////////////////////////////
// ICMP/IPv4
/////////////////////////////////////////////////////////////////////////
void vnet_server_c::process_icmpipv4(Bit8u clientid, Bit8u srv_id,
const Bit8u *ipheader, unsigned ipheader_len,
const Bit8u *l4pkt, unsigned l4pkt_len)
{
unsigned icmptype;
unsigned icmpcode;
Bit8u replybuf[ICMP_ECHO_PACKET_MAX];
if (l4pkt_len < 8) return;
icmptype = l4pkt[0];
icmpcode = l4pkt[1];
if (ip_checksum(l4pkt, l4pkt_len) != (Bit16u)0xffff) {
BX_ERROR(("icmp: invalid checksum"));
return;
}
switch (icmptype) {
case 0x08: // ECHO
if (icmpcode == 0) {
if ((14U+ipheader_len+l4pkt_len) > ICMP_ECHO_PACKET_MAX) {
return;
}
memcpy(&replybuf[14], ipheader, ipheader_len);
memcpy(&replybuf[14+ipheader_len], l4pkt, l4pkt_len);
replybuf[14+ipheader_len+0] = 0x00; // echo reply
put_net2(&replybuf[14+ipheader_len+2],0);
put_net2(&replybuf[14+ipheader_len+2],
ip_checksum(&replybuf[14+ipheader_len],l4pkt_len) ^ (Bit16u)0xffff);
host_to_guest_ipv4(clientid, srv_id, replybuf, 14U+ipheader_len+l4pkt_len);
}
break;
default:
BX_ERROR(("unhandled icmp packet: type=%u code=%u", icmptype, icmpcode));
break;
}
}
/////////////////////////////////////////////////////////////////////////
// TCP/IPv4
/////////////////////////////////////////////////////////////////////////
#define TCP_DISCONNECTED 0
#define TCP_CONNECTING 1
#define TCP_CONNECTED 2
#define TCP_DISCONNECTING 3
// TCP handler methods
tcp_handler_t vnet_server_c::get_tcp_handler(unsigned port)
{
unsigned n;
for (n = 0; n < tcpfn_used; n++) {
if (tcpfn[n].port == port)
return tcpfn[n].func;
}
return (tcp_handler_t)NULL;
}
bx_bool vnet_server_c::register_tcp_handler(unsigned port, tcp_handler_t func)
{
if (get_tcp_handler(port) != (tcp_handler_t)NULL) {
BX_ERROR(("TCP port %u is already in use", port));
return 0;
}
unsigned n;
for (n = 0; n < tcpfn_used; n++) {
if (tcpfn[n].func == (tcp_handler_t)NULL) {
break;
}
}
if (n == tcpfn_used) {
if (n >= LAYER4_LISTEN_MAX) {
BX_ERROR(("vnet: LAYER4_LISTEN_MAX is too small"));
return 0;
}
tcpfn_used++;
}
tcpfn[n].port = port;
tcpfn[n].func = func;
return 1;
}
bx_bool vnet_server_c::unregister_tcp_handler(unsigned port)
{
unsigned n;
for (n = 0; n < tcpfn_used; n++) {
if (tcpfn[n].port == port) {
tcpfn[n].func = (tcp_handler_t)NULL;
return 1;
}
}
BX_ERROR(("TCP port %u is not registered", port));
return 0;
}
// TCP connection handling functions
tcp_conn_t *tcp_connections = NULL;
tcp_conn_t *tcp_new_connection(Bit8u clientid, Bit16u src_port, Bit16u dst_port)
{
tcp_conn_t *tc = new tcp_conn_t;
memset(tc, 0, sizeof(tcp_conn_t));
tc->clientid = clientid;
tc->src_port = src_port;
tc->dst_port = dst_port;
tc->next = tcp_connections;
tcp_connections = tc;
return tc;
}
tcp_conn_t *tcp_find_connection(Bit8u clientid, Bit16u src_port, Bit16u dst_port)
{
tcp_conn_t *tc = tcp_connections;
while (tc != NULL) {
if ((tc->clientid != clientid) || (tc->src_port != src_port) || (tc->dst_port != dst_port))
tc = tc->next;
else
break;
}
return tc;
}
void tcp_remove_connection(tcp_conn_t *tc)
{
tcp_conn_t *last;
if (tcp_connections == tc) {
tcp_connections = tc->next;
} else {
last = tcp_connections;
while (last != NULL) {
if (last->next != tc)
last = last->next;
else
break;
}
if (last) {
last->next = tc->next;
}
}
delete tc;
}
void vnet_server_c::process_tcpipv4(Bit8u clientid, Bit8u srv_id,
const Bit8u *ipheader, unsigned ipheader_len,
const Bit8u *l4pkt, unsigned l4pkt_len)
{
unsigned tcphdr_len, tcpdata_len, tcprhdr_len;
Bit32u tcp_seq_num, tcp_ack_num;
Bit16u tcp_src_port, tcp_dst_port, tcp_window;
Bit8u replybuf[MIN_RX_PACKET_LEN];
const Bit8u *tcp_data;
tcp_handler_t func;
tcp_conn_t *tcp_conn;
bx_bool tcp_error = 1;
Bit8u option, optlen;
Bit16u value16;
if (l4pkt_len < 20) return;
memset(replybuf, 0, MIN_RX_PACKET_LEN);
tcp_header_t *tcphdr = (tcp_header_t *)l4pkt;
tcp_header_t *tcprhdr = (tcp_header_t *)&replybuf[34];
tcp_src_port = ntohs(tcphdr->src_port);
tcp_dst_port = ntohs(tcphdr->dst_port);
tcp_seq_num = ntohl(tcphdr->seq_num);
tcp_ack_num = ntohl(tcphdr->ack_num);
tcphdr_len = tcphdr->data_offset << 2;
tcp_window = ntohs(tcphdr->window);
tcpdata_len = l4pkt_len - tcphdr_len;
tcp_data = (Bit8u*)tcphdr + tcphdr_len;
// TODO: parse options
if (tcphdr_len > 20) {
int i = 0;
do {
option = l4pkt[20+i];
if (option < 2) {
optlen = 1;
} else {
optlen = l4pkt[20+i+1];
}
if (option == 0) { // EOL
break;
}
if (option == 2) {
value16 = get_net2(&l4pkt[20+i+2]);
if (value16 != 1460) {
BX_ERROR(("TCP: MSS = %d (unhandled)", value16));
}
} else if (option != 1) { // NOP
BX_ERROR(("TCP option %d not supported yet", option));
}
i += optlen;
} while (i < (int)(tcphdr_len - 20));
}
// check header
func = get_tcp_handler(tcp_dst_port);
tcp_conn = tcp_find_connection(clientid, tcp_src_port, tcp_dst_port);
tcprhdr_len = sizeof(tcp_header_t);
if ((srv_id == VNET_MISC) && (func != (tcp_handler_t)NULL)) {
if (tcp_conn == (tcp_conn_t*)NULL) {
if (tcphdr->flags.syn) {
tcprhdr->flags.syn = 1;
tcprhdr->flags.ack = 1;
tcprhdr->seq_num = htonl(1);
tcprhdr->ack_num = htonl(tcp_seq_num+1);
tcprhdr->window = htons(tcp_window);
// set MSS in reply to default 1460
Bit8u *opthdr = (Bit8u*)tcprhdr + tcprhdr_len;
opthdr[0] = 0x02;
opthdr[1] = 0x04;
opthdr[2] = 0x05;
opthdr[3] = 0xb4;
tcprhdr_len += 4;
tcp_conn = tcp_new_connection(clientid, tcp_src_port, tcp_dst_port);
tcp_conn->state = TCP_CONNECTING;
tcp_error = 0;
}
} else {
if (tcphdr->flags.rst) {
if (tcp_conn->data != NULL) {
tcp_conn->state = TCP_DISCONNECTING;
(*func)((void *)this, tcp_conn, NULL, 0);
}
tcp_remove_connection(tcp_conn);
return;
} else if (tcphdr->flags.fin && (tcpdata_len == 0)) {
if (tcp_conn->state == TCP_CONNECTED) {
tcprhdr->flags.fin = 1;
tcprhdr->flags.ack = 1;
tcprhdr->seq_num = htonl(tcp_conn->host_seq_num);
tcprhdr->ack_num = htonl(tcp_seq_num+1);
tcprhdr->window = htons(tcp_window);
tcp_conn->state = TCP_DISCONNECTING;
tcp_error = 0;
} else if (tcp_conn->state == TCP_DISCONNECTING) {
tcpipv4_send_ack(tcp_conn, 1);
(*func)((void *)this, tcp_conn, tcp_data, tcpdata_len);
tcp_remove_connection(tcp_conn);
return;
}
} else if (tcphdr->flags.ack) {
if (tcp_conn->state == TCP_CONNECTING) {
tcp_conn->guest_seq_num = tcp_seq_num;
tcp_conn->host_seq_num = tcp_ack_num;
tcp_conn->window = tcp_window;
(*func)((void *)this, tcp_conn, tcp_data, tcpdata_len);
tcp_conn->state = TCP_CONNECTED;
} else if (tcp_conn->state == TCP_DISCONNECTING) {
if (!tcp_conn->host_port_fin) {
(*func)((void *)this, tcp_conn, tcp_data, tcpdata_len);
tcp_remove_connection(tcp_conn);
}
} else {
tcp_conn->guest_seq_num = tcp_seq_num;
tcp_conn->window = tcp_window;
if ((tcpdata_len > 0) || (tcp_ack_num == tcp_conn->host_seq_num)) {
if (tcpdata_len > 0) {
tcpipv4_send_ack(tcp_conn, tcpdata_len);
}
(*func)((void *)this, tcp_conn, tcp_data, tcpdata_len);
}
if (tcphdr->flags.fin) {
tcp_conn->guest_seq_num++;
tcpipv4_send_fin(tcp_conn, 0);
}
}
return;
} else {
BX_ERROR(("TCP: unhandled flag"));
return;
}
}
}
if (tcp_error) {
tcprhdr->flags.syn = 0;
tcprhdr->flags.rst = 1;
tcprhdr->flags.ack = 1;
tcprhdr->seq_num = htonl(1);
tcprhdr->ack_num = htonl(tcp_seq_num+1);
tcprhdr->window = 0;
BX_ERROR(("tcp - port %u unhandled or in use", tcp_dst_port));
}
host_to_guest_tcpipv4(clientid, srv_id, tcp_dst_port, tcp_src_port, replybuf, 0,
tcprhdr_len);
}
void vnet_server_c::host_to_guest_tcpipv4(Bit8u clientid, Bit8u srv_id,
Bit16u src_port,Bit16u dst_port,
Bit8u *data, unsigned data_len, unsigned hdr_len)
{
tcp_header_t *tcphdr = (tcp_header_t *)&data[34];
// tcp pseudo-header
data[34U-12U]=0;
data[34U-11U]=0x06; // TCP
put_net2(&data[34U-10U], hdr_len+data_len);
memcpy(&data[34U-8U], dhcp->srv_ipv4addr[srv_id], 4);
memcpy(&data[34U-4U], client[clientid].ipv4addr, 4);
// tcp header
tcphdr->src_port = htons(src_port);
tcphdr->dst_port = htons(dst_port);
tcphdr->data_offset = hdr_len >> 2;
tcphdr->checksum = 0;
put_net2(&data[34U+16U], ip_checksum(&data[34U-12U],12U+hdr_len+data_len) ^ (Bit16u)0xffff);
// ip header
memset(&data[14U],0,20U);
data[14U+0] = 0x45;
data[14U+1] = 0x00;
put_net2(&data[14U+2],20U+hdr_len+data_len);
put_net2(&data[14U+4],1);
data[14U+6] = 0x00;
data[14U+7] = 0x00;
data[14U+8] = 0x07; // TTL
data[14U+9] = 0x06; // TCP
host_to_guest_ipv4(clientid, srv_id, data, data_len + hdr_len + 34U);
}
unsigned vnet_server_c::tcpipv4_send_data(tcp_conn_t *tcp_conn, const Bit8u *data,
unsigned data_len, bx_bool push)
{
Bit8u sendbuf[BX_PACKET_BUFSIZE];
tcp_header_t *tcphdr = (tcp_header_t *)&sendbuf[34];
unsigned tcphdr_len = sizeof(tcp_header_t);
Bit8u *tcp_data;
unsigned pos = 0, sendsize, total;
if (data_len > 0) {
memset(tcphdr, 0, tcphdr_len);
if (push) {
tcphdr->flags.psh = 1;
}
tcphdr->flags.ack = 1;
tcphdr->ack_num = htonl(tcp_conn->guest_seq_num);
tcphdr->window = htons(tcp_conn->window);
tcp_data = (Bit8u*)tcphdr + tcphdr_len;
do {
total = data_len - pos;
if ((total + 54U) > BX_PACKET_BUFSIZE) {
sendsize = BX_PACKET_BUFSIZE - 54U;
} else {
sendsize = total;
}
if ((pos + sendsize) > tcp_conn->window)
break;
tcphdr->seq_num = htonl(tcp_conn->host_seq_num);
if (sendsize > 0) {
memcpy(tcp_data, data + pos, sendsize);
}
host_to_guest_tcpipv4(tcp_conn->clientid, VNET_MISC, tcp_conn->dst_port,
tcp_conn->src_port, sendbuf, sendsize, tcphdr_len);
tcp_conn->host_seq_num += sendsize;
pos += sendsize;
} while (pos < data_len);
} else {
tcpipv4_send_fin(tcp_conn, 1);
}
return pos;
}
void vnet_server_c::tcpipv4_send_ack(tcp_conn_t *tcp_conn, unsigned data_len)
{
Bit8u replybuf[MIN_RX_PACKET_LEN];
tcp_header_t *tcphdr = (tcp_header_t *)&replybuf[34];
unsigned tcphdr_len = sizeof(tcp_header_t);
memset(replybuf, 0, MIN_RX_PACKET_LEN);
tcphdr->flags.ack = 1;
tcphdr->seq_num = htonl(tcp_conn->host_seq_num);
tcp_conn->guest_seq_num += data_len;
tcphdr->ack_num = htonl(tcp_conn->guest_seq_num);
tcphdr->window = htons(tcp_conn->window);
host_to_guest_tcpipv4(tcp_conn->clientid, VNET_MISC, tcp_conn->dst_port,
tcp_conn->src_port, replybuf, 0, tcphdr_len);
}
void vnet_server_c::tcpipv4_send_fin(tcp_conn_t *tcp_conn, bx_bool host_fin)
{
Bit8u replybuf[MIN_RX_PACKET_LEN];
tcp_header_t *tcphdr = (tcp_header_t *)&replybuf[34];
unsigned tcphdr_len = sizeof(tcp_header_t);
memset(replybuf, 0, MIN_RX_PACKET_LEN);
tcphdr->flags.fin = 1;
tcphdr->flags.ack = 1;
tcphdr->seq_num = htonl(tcp_conn->host_seq_num);
tcp_conn->host_seq_num++;
tcphdr->ack_num = htonl(tcp_conn->guest_seq_num);
tcphdr->window = htons(tcp_conn->window);
tcp_conn->state = TCP_DISCONNECTING;
tcp_conn->host_port_fin = host_fin;
host_to_guest_tcpipv4(tcp_conn->clientid, VNET_MISC, tcp_conn->dst_port,
tcp_conn->src_port, replybuf, 0, tcphdr_len);
}
// FTP support
enum {
FTP_STATE_LOGIN = 1,
FTP_STATE_ASKPASS,
FTP_STATE_READY,
FTP_STATE_LOGOUT
};
ftp_session_t *ftp_sessions = NULL;
ftp_session_t *ftp_new_session(tcp_conn_t *tcp_conn, Bit16u client_cmd_port)
{
ftp_session_t *fs = new ftp_session_t;
memset(fs, 0, sizeof(ftp_session_t));
fs->state = FTP_STATE_LOGIN;
fs->client_cmd_port = client_cmd_port;
fs->ascii_mode = 1;
fs->data_xfer_fd = -1;
fs->rel_path = new char[BX_PATHNAME_LEN];
strcpy(fs->rel_path, "/");
fs->next = ftp_sessions;
ftp_sessions = fs;
return fs;
}
ftp_session_t *ftp_find_cmd_session(Bit16u pasv_port)
{
ftp_session_t *fs = ftp_sessions;
while (fs != NULL) {
if (fs->pasv_port != pasv_port)
fs = fs->next;
else
break;
}
return fs;
}
void ftp_remove_session(ftp_session_t *fs)
{
ftp_session_t *last;
if (ftp_sessions == fs) {
ftp_sessions = fs->next;
} else {
last = ftp_sessions;
while (last != NULL) {
if (last->next != fs)
last = last->next;
else
break;
}
if (last) {
last->next = fs->next;
}
}
if (fs->data_xfer_fd >= 0) {
close(fs->data_xfer_fd);
}
delete [] fs->rel_path;
delete fs;
}
// FTP directory helper functions
bx_bool ftp_mkdir(const char *path)
{
#ifndef WIN32
return (mkdir(path, 0755) == 0);
#else
return (CreateDirectory(path, NULL) != 0);
#endif
}
bx_bool ftp_rmdir(const char *path)
{
#ifndef WIN32
return (rmdir(path) == 0);
#else
return (RemoveDirectory(path) != 0);
#endif
}
// FTP command decoder
enum {
FTPCMD_UNKNOWN,
FTPCMD_NOPERM,
FTPCMD_ABOR,
FTPCMD_CDUP,
FTPCMD_CWD,
FTPCMD_DELE,
FTPCMD_EPRT,
FTPCMD_EPSV,
FTPCMD_FEAT,
FTPCMD_LIST,
FTPCMD_MKD,
FTPCMD_NLST,
FTPCMD_NOOP,
FTPCMD_OPTS,
FTPCMD_PASS,
FTPCMD_PASV,
FTPCMD_PORT,
FTPCMD_PWD,
FTPCMD_QUIT,
FTPCMD_RETR,
FTPCMD_RMD,
FTPCMD_RNFR,
FTPCMD_RNTO,
FTPCMD_SIZE,
FTPCMD_STAT,
FTPCMD_STOR,
FTPCMD_STOU,
FTPCMD_SYST,
FTPCMD_TYPE,
FTPCMD_USER
};
typedef struct {
char name[8];
unsigned code;
bx_bool rw;
} ftp_cmd_t;
#define FTP_N_CMDS 28
const ftp_cmd_t ftpCmd[FTP_N_CMDS] = {
{"ABOR", FTPCMD_ABOR, 0}, {"CDUP", FTPCMD_CDUP, 0}, {"CWD", FTPCMD_CWD, 0},
{"DELE", FTPCMD_DELE, 1}, {"EPRT", FTPCMD_EPRT, 0}, {"EPSV", FTPCMD_EPSV, 0},
{"FEAT", FTPCMD_FEAT, 0}, {"LIST", FTPCMD_LIST, 0}, {"MKD", FTPCMD_MKD, 1},
{"NLST", FTPCMD_NLST, 0}, {"NOOP", FTPCMD_NOOP, 0}, {"OPTS", FTPCMD_OPTS, 0},
{"PASS", FTPCMD_PASS, 0}, {"PASV", FTPCMD_PASV, 0}, {"PORT", FTPCMD_PORT, 0},
{"PWD", FTPCMD_PWD, 0}, {"QUIT", FTPCMD_QUIT, 0}, {"RETR", FTPCMD_RETR, 0},
{"RMD", FTPCMD_RMD, 1}, {"RNFR", FTPCMD_RNFR, 1}, {"RNTO", FTPCMD_RNTO, 1},
{"SIZE", FTPCMD_SIZE, 0}, {"STAT", FTPCMD_STAT, 0}, {"STOR", FTPCMD_STOR, 1},
{"STOU", FTPCMD_STOU, 1}, {"SYST", FTPCMD_SYST, 0}, {"TYPE", FTPCMD_TYPE, 0},
{"USER", FTPCMD_USER, 0}
};
unsigned ftp_get_command(const char *cmdstr, bx_bool anonuser)
{
unsigned n, cmdcode = FTPCMD_UNKNOWN;
for (n = 0; n < FTP_N_CMDS; n++) {
if (!stricmp(cmdstr, ftpCmd[n].name)) {
if (!ftpCmd[n].rw || !anonuser) {
cmdcode = ftpCmd[n].code;
} else {
cmdcode = FTPCMD_NOPERM;
}
break;
}
}
return cmdcode;
}
// FTP service handler
void vnet_server_c::tcpipv4_ftp_handler(void *this_ptr, tcp_conn_t *tcp_conn, const Bit8u *data, unsigned data_len)
{
((vnet_server_c *)this_ptr)->tcpipv4_ftp_handler_ns(tcp_conn, data, data_len);
}
void vnet_server_c::tcpipv4_ftp_handler_ns(tcp_conn_t *tcp_conn, const Bit8u *data, unsigned data_len)
{
char *ftpcmd, *cmdstr = NULL, *arg = NULL, *opt = NULL;
char reply[256];
ftp_session_t *fs;
const Bit8u *hostip;
bx_bool pasv_port_ok, path_ok;
tcp_conn_t *tcpc_cmd = NULL, *tcpc_data =NULL;
char tmp_path[BX_PATHNAME_LEN];
unsigned len;
if (tcp_conn->dst_port == INET_PORT_FTP) {
tcpc_cmd = tcp_conn;
if (tcpc_cmd->state == TCP_CONNECTING) {
// send welcome message
ftp_send_reply(tcpc_cmd, "220 Bochs FTP server ready.");
fs = ftp_new_session(tcpc_cmd, tcpc_cmd->src_port);
tcpc_cmd->data = fs;
} else if (tcpc_cmd->state == TCP_DISCONNECTING) {
fs = (ftp_session_t*)tcpc_cmd->data;
ftp_remove_session(fs);
tcpc_cmd->data = NULL;
} else if (data_len > 0) {
// skip special sequence for abort
while ((data[0] > 0xf0) && (data_len > 0)) {
data++;
data_len--;
}
if (data_len == 0) return;
// split command and argument
ftpcmd = new char[data_len + 1];
memcpy(ftpcmd, data, data_len);
ftpcmd[data_len] = 0;
cmdstr = strtok(ftpcmd, " \r");
arg = strtok(NULL, "\r");
if (arg[0] == '\n') arg++;
fs = (ftp_session_t*)tcpc_cmd->data;
fs->cmdcode = ftp_get_command(cmdstr, fs->anonymous);
if (fs->state == FTP_STATE_LOGIN) {
if (fs->cmdcode == FTPCMD_USER) {
fs->anonymous = !strcmp(arg, "anonymous");
if (!strcmp(arg, "bochs") || fs->anonymous) {
sprintf(reply, "331 Password required for %s.", arg);
ftp_send_reply(tcpc_cmd, reply);
fs->state = FTP_STATE_ASKPASS;
} else {
ftp_send_reply(tcpc_cmd, "430 Invalid username or password.");
}
}
} else if (fs->state == FTP_STATE_ASKPASS) {
if (fs->cmdcode == FTPCMD_PASS) {
if (!strcmp(arg, "bochs") || fs->anonymous) {
if (!fs->anonymous) {
ftp_send_reply(tcpc_cmd, "230 User bochs logged in.");
} else {
ftp_send_reply(tcpc_cmd, "230 Guest login with read-only access.");
}
fs->state = FTP_STATE_READY;
} else {
ftp_send_reply(tcpc_cmd, "530 Login incorrect.");
fs->state = FTP_STATE_LOGIN;
}
}
} else if (fs->state != FTP_STATE_LOGOUT) {
if (fs->pasv_port > 0) {
tcpc_data = tcp_find_connection(tcpc_cmd->clientid, fs->client_data_port, fs->pasv_port);
if (tcpc_data == NULL) {
BX_ERROR(("FTP data connection not found"));
}
}
switch (fs->cmdcode) {
case FTPCMD_ABOR:
if (fs->data_xfer_fd >= 0) {
close(fs->data_xfer_fd);
fs->data_xfer_fd = -1;
tcpipv4_send_fin(tcpc_data, 1);
ftp_send_reply(tcpc_cmd, "426 Transfer aborted.");
ftp_send_reply(tcpc_cmd, "226 Transfer abort complete.");
} else {
ftp_send_reply(tcpc_cmd, "226 Transfer complete.");
}
break;
case FTPCMD_CDUP:
if (!strcmp(fs->rel_path, "/")) {
ftp_send_reply(tcpc_cmd, "550 CDUP operation not permitted.");
} else {
len = strlen(fs->rel_path);
do {
if (len < 2) break;
} while (fs->rel_path[--len] != '/');
fs->rel_path[len] = 0;
ftp_send_reply(tcpc_cmd, "250 CDUP command successful.");
}
break;
case FTPCMD_CWD:
if (ftp_subdir_exists(tcpc_cmd, arg, tmp_path)) {
strcpy(fs->rel_path, tmp_path);
ftp_send_reply(tcpc_cmd, "250 CWD command successful.");
}
break;
case FTPCMD_DELE:
if (ftp_file_exists(tcpc_cmd, arg, tmp_path, NULL)) {
if (unlink(tmp_path) >= 0) {
ftp_send_reply(tcpc_cmd, "250 File deletion successful.");
} else {
ftp_send_reply(tcpc_cmd, "550 File deletion failed.");
}
}
break;
case FTPCMD_FEAT:
sprintf(reply, "211- Extension supported:%c%c SIZE%c%c211 end",
13, 10, 13, 10);
ftp_send_reply(tcpc_cmd, reply);
break;
case FTPCMD_LIST:
case FTPCMD_NLST:
if (tcpc_data != NULL) {
if (arg[0] == '-') {
opt = strtok(arg, " \n");
arg += (strlen(opt) + 1);
if (arg[0] == '\n') arg++;
} else {
opt = &arg[strlen(arg)];
}
if (strlen(arg) == 0) {
strcpy(tmp_path, fs->rel_path);
path_ok = 1;
} else {
path_ok = ftp_subdir_exists(tcpc_cmd, arg, tmp_path);
}
if (path_ok) {
ftp_list_directory(tcpc_cmd, tcpc_data, opt, tmp_path);
}
}
break;
case FTPCMD_MKD:
if (!ftp_subdir_exists(tcpc_cmd, arg, tmp_path)) {
if (ftp_mkdir(tmp_path)) {
ftp_send_reply(tcpc_cmd, "257 New directory created.");
} else {
ftp_send_reply(tcpc_cmd, "550 Directory creation failed.");
}
} else {
ftp_send_reply(tcpc_cmd, "550 Directory already exists.");
}
break;
case FTPCMD_NOOP:
ftp_send_reply(tcpc_cmd, "200 Command OK.");
break;
case FTPCMD_OPTS:
sprintf(reply, "501 Feature '%s' not supported.", arg);
ftp_send_reply(tcpc_cmd, reply);
break;
case FTPCMD_EPSV:
case FTPCMD_PASV:
do {
fs->pasv_port = (((rand() & 0x7f) | 0x80) << 8) | (rand() & 0xff);
pasv_port_ok = register_tcp_handler(fs->pasv_port, tcpipv4_ftp_handler);
} while (!pasv_port_ok);
if (fs->cmdcode == FTPCMD_EPSV) {
BX_DEBUG(("Extended passive FTP mode using port %d", fs->pasv_port));
sprintf(reply, "229 Entering extended passive mode (|||%d|).",
fs->pasv_port);
} else {
BX_DEBUG(("Passive FTP mode using port %d", fs->pasv_port));
hostip = dhcp->srv_ipv4addr[VNET_MISC];
sprintf(reply, "227 Entering passive mode (%d, %d, %d, %d, %d, %d).",
hostip[0], hostip[1], hostip[2], hostip[3], (fs->pasv_port >> 8),
(fs->pasv_port & 0xff));
}
ftp_send_reply(tcpc_cmd, reply);
break;
case FTPCMD_PWD:
sprintf(reply, "257 \"%s\" is current directory.", fs->rel_path);
ftp_send_reply(tcpc_cmd, reply);
break;
case FTPCMD_QUIT:
if (fs->pasv_port > 0) {
if (fs->data_xfer_fd >= 0) {
close(fs->data_xfer_fd);
}
unregister_tcp_handler(fs->pasv_port);
}
ftp_send_reply(tcpc_cmd, "221 Goodbye.");
fs->state = FTP_STATE_LOGOUT;
break;
case FTPCMD_RETR:
if (tcpc_data != NULL) {
ftp_send_file(tcpc_cmd, tcpc_data, arg);
}
break;
case FTPCMD_RMD:
if (ftp_subdir_exists(tcpc_cmd, arg, tmp_path)) {
if (ftp_rmdir(tmp_path)) {
ftp_send_reply(tcpc_cmd, "250 RMD operation successful.");
} else {
ftp_send_reply(tcpc_cmd, "550 RMD operation failed.");
}
}
break;
case FTPCMD_RNFR:
if (ftp_file_exists(tcpc_cmd, arg, tmp_path, NULL)) {
fs->last_fname = new char[strlen(tmp_path)+1];
strcpy(fs->last_fname, tmp_path);
ftp_send_reply(tcpc_cmd, "350 File exists. Ready for new name.");
}
break;
case FTPCMD_RNTO:
if (fs->last_fname != NULL) {
if (!ftp_file_exists(tcpc_cmd, arg, tmp_path, NULL)) {
if (rename(fs->last_fname, tmp_path) == 0) {
ftp_send_reply(tcpc_cmd, "250 File renamed successfully.");
} else {
ftp_send_reply(tcpc_cmd, "550 Rename operation failed.");
}
}
delete [] fs->last_fname;
fs->last_fname = NULL;
} else {
ftp_send_reply(tcpc_cmd, "550 Rename operation not permitted.");
}
break;
case FTPCMD_SIZE:
ftp_get_filesize(tcpc_cmd, arg);
break;
case FTPCMD_STAT:
ftp_send_status(tcpc_cmd);
break;
case FTPCMD_STOR:
case FTPCMD_STOU:
if (tcpc_data != NULL) {
ftp_recv_file(tcpc_cmd, tcpc_data, arg);
}
break;
case FTPCMD_SYST:
ftp_send_reply(tcpc_cmd, "215 UNIX Type: L8 Version: Bochs 2.6.11");
break;
case FTPCMD_TYPE:
if (!stricmp(arg, "A") || !stricmp(arg, "I")) {
sprintf(reply, "200 Type set to %s.", arg);
fs->ascii_mode = !stricmp(arg, "A");
} else {
sprintf(reply, "501 Type %s not supported.", arg);
}
ftp_send_reply(tcpc_cmd, reply);
break;
case FTPCMD_EPRT:
case FTPCMD_PORT:
sprintf(reply, "502 %s command not supported by server", cmdstr);
ftp_send_reply(tcpc_cmd, reply);
break;
case FTPCMD_NOPERM:
sprintf(reply, "550 %s operation not permitted.", cmdstr);
ftp_send_reply(tcpc_cmd, reply);
break;
default:
sprintf(reply, "502 Command '%s' not implemented.", cmdstr);
ftp_send_reply(tcpc_cmd, reply);
}
}
delete [] ftpcmd;
}
} else {
// FTP data port
tcpc_data = tcp_conn;
fs = ftp_find_cmd_session(tcpc_data->dst_port);
tcpc_cmd = tcp_find_connection(tcpc_data->clientid, fs->client_cmd_port,
INET_PORT_FTP);
if ((fs == NULL) || (tcpc_cmd == NULL)) {
BX_ERROR(("FTP command connection not found"));
return;
}
if (tcpc_data->state == TCP_CONNECTING) {
fs->client_data_port = tcpc_data->src_port;
tcpc_data->data = fs;
} else if (tcpc_data->state == TCP_DISCONNECTING) {
if (fs->data_xfer_fd >= 0) {
close(fs->data_xfer_fd);
fs->data_xfer_fd = -1;
if (fs->last_fname != NULL) {
sprintf(reply, "226 Transfer complete (unique file name %s).",
fs->last_fname);
ftp_send_reply(tcpc_cmd, reply);
delete [] fs->last_fname;
fs->last_fname = NULL;
} else {
ftp_send_reply(tcpc_cmd, "226 Transfer complete.");
}
}
fs->pasv_port = 0;
unregister_tcp_handler(tcpc_data->dst_port);
} else {
if (data_len > 0) {
if (fs->data_xfer_fd >= 0) {
write(fs->data_xfer_fd, data, data_len);
} else {
BX_ERROR(("FTP data port %d: unexpected data", fs->pasv_port));
}
} else {
if (fs->data_xfer_fd >= 0) {
ftp_send_data(tcpc_cmd, tcpc_data);
} else {
tcpipv4_send_fin(tcpc_data, 1);
}
}
}
}
}
bx_bool vnet_server_c::ftp_file_exists(tcp_conn_t *tcpc_cmd, const char *arg,
char *path, unsigned *size)
{
ftp_session_t *fs = (ftp_session_t*)tcpc_cmd->data;
bx_bool exists = 0, notfile = 1;
#ifndef WIN32
struct stat stat_buf;
#endif
if (size != NULL) {
*size = 0;
}
if (arg != NULL) {
if (arg[0] != '/') {
sprintf(path, "%s%s/%s", tftp_root, fs->rel_path, arg);
} else {
sprintf(path, "%s%s", tftp_root, arg);
}
}
#ifdef WIN32
HANDLE hFile = CreateFile(path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_RANDOM_ACCESS, NULL);
if (hFile != INVALID_HANDLE_VALUE) {
ULARGE_INTEGER FileSize;
FileSize.LowPart = GetFileSize(hFile, &FileSize.HighPart);
CloseHandle(hFile);
exists = 1;
if (((FileSize.LowPart != INVALID_FILE_SIZE) || (GetLastError() == NO_ERROR)) &&
(FileSize.HighPart == 0)) {
if (size != NULL) {
*size = FileSize.LowPart;
}
notfile = 0;
}
}
#else
int fd = open(path, O_RDONLY
#ifdef O_BINARY
| O_BINARY
#endif
);
if (fd >= 0) {
if (fstat(fd, &stat_buf) == 0) {
if (S_ISREG(stat_buf.st_mode)) {
notfile = 0;
}
if (size != NULL) {
*size = stat_buf.st_size;
}
exists = 1;
}
close(fd);
}
#endif
if (!exists) {
if ((fs->cmdcode != FTPCMD_RNTO) && (fs->cmdcode != FTPCMD_STOU)) {
ftp_send_reply(tcpc_cmd, "550 File not found.");
}
} else {
if (fs->cmdcode == FTPCMD_RNTO) {
ftp_send_reply(tcpc_cmd, "550 File exists.");
} else if (notfile) {
ftp_send_reply(tcpc_cmd, "550 Not a regular file.");
}
}
return (exists && !notfile);
}
bx_bool vnet_server_c::ftp_subdir_exists(tcp_conn_t *tcpc_cmd, const char *arg,
char *path)
{
ftp_session_t *fs = (ftp_session_t*)tcpc_cmd->data;
char abspath[BX_PATHNAME_LEN];
char relpath[BX_PATHNAME_LEN];
bx_bool exists = 0, notdir = 1;
#ifndef WIN32
DIR *dir;
#endif
if (arg[0] != '/') {
if (!strcmp(fs->rel_path, "/")) {
sprintf(relpath, "/%s", arg);
} else {
sprintf(relpath, "%s/%s", fs->rel_path, arg);
}
} else {
strcpy(relpath, arg);
}
if (!strcmp(relpath, "/")) {
strcpy(abspath, tftp_root);
} else {
sprintf(abspath, "%s%s", tftp_root, relpath);
}
#ifndef WIN32
dir = opendir(abspath);
if (dir != NULL) {
closedir(dir);
exists = 1;
} else if (errno != ENOTDIR) {
notdir = 0;
}
#else
DWORD dwAttrib = GetFileAttributes(abspath);
exists = ((dwAttrib != INVALID_FILE_ATTRIBUTES) &&
((dwAttrib & FILE_ATTRIBUTE_DIRECTORY) != 0));
if (!exists) {
notdir = ((dwAttrib != INVALID_FILE_ATTRIBUTES) &&
((dwAttrib & FILE_ATTRIBUTE_DIRECTORY) == 0));
}
#endif
if (exists) {
if (fs->cmdcode == FTPCMD_RMD) {
strcpy(path, abspath);
} else {
strcpy(path, relpath);
}
} else {
if (fs->cmdcode == FTPCMD_MKD) {
strcpy(path, abspath);
} else if (notdir) {
ftp_send_reply(tcpc_cmd, "550 Not a directory.");
} else {
ftp_send_reply(tcpc_cmd, "550 Directory not found.");
}
}
return exists;
}
void vnet_server_c::ftp_send_reply(tcp_conn_t *tcp_conn, const char *msg)
{
if (strlen(msg) > 0) {
char *reply = new char[strlen(msg) + 3];
sprintf(reply, "%s%c%c", msg, 13, 10);
tcpipv4_send_data(tcp_conn, (Bit8u*)reply, strlen(reply), 1);
delete [] reply;
}
}
void vnet_server_c::ftp_send_status(tcp_conn_t *tcp_conn)
{
ftp_session_t *fs = (ftp_session_t*)tcp_conn->data;
char reply[256], linebuf[80];
Bit8u *ipv4addr = client[tcp_conn->clientid].ipv4addr;
sprintf(reply, "211- Bochs FTP server status:%c%c", 13, 10);
sprintf(linebuf, " Connected to %u.%u.%u.%u%c%c", ipv4addr[0], ipv4addr[1],
ipv4addr[2], ipv4addr[3], 13, 10);
strcat(reply, linebuf);
if (fs->anonymous) {
sprintf(linebuf, " Logged in anonymously%c%c", 13, 10);
} else {
sprintf(linebuf, " Logged in as ftpuser%c%c", 13, 10);
}
strcat(reply, linebuf);
sprintf(linebuf, " Type: ASCII, Form: Nonprint; STRUcture: File; Transfer MODE: Stream%c%c", 13, 10);
strcat(reply, linebuf);
sprintf(linebuf, " No data connection%c%c", 13, 10);
strcat(reply, linebuf);
sprintf(linebuf, "211 End of status%c%c", 13, 10);
strcat(reply, linebuf);
tcpipv4_send_data(tcp_conn, (Bit8u*)reply, strlen(reply), 1);
}
void vnet_server_c::ftp_send_data_prep(tcp_conn_t *tcpc_cmd, tcp_conn_t *tcpc_data,
const char *path, unsigned data_len)
{
ftp_session_t *fs = (ftp_session_t*)tcpc_cmd->data;
fs->data_xfer_fd = open(path, O_RDONLY
#ifdef O_BINARY
| O_BINARY
#endif
);
fs->data_xfer_size = data_len;
fs->data_xfer_pos = 0;
ftp_send_data(tcpc_cmd, tcpc_data);
}
void vnet_server_c::ftp_send_data(tcp_conn_t *tcpc_cmd, tcp_conn_t *tcpc_data)
{
ftp_session_t *fs = (ftp_session_t*)tcpc_cmd->data;
Bit8u *buffer = NULL;
unsigned data_len = fs->data_xfer_size - fs->data_xfer_pos;
if (tcpc_data->window == 0)
return;
if (data_len > tcpc_data->window) {
data_len = tcpc_data->window;
}
if (data_len > 0) {
buffer = new Bit8u[data_len];
lseek(fs->data_xfer_fd, fs->data_xfer_pos, SEEK_SET);
read(fs->data_xfer_fd, buffer, data_len);
}
fs->data_xfer_pos += tcpipv4_send_data(tcpc_data, buffer, data_len, 0);
if (fs->data_xfer_pos == fs->data_xfer_size) {
ftp_send_reply(tcpc_cmd, "226 Transfer complete.");
close(fs->data_xfer_fd);
fs->data_xfer_fd = -1;
if (strlen(fs->dirlist_tmp) > 0) {
unlink(fs->dirlist_tmp);
fs->dirlist_tmp[0] = 0;
}
}
if (data_len > 0) {
delete [] buffer;
}
}
void vnet_server_c::ftp_list_directory(tcp_conn_t *tcpc_cmd, tcp_conn_t *tcpc_data,
const char *options, const char *subdir)
{
ftp_session_t *fs;
char abspath[BX_PATHNAME_LEN], reply[80];
char linebuf[BX_PATHNAME_LEN], tmptime[20];
unsigned size = 0;
int fd = -1;
bx_bool nlst, hidden = 0;
#ifndef WIN32
DIR *dir;
struct dirent *dent;
struct stat st;
char path[BX_PATHNAME_LEN];
time_t now = time(NULL);
#endif
fs = (ftp_session_t*)tcpc_cmd->data;
nlst = (fs->cmdcode == FTPCMD_NLST);
if (nlst) {
hidden = 1;
} else {
hidden = (strchr(options, 'a') != NULL);
}
sprintf(reply, "150 Opening %s mode connection for file list.",
fs->ascii_mode ? "ASCII":"BINARY");
ftp_send_reply(tcpc_cmd, reply);
if (!strcmp(subdir, "/")) {
strcpy(abspath, tftp_root);
} else {
sprintf(abspath, "%s%s", tftp_root, subdir);
}
strcpy(fs->dirlist_tmp, "dirlist.XXXXXX");
#if BX_HAVE_MKSTEMP
fd = mkstemp(fs->dirlist_tmp);
#else
mktemp(fs->dirlist_tmp);
fd = open(fs->dirlist_tmp, O_CREAT | O_WRONLY | O_TRUNC
#ifdef O_BINARY
| O_BINARY
#endif
, 0644);
#endif
if (fd >= 0) {
#ifndef WIN32
setlocale(LC_ALL, "en_US");
dir = opendir(abspath);
if (dir != NULL) {
while ((dent=readdir(dir)) != NULL) {
linebuf[0] = 0;
if (strcmp(dent->d_name, ".") && strcmp(dent->d_name, "..") &&
(hidden || (dent->d_name[0] != '.'))) {
sprintf(path, "%s/%s", abspath, dent->d_name);
if (!nlst) {
if (stat(path, &st) >= 0) {
if (st.st_mtime < (now - 31536000)) {
strftime(tmptime, 20, "%b %d %Y", localtime(&st.st_mtime));
} else {
strftime(tmptime, 20, "%b %d %H:%M", localtime(&st.st_mtime));
}
if (S_ISDIR(st.st_mode)) {
sprintf(linebuf, "drwxrwxr-x 1 ftp ftp %ld %s %s%c%c", st.st_size,
tmptime, dent->d_name, 13, 10);
} else {
sprintf(linebuf, "-rw-rw-r-- 1 ftp ftp %ld %s %s%c%c", st.st_size,
tmptime, dent->d_name, 13, 10);
}
}
} else {
sprintf(linebuf, "%s%c%c", dent->d_name, 13, 10);
}
if (strlen(linebuf) > 0) {
write(fd, linebuf, strlen(linebuf));
size += strlen(linebuf);
}
}
}
closedir(dir);
}
#else
WIN32_FIND_DATA finddata;
HANDLE hFind;
SYSTEMTIME gmtsystime, systime, now;
TIME_ZONE_INFORMATION tzi;
const char month[][12] = {"Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul",
"Aug", "Sep", "Oct", "Nov", "Dec"};
char filter[MAX_PATH];
sprintf(filter, "%s\\*.*", abspath);
hFind = FindFirstFile(filter, &finddata);
if (hFind != INVALID_HANDLE_VALUE) {
do {
linebuf[0] = 0;
if (strcmp(finddata.cFileName, ".") && strcmp(finddata.cFileName, "..") &&
(hidden || (finddata.cFileName[0] != '.'))) {
if (!nlst) {
FileTimeToSystemTime(&finddata.ftLastWriteTime, &gmtsystime);
GetTimeZoneInformation(&tzi);
SystemTimeToTzSpecificLocalTime(&tzi, &gmtsystime, &systime);
GetLocalTime(&now);
if ((systime.wYear == now.wYear) ||
((systime.wYear == (now.wYear - 1)) && (systime.wMonth > now.wMonth))) {
sprintf(tmptime, "%s %d %02d:%02d", month[systime.wMonth-1], systime.wDay,
systime.wHour, systime.wMinute);
} else {
sprintf(tmptime, "%s %d %d", month[systime.wMonth-1], systime.wDay,
systime.wYear);
}
if (finddata.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
sprintf(linebuf, "drwxrwxr-x 1 ftp ftp %ld %s %s%c%c", 0, tmptime,
finddata.cFileName, 13, 10);
} else {
sprintf(linebuf, "-rw-rw-r-- 1 ftp ftp %ld %s %s%c%c", finddata.nFileSizeLow,
tmptime, finddata.cFileName, 13, 10);
}
} else {
sprintf(linebuf, "%s%c%c", finddata.cFileName, 13, 10);
}
if (strlen(linebuf) > 0) {
write(fd, linebuf, strlen(linebuf));
size += strlen(linebuf);
}
}
} while (FindNextFile(hFind, &finddata));
FindClose(hFind);
}
#endif
close(fd);
}
ftp_send_data_prep(tcpc_cmd, tcpc_data, fs->dirlist_tmp, size);
}
void vnet_server_c::ftp_recv_file(tcp_conn_t *tcpc_cmd, tcp_conn_t *tcpc_data,
const char *fname)
{
char path[BX_PATHNAME_LEN], tmp_path[BX_PATHNAME_LEN+4], *cptr, reply[80];
ftp_session_t *fs = (ftp_session_t*)tcpc_cmd->data;
int fd = -1;
bx_bool exists;
Bit8u n = 1;
exists = ftp_file_exists(tcpc_cmd, fname, path, NULL);
if (exists && (fs->cmdcode == FTPCMD_STOU)) {
do {
sprintf(tmp_path, "%s.%d", path, n++);
} while (ftp_file_exists(tcpc_cmd, NULL, tmp_path, NULL));
strcpy(path, tmp_path);
cptr = strrchr(path, '/');
cptr++;
fs->last_fname = new char[strlen(cptr)+1];
strcpy(fs->last_fname, cptr);
}
fd = open(path, O_CREAT | O_WRONLY | O_TRUNC
#ifdef O_BINARY
| O_BINARY
#endif
, 0644);
if (fd >= 0) {
sprintf(reply, "150 Opening %s mode connection to receive file.",
fs->ascii_mode ? "ASCII":"BINARY");
ftp_send_reply(tcpc_cmd, reply);
fs->data_xfer_fd = fd;
} else {
ftp_send_reply(tcpc_cmd, "550 File creation failed.");
}
}
void vnet_server_c::ftp_send_file(tcp_conn_t *tcpc_cmd, tcp_conn_t *tcpc_data,
const char *arg)
{
ftp_session_t *fs = (ftp_session_t*)tcpc_cmd->data;
char path[BX_PATHNAME_LEN], reply[80];
unsigned size = 0;
if (ftp_file_exists(tcpc_cmd, arg, path, &size)) {
sprintf(reply, "150 Opening %s mode connection to send file.",
fs->ascii_mode ? "ASCII":"BINARY");
ftp_send_reply(tcpc_cmd, reply);
ftp_send_data_prep(tcpc_cmd, tcpc_data, path, size);
}
}
void vnet_server_c::ftp_get_filesize(tcp_conn_t *tcp_conn, const char *arg)
{
char path[BX_PATHNAME_LEN];
char reply[20];
unsigned size = 0;
if (ftp_file_exists(tcp_conn, arg, path, &size)) {
sprintf(reply, "213 %d", size);
ftp_send_reply(tcp_conn, reply);
} else {
ftp_send_reply(tcp_conn, "550 File not found.");
}
}
// Layer 4 handler methods
layer4_handler_t vnet_server_c::get_layer4_handler(
unsigned ipprotocol, unsigned port)
{
unsigned n;
for (n = 0; n < l4data_used; n++) {
if (l4data[n].ipprotocol == ipprotocol && l4data[n].port == port)
return l4data[n].func;
}
return (layer4_handler_t)NULL;
}
bx_bool vnet_server_c::register_layer4_handler(
unsigned ipprotocol, unsigned port,layer4_handler_t func)
{
if (get_layer4_handler(ipprotocol,port) != (layer4_handler_t)NULL) {
BX_ERROR(("IP protocol 0x%02x port %u is already in use",
ipprotocol, port));
return false;
}
unsigned n;
for (n = 0; n < l4data_used; n++) {
if (l4data[n].func == (layer4_handler_t)NULL) {
break;
}
}
if (n == l4data_used) {
if (n >= LAYER4_LISTEN_MAX) {
BX_ERROR(("vnet: LAYER4_LISTEN_MAX is too small"));
return false;
}
l4data_used++;
}
l4data[n].ipprotocol = ipprotocol;
l4data[n].port = port;
l4data[n].func = func;
return true;
}
bx_bool vnet_server_c::unregister_layer4_handler(
unsigned ipprotocol, unsigned port)
{
unsigned n;
for (n = 0; n < l4data_used; n++) {
if (l4data[n].ipprotocol == ipprotocol && l4data[n].port == port) {
l4data[n].func = (layer4_handler_t)NULL;
return true;
}
}
BX_ERROR(("IP protocol 0x%02x port %u is not registered",
ipprotocol,port));
return false;
}
/////////////////////////////////////////////////////////////////////////
// UDP/IPv4
/////////////////////////////////////////////////////////////////////////
void vnet_server_c::process_udpipv4(Bit8u clientid, Bit8u srv_id,
const Bit8u *ipheader, unsigned ipheader_len,
const Bit8u *l4pkt, unsigned l4pkt_len)
{
unsigned udp_dst_port;
unsigned udp_src_port;
int udp_len = 0;
Bit8u replybuf[BX_PACKET_BUFSIZE];
Bit8u *udpreply = &replybuf[42];
layer4_handler_t func;
if (l4pkt_len < 8) return;
udp_header_t *udphdr = (udp_header_t *)l4pkt;
udp_src_port = ntohs(udphdr->src_port);
udp_dst_port = ntohs(udphdr->dst_port);
// udp_len = ntohs(udphdr->length);
if ((srv_id == VNET_DNS) != (udp_dst_port ==INET_PORT_DOMAIN))
return;
func = get_layer4_handler(0x11, udp_dst_port);
if (func != (layer4_handler_t)NULL) {
udp_len = (*func)((void *)this,ipheader, ipheader_len,
udp_src_port, udp_dst_port, &l4pkt[8], l4pkt_len-8, udpreply);
} else {
BX_ERROR(("udp - unhandled port %u", udp_dst_port));
}
if (udp_len > 0) {
if ((udp_len + 42U) > BX_PACKET_BUFSIZE) {
BX_ERROR(("generated udp data is too long"));
return;
}
// udp pseudo-header
replybuf[34U-12U] = 0;
replybuf[34U-11U] = 0x11; // UDP
put_net2(&replybuf[34U-10U], 8U+udp_len);
memcpy(&replybuf[34U-8U], dhcp->srv_ipv4addr[srv_id], 4);
memcpy(&replybuf[34U-4U], client[clientid].ipv4addr, 4);
// udp header
put_net2(&replybuf[34U+0], udp_dst_port);
put_net2(&replybuf[34U+2], udp_src_port);
put_net2(&replybuf[34U+4],8U+udp_len);
put_net2(&replybuf[34U+6],0);
put_net2(&replybuf[34U+6], ip_checksum(&replybuf[34U-12U],12U+8U+udp_len) ^ (Bit16u)0xffff);
// ip header
memset(&replybuf[14U], 0, 20U);
replybuf[14U+0] = 0x45;
replybuf[14U+1] = 0x00;
put_net2(&replybuf[14U+2], 20U+8U+udp_len);
put_net2(&replybuf[14U+4], 1);
replybuf[14U+6] = 0x00;
replybuf[14U+7] = 0x00;
replybuf[14U+8] = 0x07; // TTL
replybuf[14U+9] = 0x11; // UDP
host_to_guest_ipv4(clientid, srv_id, replybuf, udp_len + 42U);
}
}
int vnet_server_c::udpipv4_dhcp_handler(void *this_ptr, const Bit8u *ipheader,
unsigned ipheader_len, unsigned sourceport, unsigned targetport,
const Bit8u *l4pkt, unsigned l4pkt_len, Bit8u *reply)
{
return ((vnet_server_c *)this_ptr)->udpipv4_dhcp_handler_ns(ipheader,
ipheader_len, sourceport, targetport, l4pkt, l4pkt_len, reply);
}
int vnet_server_c::udpipv4_dhcp_handler_ns(const Bit8u *ipheader,
unsigned ipheader_len, unsigned sourceport, unsigned targetport,
const Bit8u *data, unsigned data_len, Bit8u *reply)
{
Bit8u clientid = 0;
const Bit8u *opts;
unsigned opts_len;
unsigned extcode;
unsigned extlen;
const Bit8u *extdata;
unsigned dhcpmsgtype = 0;
bx_bool found_serverid = 0;
bx_bool found_guest_ipaddr = 0;
Bit32u leasetime = BX_MAX_BIT32U;
const Bit8u *dhcpreqparams = NULL;
unsigned dhcpreqparams_len = 0;
Bit8u *replyopts;
Bit8u replybuf[576];
unsigned hostname_len;
Bit16u maxsize = 0;
if (data_len < (236U+4U)) return 0;
if (data[0] != BOOTREQUEST) return 0;
if (data[1] != 1 || data[2] != 6) return 0;
if (!find_client(&data[28U], &clientid)) return 0;
hostname_len = strlen(client[clientid].hostname);
if (data[236] != 0x63 || data[237] != 0x82 ||
data[238] != 0x53 || data[239] != 0x63) return 0;
opts = &data[240];
opts_len = data_len - 240U;
while (1) {
if (opts_len < 1) {
BX_ERROR(("dhcp: invalid request"));
return 0;
}
extcode = *opts++;
opts_len--;
if (extcode == BOOTPOPT_PADDING) continue;
if (extcode == BOOTPOPT_END) break;
if (opts_len < 1) {
BX_ERROR(("dhcp: invalid request"));
return 0;
}
extlen = *opts++;
opts_len--;
if (opts_len < extlen) {
BX_ERROR(("dhcp: invalid request"));
return 0;
}
extdata = opts;
opts += extlen;
opts_len -= extlen;
switch (extcode) {
case BOOTPOPT_DHCP_MESSAGETYPE:
if (extlen != 1)
break;
dhcpmsgtype = *extdata;
break;
case BOOTPOPT_PARAMETER_REQUEST_LIST:
if (extlen < 1)
break;
dhcpreqparams = extdata;
dhcpreqparams_len = extlen;
break;
case BOOTPOPT_SERVER_IDENTIFIER:
if (extlen != 4)
break;
if (memcmp(extdata, dhcp->srv_ipv4addr[VNET_SRV], 4)) {
BX_INFO(("dhcp: request to another server"));
return 0;
}
found_serverid = 1;
break;
case BOOTPOPT_IP_ADDRESS_LEASE_TIME:
if (extlen != 4)
break;
leasetime = get_net4(&extdata[0]);
break;
case BOOTPOPT_REQUESTED_IP_ADDRESS:
if (extlen != 4)
break;
if (!memcmp(extdata, client[clientid].default_ipv4addr, 4)) {
found_guest_ipaddr = 1;
memcpy(client[clientid].ipv4addr, client[clientid].default_ipv4addr, 4);
}
break;
case BOOTPOPT_HOST_NAME:
if (extlen < 1)
break;
if ((hostname_len == 0) && (extlen < 256)) {
memcpy(client[clientid].hostname, extdata, extlen);
client[clientid].hostname[extlen] = 0;
hostname_len = extlen;
}
break;
case BOOTPOPT_MAX_DHCP_MESSAGE_SIZE:
if (extlen < 2)
break;
maxsize = get_net2(&extdata[0]);
if (maxsize < 548U) {
BX_ERROR(("invalid max. DHCP message size = %d", maxsize));
}
break;
default:
BX_ERROR(("extcode %d not supported yet (len=%d)", extcode, extlen));
break;
}
}
memset(&replybuf[0],0,sizeof(replybuf));
replybuf[0] = BOOTREPLY;
replybuf[1] = 1;
replybuf[2] = 6;
memcpy(&replybuf[4],&data[4],4);
memcpy(&replybuf[16], client[clientid].default_ipv4addr, 4);
memcpy(&replybuf[20], dhcp->srv_ipv4addr[VNET_SRV], 4);
memcpy(&replybuf[28],&data[28],6);
memcpy(&replybuf[44],"vnet",4);
if (strlen(dhcp->bootfile) > 0) {
memcpy(&replybuf[108], dhcp->bootfile, strlen(dhcp->bootfile));
}
replybuf[236] = 0x63;
replybuf[237] = 0x82;
replybuf[238] = 0x53;
replybuf[239] = 0x63;
replyopts = &replybuf[240];
opts_len = sizeof(replybuf)/sizeof(replybuf[0])-240;
switch (dhcpmsgtype) {
case DHCPDISCOVER:
BX_DEBUG(("dhcp server: DHCPDISCOVER"));
// reset guest address; answer must be broadcasted to unconfigured IP
memcpy(client[clientid].ipv4addr, broadcast_ipv4addr[1], 4);
*replyopts ++ = BOOTPOPT_DHCP_MESSAGETYPE;
*replyopts ++ = 1;
*replyopts ++ = DHCPOFFER;
opts_len -= 3;
break;
case DHCPREQUEST:
BX_DEBUG(("dhcp server: DHCPREQUEST"));
// check ciaddr.
if (found_serverid || found_guest_ipaddr || (!memcmp(&data[12], client[clientid].default_ipv4addr, 4))) {
*replyopts ++ = BOOTPOPT_DHCP_MESSAGETYPE;
*replyopts ++ = 1;
*replyopts ++ = DHCPACK;
opts_len -= 3;
} else {
*replyopts ++ = BOOTPOPT_DHCP_MESSAGETYPE;
*replyopts ++ = 1;
*replyopts ++ = DHCPNAK;
opts_len -= 3;
}
break;
default:
BX_ERROR(("dhcp server: unsupported message type %u",dhcpmsgtype));
return 0;
}
// default DHCP options must be on top of reply
BX_DEBUG(("provide BOOTPOPT_SERVER_IDENTIFIER"));
opts_len -= 6;
*replyopts ++ = BOOTPOPT_SERVER_IDENTIFIER;
*replyopts ++ = 4;
memcpy(replyopts, dhcp->srv_ipv4addr[VNET_SRV], 4);
replyopts += 4;
BX_DEBUG(("provide BOOTPOPT_IP_ADDRESS_LEASE_TIME"));
opts_len -= 6;
*replyopts ++ = BOOTPOPT_IP_ADDRESS_LEASE_TIME;
*replyopts ++ = 4;
if (leasetime < DEFAULT_LEASE_TIME) {
put_net4(replyopts, leasetime);
} else {
put_net4(replyopts, DEFAULT_LEASE_TIME);
}
replyopts += 4;
if (hostname_len > 0) {
BX_DEBUG(("provide BOOTPOPT_HOST_NAME"));
opts_len -= (hostname_len + 2);
*replyopts ++ = BOOTPOPT_HOST_NAME;
*replyopts ++ = hostname_len;
memcpy(replyopts, client[clientid].hostname, hostname_len);
replyopts += hostname_len;
}
while (dhcpreqparams_len-- > 0) {
switch (*dhcpreqparams++) {
case BOOTPOPT_SUBNETMASK:
BX_DEBUG(("provide BOOTPOPT_SUBNETMASK"));
if (opts_len < 6) {
BX_ERROR(("option buffer is insufficient"));
return 0;
}
opts_len -= 6;
*replyopts ++ = BOOTPOPT_SUBNETMASK;
*replyopts ++ = 4;
memcpy(replyopts,subnetmask_ipv4addr,4);
replyopts += 4;
break;
case BOOTPOPT_ROUTER_OPTION:
BX_DEBUG(("provide BOOTPOPT_ROUTER_OPTION"));
if (opts_len < 6) {
BX_ERROR(("option buffer is insufficient"));
return 0;
}
opts_len -= 6;
*replyopts ++ = BOOTPOPT_ROUTER_OPTION;
*replyopts ++ = 4;
memcpy(replyopts, dhcp->srv_ipv4addr[VNET_SRV], 4);
replyopts += 4;
break;
case BOOTPOPT_DOMAIN_NAMESERVER:
BX_DEBUG(("provide BOOTPOPT_DOMAIN_NAMESERVER"));
if (opts_len < 6) {
BX_ERROR(("option buffer is insufficient"));
return 0;
}
opts_len -= 6;
*replyopts ++ = BOOTPOPT_DOMAIN_NAMESERVER;
*replyopts ++ = 4;
memcpy(replyopts, dhcp->srv_ipv4addr[VNET_DNS], 4);
replyopts += 4;
break;
case BOOTPOPT_BROADCAST_ADDRESS:
BX_DEBUG(("provide BOOTPOPT_BROADCAST_ADDRESS"));
if (opts_len < 6) {
BX_ERROR(("option buffer is insufficient"));
return 0;
}
opts_len -= 6;
*replyopts ++ = BOOTPOPT_BROADCAST_ADDRESS;
*replyopts ++ = 4;
memcpy(replyopts, dhcp->srv_ipv4addr[VNET_SRV], 3);
replyopts += 3;
*replyopts ++ = 0xff;
break;
case BOOTPOPT_RENEWAL_TIME:
BX_DEBUG(("provide BOOTPOPT_RENEWAL_TIME"));
if (opts_len < 6) {
BX_ERROR(("option buffer is insufficient"));
return 0;
}
opts_len -= 6;
*replyopts ++ = BOOTPOPT_RENEWAL_TIME;
*replyopts ++ = 4;
put_net4(replyopts, 600);
replyopts += 4;
break;
case BOOTPOPT_REBINDING_TIME:
BX_DEBUG(("provide BOOTPOPT_REBINDING_TIME"));
if (opts_len < 6) {
BX_ERROR(("option buffer is insufficient"));
return 0;
}
opts_len -= 6;
*replyopts ++ = BOOTPOPT_REBINDING_TIME;
*replyopts ++ = 4;
put_net4(replyopts, 1800);
replyopts += 4;
break;
default:
if (*(dhcpreqparams-1) != 0) {
BX_ERROR(("dhcp server: requested parameter %u not supported yet",*(dhcpreqparams-1)));
}
break;
}
}
if (opts_len < 1) {
BX_ERROR(("option buffer is insufficient"));
return 0;
}
opts_len -= 2;
*replyopts ++ = BOOTPOPT_END;
opts_len = replyopts - &replybuf[0];
if (opts_len < (236U+64U)) {
opts_len = (236U+64U); // BOOTP
}
if (opts_len < (548U)) {
opts_len = 548U; // DHCP
}
memcpy(reply, replybuf, opts_len);
return opts_len;
}
// TFTP support
tftp_session_t *tftp_sessions = NULL;
tftp_session_t *tftp_new_session(Bit16u req_tid, bx_bool mode, const char *tpath, const char *tname)
{
tftp_session_t *s = new tftp_session_t;
s->tid = req_tid;
s->iswrite = mode;
s->options = 0;
s->blksize_val = TFTP_DEFAULT_BLKSIZE;
s->timeout_val = TFTP_DEFAULT_TIMEOUT;
s->next = tftp_sessions;
tftp_sessions = s;
if ((strlen(tname) > 0) && ((strlen(tpath) + strlen(tname)) < BX_PATHNAME_LEN)) {
sprintf(s->filename, "%s/%s", tpath, tname);
} else {
s->filename[0] = 0;
}
return s;
}
tftp_session_t *tftp_find_session(Bit16u tid)
{
tftp_session_t *s = tftp_sessions;
while (s != NULL) {
if (s->tid != tid)
s = s->next;
else
break;
}
return s;
}
void tftp_remove_session(tftp_session_t *s)
{
tftp_session_t *last;
if (tftp_sessions == s) {
tftp_sessions = s->next;
} else {
last = tftp_sessions;
while (last != NULL) {
if (last->next != s)
last = last->next;
else
break;
}
if (last) {
last->next = s->next;
}
}
delete s;
}
void tftp_update_timestamp(tftp_session_t *s)
{
#ifndef BXHUB
s->timestamp = (unsigned)(bx_pc_system.time_usec() / 1000000);
#else
s->timestamp = (unsigned)time(NULL);
#endif
}
void tftp_timeout_check()
{
#ifndef BXHUB
unsigned curtime = (unsigned)(bx_pc_system.time_usec() / 1000000);
#else
unsigned curtime = (unsigned)time(NULL);
#endif
tftp_session_t *next, *s = tftp_sessions;
while (s != NULL) {
if ((curtime - s->timestamp) > s->timeout_val) {
next = s->next;
tftp_remove_session(s);
s = next;
} else {
s = s->next;
}
}
}
int tftp_send_error(Bit8u *buffer, unsigned code, const char *msg, tftp_session_t *s)
{
put_net2(buffer, TFTP_ERROR);
put_net2(buffer + 2, code);
strcpy((char*)buffer + 4, msg);
if (s != NULL) {
tftp_remove_session(s);
}
return (strlen(msg) + 5);
}
int tftp_send_data(Bit8u *buffer, unsigned block_nr, tftp_session_t *s)
{
char msg[BX_PATHNAME_LEN + 16];
int rd;
FILE *fp = fopen(s->filename, "rb");
if (!fp) {
sprintf(msg, "File not found: %s", s->filename);
return tftp_send_error(buffer, 1, msg, s);
}
if (fseek(fp, (block_nr - 1) * s->blksize_val, SEEK_SET) < 0) {
fclose(fp);
return tftp_send_error(buffer, 3, "Block not seekable", s);
}
rd = fread(buffer + 4, 1, s->blksize_val, fp);
fclose(fp);
if (rd < 0) {
return tftp_send_error(buffer, 3, "Block not readable", s);
}
put_net2(buffer, TFTP_DATA);
put_net2(buffer + 2, block_nr);
if (rd < (int)s->blksize_val) {
tftp_remove_session(s);
} else {
tftp_update_timestamp(s);
}
return (rd + 4);
}
int tftp_send_ack(Bit8u *buffer, unsigned block_nr)
{
put_net2(buffer, TFTP_ACK);
put_net2(buffer + 2, block_nr);
return 4;
}
int tftp_send_optack(Bit8u *buffer, tftp_session_t *s)
{
Bit8u *p = buffer;
put_net2(p, TFTP_OPTACK);
p += 2;
if (s->options & TFTP_OPTION_TSIZE) {
*p++='t'; *p++='s'; *p++='i'; *p++='z'; *p++='e'; *p++='\0';
sprintf((char *)p, "%lu", (unsigned long)s->tsize_val);
p += strlen((const char *)p) + 1;
}
if (s->options & TFTP_OPTION_BLKSIZE) {
*p++='b'; *p++='l'; *p++='k'; *p++='s'; *p++='i'; *p++='z'; *p++='e'; *p++='\0';
sprintf((char *)p, "%u", s->blksize_val);
p += strlen((const char *)p) + 1;
}
if (s->options & TFTP_OPTION_TIMEOUT) {
*p++='t'; *p++='i'; *p++='m'; *p++='e'; *p++='o'; *p++='u'; *p++='t'; *p++='\0';
sprintf((char *)p, "%u", s->timeout_val);
p += strlen((const char *)p) + 1;
}
tftp_update_timestamp(s);
return (p - buffer);
}
void vnet_server_c::tftp_parse_options(const char *mode, const Bit8u *data,
unsigned data_len, tftp_session_t *s)
{
while (mode < (const char*)data + data_len) {
if (memcmp(mode, "octet\0", 6) == 0) {
s->options |= TFTP_OPTION_OCTET;
mode += 6;
} else if (memcmp(mode, "tsize\0", 6) == 0) {
s->options |= TFTP_OPTION_TSIZE; // size needed
mode += 6;
if (s->iswrite) {
s->tsize_val = atoi(mode);
}
mode += strlen(mode)+1;
} else if (memcmp(mode, "blksize\0", 8) == 0) {
s->options |= TFTP_OPTION_BLKSIZE;
mode += 8;
s->blksize_val = atoi(mode);
if (s->blksize_val > TFTP_BUFFER_SIZE) {
BX_ERROR(("tftp req: blksize value %d not supported - using %d instead",
s->blksize_val, TFTP_BUFFER_SIZE));
s->blksize_val = TFTP_BUFFER_SIZE;
}
mode += strlen(mode)+1;
} else if (memcmp(mode, "timeout\0", 8) == 0) {
s->options |= TFTP_OPTION_TIMEOUT;
mode += 8;
s->timeout_val = atoi(mode);
if ((s->timeout_val < 1) || (s->timeout_val > 255)) {
BX_ERROR(("tftp req: timeout value %d not supported - using %d instead",
s->timeout_val, TFTP_DEFAULT_TIMEOUT));
s->timeout_val = TFTP_DEFAULT_TIMEOUT;
}
mode += strlen(mode)+1;
} else {
BX_ERROR(("tftp req: unknown option %s", mode));
break;
}
}
}
int vnet_server_c::udpipv4_tftp_handler(void *this_ptr, const Bit8u *ipheader,
unsigned ipheader_len, unsigned sourceport, unsigned targetport,
const Bit8u *data, unsigned data_len, Bit8u *reply)
{
return ((vnet_server_c *)this_ptr)->udpipv4_tftp_handler_ns(ipheader,
ipheader_len, sourceport, targetport, data, data_len, reply);
}
int vnet_server_c::udpipv4_tftp_handler_ns(const Bit8u *ipheader,
unsigned ipheader_len, unsigned sourceport, unsigned targetport,
const Bit8u *data, unsigned data_len, Bit8u *reply)
{
FILE *fp;
unsigned block_nr;
unsigned tftp_len;
unsigned req_tid = sourceport;
tftp_session_t *s;
char msg[BX_PATHNAME_LEN + 16];
tftp_timeout_check();
s = tftp_find_session(req_tid);
switch (get_net2(data)) {
case TFTP_RRQ:
{
if (s != NULL) {
tftp_remove_session(s);
}
strncpy((char*)reply, (const char*)data + 2, data_len - 2);
reply[data_len - 4] = 0;
s = tftp_new_session(req_tid, 0, tftp_root, (const char*)reply);
if (strlen(s->filename) == 0) {
return tftp_send_error(reply, 1, "Illegal file name", s);
}
fp = fopen(s->filename, "rb");
if (!fp) {
sprintf(msg, "File not found: %s", s->filename);
return tftp_send_error(reply, 1, msg, s);
} else {
fclose(fp);
}
// options
if (strlen((char*)reply) < data_len - 2) {
const char *mode = (const char*)data + 2 + strlen((char*)reply) + 1;
tftp_parse_options(mode, data, data_len, s);
}
if (!(s->options & TFTP_OPTION_OCTET)) {
return tftp_send_error(reply, 4, "Unsupported transfer mode", NULL);
}
if (s->options & TFTP_OPTION_TSIZE) {
struct stat stbuf;
if (stat(s->filename, &stbuf) < 0) {
s->options &= ~TFTP_OPTION_TSIZE;
} else {
s->tsize_val = (size_t)stbuf.st_size;
BX_DEBUG(("TFTP RRQ: filesize=%lu", (unsigned long)s->tsize_val));
}
}
if ((s->options & ~TFTP_OPTION_OCTET) > 0) {
return tftp_send_optack(reply, s);
} else {
return tftp_send_data(reply, 1, s);
}
}
break;
case TFTP_WRQ:
{
if (s != NULL) {
tftp_remove_session(s);
}
strncpy((char*)reply, (const char*)data + 2, data_len - 2);
reply[data_len - 4] = 0;
s = tftp_new_session(req_tid, 1, tftp_root, (const char*)reply);
if (strlen(s->filename) == 0) {
return tftp_send_error(reply, 1, "Illegal file name", s);
}
// options
if (strlen((char*)reply) < data_len - 2) {
const char *mode = (const char*)data + 2 + strlen((char*)reply) + 1;
tftp_parse_options(mode, data, data_len, s);
}
if (!(s->options & TFTP_OPTION_OCTET)) {
return tftp_send_error(reply, 4, "Unsupported transfer mode", NULL);
}
fp = fopen(s->filename, "rb");
if (fp) {
fclose(fp);
return tftp_send_error(reply, 6, "File exists", s);
}
fp = fopen(s->filename, "wb");
if (!fp) {
return tftp_send_error(reply, 2, "Access violation", s);
}
fclose(fp);
if ((s->options & ~TFTP_OPTION_OCTET) > 0) {
return tftp_send_optack(reply, s);
} else {
tftp_update_timestamp(s);
return tftp_send_ack(reply, 0);
}
}
break;
case TFTP_DATA:
if (s != NULL) {
if (s->iswrite == 1) {
block_nr = get_net2(data + 2);
strncpy((char*)reply, (const char*)data + 4, data_len - 4);
tftp_len = data_len - 4;
reply[tftp_len] = 0;
if (tftp_len <= s->blksize_val) {
fp = fopen(s->filename, "ab");
if (!fp) {
return tftp_send_error(reply, 2, "Access violation", s);
}
if (fseek(fp, (block_nr - 1) * TFTP_BUFFER_SIZE, SEEK_SET) < 0) {
fclose(fp);
return tftp_send_error(reply, 3, "Block not seekable", s);
}
fwrite(reply, 1, tftp_len, fp);
fclose(fp);
if (tftp_len < s->blksize_val) {
tftp_remove_session(s);
} else {
tftp_update_timestamp(s);
}
return tftp_send_ack(reply, block_nr);
} else {
return tftp_send_error(reply, 4, "Illegal request", s);
}
} else {
return tftp_send_error(reply, 4, "Illegal request", s);
}
} else {
return tftp_send_error(reply, 5, "Unknown transfer ID", s);
}
break;
case TFTP_ACK:
if (s != NULL) {
if (s->iswrite == 0) {
return tftp_send_data(reply, get_net2(data + 2) + 1, s);
} else {
return tftp_send_error(reply, 4, "Illegal request", s);
}
}
break;
case TFTP_ERROR:
if (s != NULL) {
tftp_remove_session(s);
}
break;
default:
BX_ERROR(("TFTP unknown opt %d", get_net2(data)));
}
return 0;
}
int vnet_server_c::udpipv4_dns_handler(void *this_ptr, const Bit8u *ipheader,
unsigned ipheader_len, unsigned sourceport, unsigned targetport,
const Bit8u *data, unsigned data_len, Bit8u *reply)
{
return ((vnet_server_c *)this_ptr)->udpipv4_dns_handler_ns(ipheader,
ipheader_len, sourceport, targetport, data, data_len, reply);
}
int vnet_server_c::udpipv4_dns_handler_ns(const Bit8u *ipheader,
unsigned ipheader_len, unsigned sourceport, unsigned targetport,
const Bit8u *data, unsigned data_len, Bit8u *reply)
{
char host[256], suffix[16];
Bit8u len1, p1, p2;
Bit8u ipaddr[4];
Bit16u val16[6], qtype, qclass;
bx_bool host_found = 0;
int i, n, rlen = 0, tmpval[4];
for (i = 0; i < 6; i++) {
val16[i] = get_net2(&data[i * 2]);
}
if ((val16[1] != 0x0100) || (val16[2] != 0x0001))
return 0;
p1 = 12;
p2 = 0;
while (data[p1] != 0) {
len1 = data[p1];
if (p2 > 0) {
host[p2++] = '.';
}
memcpy(&host[p2], &data[p1 + 1], len1);
p2 += len1;
p1 += (len1 + 1);
};
host[p2] = 0;
qtype = get_net2(&data[p1 + 1]);
qclass = get_net2(&data[p1 + 3]);
if (((qtype != 1) && (qtype != 12)) || (qclass != 1)) {
memcpy(reply, data, data_len);
put_net2(&reply[2], 0x8181);
return data_len;
}
if (qtype == 12) {
n = sscanf(host, "%d.%d.%d.%d.%s", &tmpval[3], &tmpval[2], &tmpval[1], &tmpval[0], suffix);
if ((n == 5) && (!strcmp(suffix, "in-addr.arpa"))) {
for (i = 0; i < 4; i++) {
ipaddr[i] = (Bit8u)tmpval[i];
}
if (!memcmp(&ipaddr, dhcp->srv_ipv4addr[VNET_SRV], 4)) {
host_found = 1;
strcpy(host, "vnet");
} else if (!memcmp(&ipaddr, dhcp->srv_ipv4addr[VNET_DNS], 4)) {
host_found = 1;
strcpy(host, "vnet-dns");
} else if (!memcmp(&ipaddr, dhcp->srv_ipv4addr[VNET_MISC], 4)) {
host_found = 1;
strcpy(host, "vnet-ftp");
} else {
for (i = 0; i < VNET_MAX_CLIENTS; i++) {
if (client[i].init) {
if (!memcmp(&ipaddr, client[i].ipv4addr, 4)) {
host_found = 1;
strcpy(host, client[i].hostname);
break;
}
}
}
}
} else {
memcpy(reply, data, data_len);
put_net2(&reply[2], 0x8181);
return data_len;
}
} else {
if (!stricmp(host, "vnet")) {
host_found = 1;
memcpy(&ipaddr, dhcp->srv_ipv4addr[VNET_SRV], 4);
} else if (!stricmp(host, "vnet-dns")) {
host_found = 1;
memcpy(&ipaddr, dhcp->srv_ipv4addr[VNET_DNS], 4);
} else if (!stricmp(host, "vnet-ftp")) {
host_found = 1;
memcpy(&ipaddr, dhcp->srv_ipv4addr[VNET_MISC], 4);
} else {
for (i = 0; i < VNET_MAX_CLIENTS; i++) {
if (client[i].init) {
if (!stricmp(host, client[i].hostname)) {
host_found = 1;
memcpy(&ipaddr, client[i].ipv4addr, 4);
break;
}
}
}
}
}
if (host_found) {
memcpy(reply, data, data_len);
put_net2(&reply[2], 0x8180);
put_net2(&reply[6], 0x0001);
reply[data_len] = 0xc0;
reply[data_len + 1] = 12;
put_net2(&reply[data_len + 2], qtype);
put_net2(&reply[data_len + 4], qclass);
put_net4(&reply[data_len + 6], 86400);
if (qtype == 12) {
len1 = (Bit8u)strlen(host);
put_net2(&reply[data_len + 10], len1 + 2);
reply[data_len + 12] = len1;
memcpy(&reply[data_len + 13], host, len1);
reply[data_len + len1 + 13] = 0;
rlen = data_len + len1 + 14;
} else {
put_net2(&reply[data_len + 10], 4);
memcpy(&reply[data_len + 12], ipaddr, 4);
rlen = data_len + 16;
}
} else {
BX_ERROR(("DNS: unknown host '%s'", host));
memcpy(reply, data, data_len);
put_net2(&reply[2], 0x8183);
rlen = data_len;
}
return rlen;
}
#endif /* if BX_NETWORKING */
Reference in New Issue View Git Blame Copy Permalink