|
| 1 | +// SPDX-License-Identifier: MIT |
| 2 | + |
| 3 | +// SPDX-FileCopyrightText: 2022 Stefan Schmidt |
| 4 | + |
| 5 | +#include "net.h" |
| 6 | + |
| 7 | +#include <assert.h> |
| 8 | +#include <stdbool.h> |
| 9 | + |
| 10 | +#include <nxdk/configsector.h> |
| 11 | +#include <netif/etharp.h> |
| 12 | +#include <lwip/dhcp.h> |
| 13 | +#include <lwip/dns.h> |
| 14 | +#include <lwip/netif.h> |
| 15 | +#include <lwip/netifapi.h> |
| 16 | +#include <lwip/tcpip.h> |
| 17 | +#include <lwip/timeouts.h> |
| 18 | +#include <pktdrv.h> |
| 19 | +#include <xboxkrnl/xboxkrnl.h> |
| 20 | + |
| 21 | + |
| 22 | +err_t nforceif_init(struct netif *netif); |
| 23 | + |
| 24 | +struct netif *g_pnetif; |
| 25 | +static struct netif nforce_netif; |
| 26 | + |
| 27 | +static void tcpip_init_done(void *arg) |
| 28 | +{ |
| 29 | + KEVENT *init_complete = arg; |
| 30 | + KeSetEvent(init_complete, IO_NO_INCREMENT, FALSE); |
| 31 | +} |
| 32 | + |
| 33 | +static void packet_timer(void *arg) |
| 34 | +{ |
| 35 | + LWIP_UNUSED_ARG(arg); |
| 36 | + Pktdrv_ReceivePackets(); |
| 37 | + sys_timeout(5 /* ms */, packet_timer, NULL); |
| 38 | +} |
| 39 | + |
| 40 | +int nxNetInit(const nx_net_parameters_t *parameters) |
| 41 | +{ |
| 42 | + ip4_addr_t ipaddr, netmask, gateway; |
| 43 | + bool ipv4_dhcp; |
| 44 | + ip_addr_t dns[2]; |
| 45 | + bool dns_override = false; |
| 46 | + |
| 47 | + if (!parameters || parameters->ipv4_mode == NX_NET_AUTO) { |
| 48 | + nxdk_network_config_sector_t configSector; |
| 49 | + if (!nxLoadNetworkConfig(&configSector)) { |
| 50 | + return -1; |
| 51 | + } |
| 52 | + |
| 53 | + if (configSector.dhcpFlags & NXDK_NETWORK_CONFIG_MANUAL_IP) { |
| 54 | + ipv4_dhcp = false; |
| 55 | + IP4_ADDR(&ipaddr, configSector.manual.ip >> 0 & 0xff, |
| 56 | + (configSector.manual.ip >> 8 & 0xff), |
| 57 | + (configSector.manual.ip >> 16 & 0xff), |
| 58 | + (configSector.manual.ip >> 24 & 0xff)); |
| 59 | + IP4_ADDR(&gateway, configSector.manual.defaultGateway >> 0 & 0xff, |
| 60 | + (configSector.manual.defaultGateway >> 8 & 0xff), |
| 61 | + (configSector.manual.defaultGateway >> 16 & 0xff), |
| 62 | + (configSector.manual.defaultGateway >> 24 & 0xff)); |
| 63 | + IP4_ADDR(&netmask, configSector.manual.subnetMask >> 0 & 0xff, |
| 64 | + (configSector.manual.subnetMask >> 8 & 0xff), |
| 65 | + (configSector.manual.subnetMask >> 16 & 0xff), |
| 66 | + (configSector.manual.subnetMask >> 24 & 0xff));; |
| 67 | + } else { |
| 68 | + ipv4_dhcp = true; |
| 69 | + } |
| 70 | + |
| 71 | + if (configSector.dhcpFlags & NXDK_NETWORK_CONFIG_MANUAL_DNS) { |
| 72 | + IP4_ADDR(&dns[0].u_addr.ip4, configSector.manual.primaryDns >> 0 & 0xff, |
| 73 | + (configSector.manual.primaryDns >> 8 & 0xff), |
| 74 | + (configSector.manual.primaryDns >> 16 & 0xff), |
| 75 | + (configSector.manual.primaryDns >> 24 & 0xff)); |
| 76 | + IP4_ADDR(&dns[1].u_addr.ip4, configSector.manual.secondaryDns >> 0 & 0xff, |
| 77 | + (configSector.manual.secondaryDns >> 8 & 0xff), |
| 78 | + (configSector.manual.secondaryDns >> 16 & 0xff), |
| 79 | + (configSector.manual.secondaryDns >> 24 & 0xff)); |
| 80 | + dns[0].type = IPADDR_TYPE_V4; |
| 81 | + dns[1].type = IPADDR_TYPE_V4; |
| 82 | + dns_override = true; |
| 83 | + } |
| 84 | + } else if (parameters->ipv4_mode == NX_NET_DHCP) { |
| 85 | + ipv4_dhcp = true; |
| 86 | + } else if (parameters->ipv4_mode == NX_NET_STATIC) { |
| 87 | + ipv4_dhcp = false; |
| 88 | + IP4_ADDR(&ipaddr, parameters->ipv4_ip >> 0 & 0xff, |
| 89 | + (parameters->ipv4_ip >> 8 & 0xff), |
| 90 | + (parameters->ipv4_ip >> 16 & 0xff), |
| 91 | + (parameters->ipv4_ip >> 24 & 0xff)); |
| 92 | + IP4_ADDR(&gateway, parameters->ipv4_gateway >> 0 & 0xff, |
| 93 | + (parameters->ipv4_gateway >> 8 & 0xff), |
| 94 | + (parameters->ipv4_gateway >> 16 & 0xff), |
| 95 | + (parameters->ipv4_gateway >> 24 & 0xff)); |
| 96 | + IP4_ADDR(&netmask, parameters->ipv4_netmask >> 0 & 0xff, |
| 97 | + (parameters->ipv4_netmask >> 8 & 0xff), |
| 98 | + (parameters->ipv4_netmask >> 16 & 0xff), |
| 99 | + (parameters->ipv4_netmask >> 24 & 0xff)); |
| 100 | + } else { |
| 101 | + assert(false); |
| 102 | + return -1; |
| 103 | + } |
| 104 | + |
| 105 | + if (ipv4_dhcp) { |
| 106 | + IP4_ADDR(&gateway, 0, 0, 0, 0); |
| 107 | + IP4_ADDR(&ipaddr, 0, 0, 0, 0); |
| 108 | + IP4_ADDR(&netmask, 0, 0, 0, 0); |
| 109 | + } |
| 110 | + |
| 111 | + KEVENT tcpip_init_complete; |
| 112 | + KeInitializeEvent(&tcpip_init_complete, SynchronizationEvent, FALSE); |
| 113 | + tcpip_init(tcpip_init_done, &tcpip_init_complete); |
| 114 | + KeWaitForSingleObject(&tcpip_init_complete, Executive, KernelMode, FALSE, NULL); |
| 115 | + |
| 116 | + g_pnetif = &nforce_netif; |
| 117 | + err_t err = netifapi_netif_add(&nforce_netif, &ipaddr, &netmask, &gateway, NULL, nforceif_init, ethernet_input); |
| 118 | + if (err != ERR_OK) { |
| 119 | + debugPrint("netif_add failed\n"); |
| 120 | + return -1; |
| 121 | + } |
| 122 | + |
| 123 | + netifapi_netif_set_default(&nforce_netif); |
| 124 | + netifapi_netif_set_up(&nforce_netif); |
| 125 | + |
| 126 | + packet_timer(NULL); |
| 127 | + |
| 128 | + if (ipv4_dhcp) { |
| 129 | + netifapi_dhcp_start(&nforce_netif); |
| 130 | + |
| 131 | + LARGE_INTEGER duration; |
| 132 | + duration.QuadPart = ((LONGLONG)1000) * -10000; |
| 133 | + int i = 0; |
| 134 | + while (dhcp_supplied_address(&nforce_netif) == 0) { |
| 135 | + i++; |
| 136 | + if (i == 10) { |
| 137 | + return -2; |
| 138 | + } |
| 139 | + KeDelayExecutionThread(KernelMode, FALSE, &duration); |
| 140 | + } |
| 141 | + } |
| 142 | + |
| 143 | + if (parameters) { |
| 144 | + if (parameters->ipv4_dns1 != 0 || parameters->ipv4_dns2 != 0) { |
| 145 | + IP4_ADDR(&dns[0].u_addr.ip4, parameters->ipv4_dns1 >> 0 & 0xff, |
| 146 | + (parameters->ipv4_dns1 >> 8 & 0xff), |
| 147 | + (parameters->ipv4_dns1 >> 16 & 0xff), |
| 148 | + (parameters->ipv4_dns1 >> 24 & 0xff)); |
| 149 | + IP4_ADDR(&dns[1].u_addr.ip4, parameters->ipv4_dns2 >> 0 & 0xff, |
| 150 | + (parameters->ipv4_dns2 >> 8 & 0xff), |
| 151 | + (parameters->ipv4_dns2 >> 16 & 0xff), |
| 152 | + (parameters->ipv4_dns2 >> 24 & 0xff)); |
| 153 | + dns[0].type = IPADDR_TYPE_V4; |
| 154 | + dns[1].type = IPADDR_TYPE_V4; |
| 155 | + dns_override = true; |
| 156 | + } |
| 157 | + } |
| 158 | + |
| 159 | + if (dns_override) { |
| 160 | + dns_setserver(2, dns); |
| 161 | + } |
| 162 | + |
| 163 | + return 0; |
| 164 | +} |
0 commit comments