#include <sys/systm.h>
#include <sys/kern_control.h>
#include <net/kpi_protocol.h>
#include <net/kpi_interface.h>
#include <sys/socket.h>
#include <net/if.h>
#include <net/if_types.h>
#include <net/bpf.h>
#include <net/if_utun.h>
#include <libkern/OSMalloc.h>
#include <libkern/OSAtomic.h>
#include <sys/mbuf.h>
#include <sys/sockio.h>
#include <netinet/in.h>
#include <netinet6/in6_var.h>
#include <netinet6/in6_var.h>
#include <sys/kauth.h>
static errno_t utun_ctl_connect(kern_ctl_ref kctlref, struct sockaddr_ctl *sac,
void **unitinfo);
static errno_t utun_ctl_disconnect(kern_ctl_ref kctlref, u_int32_t unit,
void *unitinfo);
static errno_t utun_ctl_send(kern_ctl_ref kctlref, u_int32_t unit,
void *unitinfo, mbuf_t m, int flags);
static errno_t utun_ctl_getopt(kern_ctl_ref kctlref, u_int32_t unit, void *unitinfo,
int opt, void *data, size_t *len);
static errno_t utun_ctl_setopt(kern_ctl_ref kctlref, u_int32_t unit, void *unitinfo,
int opt, void *data, size_t len);
static errno_t utun_output(ifnet_t interface, mbuf_t data);
static errno_t utun_demux(ifnet_t interface, mbuf_t data, char *frame_header,
protocol_family_t *protocol);
static errno_t utun_framer(ifnet_t interface, mbuf_t *packet,
const struct sockaddr *dest, const char *desk_linkaddr,
const char *frame_type
#if KPI_INTERFACE_EMBEDDED
,
u_int32_t *prepend_len, u_int32_t *postpend_len
#endif
);
static errno_t utun_add_proto(ifnet_t interface, protocol_family_t protocol,
const struct ifnet_demux_desc *demux_array,
u_int32_t demux_count);
static errno_t utun_del_proto(ifnet_t interface, protocol_family_t protocol);
static errno_t utun_ioctl(ifnet_t interface, u_long cmd, void *data);
static void utun_detached(ifnet_t interface);
static errno_t utun_attach_proto(ifnet_t interface, protocol_family_t proto);
static errno_t utun_proto_input(ifnet_t interface, protocol_family_t protocol,
mbuf_t m, char *frame_header);
static errno_t utun_proto_pre_output(ifnet_t interface, protocol_family_t protocol,
mbuf_t *packet, const struct sockaddr *dest, void *route,
char *frame_type, char *link_layer_dest);
__private_extern__ errno_t utun_pkt_input (struct utun_pcb *pcb, mbuf_t m);
static kern_ctl_ref utun_kctlref;
static u_int32_t utun_family;
static OSMallocTag utun_malloc_tag;
static SInt32 utun_ifcount = 0;
void*
utun_alloc(size_t size)
{
size_t *mem = OSMalloc(size + sizeof(size_t), utun_malloc_tag);
if (mem) {
*mem = size + sizeof(size_t);
mem++;
}
return (void*)mem;
}
void
utun_free(void *ptr)
{
size_t *size = ptr;
size--;
OSFree(size, *size, utun_malloc_tag);
}
errno_t
utun_register_control(void)
{
struct kern_ctl_reg kern_ctl;
errno_t result = 0;
utun_malloc_tag = OSMalloc_Tagalloc(UTUN_CONTROL_NAME, OSMT_DEFAULT);
result = mbuf_tag_id_find(UTUN_CONTROL_NAME, &utun_family);
if (result != 0) {
printf("utun_register_control - mbuf_tag_id_find_internal failed: %d\n", result);
return result;
}
bzero(&kern_ctl, sizeof(kern_ctl));
strncpy(kern_ctl.ctl_name, UTUN_CONTROL_NAME, sizeof(kern_ctl.ctl_name));
kern_ctl.ctl_name[sizeof(kern_ctl.ctl_name) - 1] = 0;
kern_ctl.ctl_flags = CTL_FLAG_PRIVILEGED;
kern_ctl.ctl_sendsize = 64 * 1024;
kern_ctl.ctl_recvsize = 64 * 1024;
kern_ctl.ctl_connect = utun_ctl_connect;
kern_ctl.ctl_disconnect = utun_ctl_disconnect;
kern_ctl.ctl_send = utun_ctl_send;
kern_ctl.ctl_setopt = utun_ctl_setopt;
kern_ctl.ctl_getopt = utun_ctl_getopt;
result = ctl_register(&kern_ctl, &utun_kctlref);
if (result != 0) {
printf("utun_register_control - ctl_register failed: %d\n", result);
return result;
}
if ((result = proto_register_plumber(PF_INET, utun_family,
utun_attach_proto, NULL)) != 0) {
printf("utun_register_control - proto_register_plumber(PF_INET, %d) failed: %d\n",
utun_family, result);
ctl_deregister(utun_kctlref);
return result;
}
if ((result = proto_register_plumber(PF_INET6, utun_family,
utun_attach_proto, NULL)) != 0) {
proto_unregister_plumber(PF_INET, utun_family);
ctl_deregister(utun_kctlref);
printf("utun_register_control - proto_register_plumber(PF_INET6, %d) failed: %d\n",
utun_family, result);
return result;
}
return 0;
}
static errno_t
utun_ctl_connect(
kern_ctl_ref kctlref,
struct sockaddr_ctl *sac,
void **unitinfo)
{
struct ifnet_init_params utun_init;
struct utun_pcb *pcb;
errno_t result;
struct ifnet_stats_param stats;
pcb = utun_alloc(sizeof(*pcb));
if (pcb == NULL)
return ENOMEM;
bzero(pcb, sizeof(*pcb));
*unitinfo = pcb;
pcb->utun_ctlref = kctlref;
pcb->utun_unit = sac->sc_unit;
printf("utun_ctl_connect: creating interface utun%d\n", pcb->utun_unit - 1);
bzero(&utun_init, sizeof(utun_init));
utun_init.name = "utun";
utun_init.unit = pcb->utun_unit - 1;
utun_init.family = utun_family;
utun_init.type = IFT_OTHER;
utun_init.output = utun_output;
utun_init.demux = utun_demux;
utun_init.framer = utun_framer;
utun_init.add_proto = utun_add_proto;
utun_init.del_proto = utun_del_proto;
utun_init.softc = pcb;
utun_init.ioctl = utun_ioctl;
utun_init.detach = utun_detached;
result = ifnet_allocate(&utun_init, &pcb->utun_ifp);
if (result != 0) {
printf("utun_ctl_connect - ifnet_allocate failed: %d\n", result);
utun_free(pcb);
return result;
}
OSIncrementAtomic(&utun_ifcount);
ifnet_set_mtu(pcb->utun_ifp, 1500);
ifnet_set_flags(pcb->utun_ifp, IFF_UP | IFF_MULTICAST | IFF_POINTOPOINT, 0xffff);
ifnet_set_eflags(pcb->utun_ifp, IFEF_NOAUTOIPV6LL, IFEF_NOAUTOIPV6LL);
bzero(&stats, sizeof(struct ifnet_stats_param));
ifnet_set_stat(pcb->utun_ifp, &stats);
result = ifnet_attach(pcb->utun_ifp, NULL);
if (result != 0) {
printf("utun_ctl_connect - ifnet_allocate failed: %d\n", result);
ifnet_release(pcb->utun_ifp);
utun_free(pcb);
}
if (result == 0)
bpfattach(pcb->utun_ifp, DLT_NULL, 4);
if (result == 0)
ifnet_set_flags(pcb->utun_ifp, IFF_RUNNING, IFF_RUNNING);
return result;
}
static errno_t
utun_detach_ip(
ifnet_t interface,
protocol_family_t protocol,
socket_t pf_socket)
{
errno_t result = EPROTONOSUPPORT;
if (protocol == PF_INET) {
struct ifreq ifr;
bzero(&ifr, sizeof(ifr));
snprintf(ifr.ifr_name, sizeof(ifr.ifr_name), "%s%d",
ifnet_name(interface), ifnet_unit(interface));
result = sock_ioctl(pf_socket, SIOCPROTODETACH, &ifr);
}
else if (protocol == PF_INET6) {
struct in6_ifreq ifr6;
bzero(&ifr6, sizeof(ifr6));
snprintf(ifr6.ifr_name, sizeof(ifr6.ifr_name), "%s%d",
ifnet_name(interface), ifnet_unit(interface));
result = sock_ioctl(pf_socket, SIOCPROTODETACH_IN6, &ifr6);
}
return result;
}
static void
utun_remove_address(
ifnet_t interface,
protocol_family_t protocol,
ifaddr_t address,
socket_t pf_socket)
{
errno_t result = 0;
if (protocol == PF_INET) {
struct ifreq ifr;
bzero(&ifr, sizeof(ifr));
snprintf(ifr.ifr_name, sizeof(ifr.ifr_name), "%s%d",
ifnet_name(interface), ifnet_unit(interface));
result = ifaddr_address(address, &ifr.ifr_addr, sizeof(ifr.ifr_addr));
if (result != 0) {
printf("utun_remove_address - ifaddr_address failed: %d", result);
}
else {
result = sock_ioctl(pf_socket, SIOCDIFADDR, &ifr);
if (result != 0) {
printf("utun_remove_address - SIOCDIFADDR failed: %d", result);
}
}
}
else if (protocol == PF_INET6) {
struct in6_ifreq ifr6;
bzero(&ifr6, sizeof(ifr6));
snprintf(ifr6.ifr_name, sizeof(ifr6.ifr_name), "%s%d",
ifnet_name(interface), ifnet_unit(interface));
result = ifaddr_address(address, (struct sockaddr*)&ifr6.ifr_addr,
sizeof(ifr6.ifr_addr));
if (result != 0) {
printf("utun_remove_address - ifaddr_address failed (v6): %d",
result);
}
else {
result = sock_ioctl(pf_socket, SIOCDIFADDR_IN6, &ifr6);
if (result != 0) {
printf("utun_remove_address - SIOCDIFADDR_IN6 failed: %d",
result);
}
}
}
}
static void
utun_cleanup_family(
ifnet_t interface,
protocol_family_t protocol)
{
errno_t result = 0;
socket_t pf_socket = NULL;
ifaddr_t *addresses = NULL;
int i;
if (protocol != PF_INET && protocol != PF_INET6) {
printf("utun_cleanup_family - invalid protocol family %d\n", protocol);
return;
}
result = sock_socket(protocol, SOCK_DGRAM, 0, NULL, NULL, &pf_socket);
if (result != 0) {
if (result != EAFNOSUPPORT)
printf("utun_cleanup_family - failed to create %s socket: %d\n",
protocol == PF_INET ? "IP" : "IPv6", result);
goto cleanup;
}
sock_setpriv(pf_socket, 1);
result = utun_detach_ip(interface, protocol, pf_socket);
if (result == 0 || result == ENXIO) {
goto cleanup;
}
else if (result != EBUSY) {
printf("utun_cleanup_family - utun_detach_ip failed: %d\n", result);
goto cleanup;
}
result = ifnet_get_address_list_family(interface, &addresses, protocol);
if (result != 0) {
printf("fnet_get_address_list_family(%s%d, 0xblah, %s) - failed: %d\n",
ifnet_name(interface), ifnet_unit(interface),
protocol == PF_INET ? "PF_INET" : "PF_INET6", result);
goto cleanup;
}
for (i = 0; addresses[i] != 0; i++) {
utun_remove_address(interface, protocol, addresses[i], pf_socket);
}
ifnet_free_address_list(addresses);
addresses = NULL;
result = utun_detach_ip(interface, protocol, pf_socket);
if (result != 0 && result != ENXIO) {
printf("utun_cleanup_family - utun_detach_ip failed: %d\n", result);
}
cleanup:
if (pf_socket != NULL)
sock_close(pf_socket);
if (addresses != NULL)
ifnet_free_address_list(addresses);
}
static errno_t
utun_ctl_disconnect(
__unused kern_ctl_ref kctlref,
__unused u_int32_t unit,
void *unitinfo)
{
struct utun_pcb *pcb = unitinfo;
ifnet_t ifp = pcb->utun_ifp;
errno_t result = 0;
utun_cleanup_crypto(pcb);
pcb->utun_ctlref = NULL;
pcb->utun_unit = 0;
utun_cleanup_family(ifp, AF_INET);
utun_cleanup_family(ifp, AF_INET6);
if ((result = ifnet_detach(ifp)) != 0) {
printf("utun_ctl_disconnect - ifnet_detach failed: %d\n", result);
}
if ((result = ifnet_release(ifp)) != 0) {
printf("utun_ctl_disconnect - ifnet_release failed: %d\n", result);
}
return 0;
}
static errno_t
utun_ctl_send(
__unused kern_ctl_ref kctlref,
__unused u_int32_t unit,
void *unitinfo,
mbuf_t m,
__unused int flags)
{
return utun_pkt_input((struct utun_pcb *)unitinfo, m);
}
static errno_t
utun_ctl_setopt(
__unused kern_ctl_ref kctlref,
__unused u_int32_t unit,
void *unitinfo,
int opt,
void *data,
size_t len)
{
struct utun_pcb *pcb = unitinfo;
errno_t result = 0;
switch (opt) {
case UTUN_OPT_FLAGS:
case UTUN_OPT_EXT_IFDATA_STATS:
if (kauth_cred_issuser(kauth_cred_get()) == 0) {
return EPERM;
}
break;
}
switch (opt) {
case UTUN_OPT_FLAGS:
if (len != sizeof(u_int32_t))
result = EMSGSIZE;
else
pcb->utun_flags = *(u_int32_t *)data;
break;
case UTUN_OPT_ENABLE_CRYPTO:
result = utun_ctl_enable_crypto(kctlref, unit, unitinfo, opt, data, len);
break;
case UTUN_OPT_CONFIG_CRYPTO_KEYS:
result = utun_ctl_config_crypto_keys(kctlref, unit, unitinfo, opt, data, len);
break;
case UTUN_OPT_UNCONFIG_CRYPTO_KEYS:
result = utun_ctl_unconfig_crypto_keys(kctlref, unit, unitinfo, opt, data, len);
break;
case UTUN_OPT_DISABLE_CRYPTO:
result = utun_ctl_disable_crypto(kctlref, unit, unitinfo, opt, data, len);
break;
case UTUN_OPT_STOP_CRYPTO_DATA_TRAFFIC:
result = utun_ctl_stop_crypto_data_traffic(kctlref, unit, unitinfo, opt, data, len);
break;
case UTUN_OPT_START_CRYPTO_DATA_TRAFFIC:
result = utun_ctl_start_crypto_data_traffic(kctlref, unit, unitinfo, opt, data, len);
break;
case UTUN_OPT_EXT_IFDATA_STATS:
if (len != sizeof(int)) {
result = EMSGSIZE;
break;
}
pcb->utun_ext_ifdata_stats = (*(int *)data) ? 1 : 0;
break;
case UTUN_OPT_INC_IFDATA_STATS_IN:
case UTUN_OPT_INC_IFDATA_STATS_OUT: {
struct utun_stats_param *utsp = (struct utun_stats_param *)data;
if (utsp == NULL || len < sizeof(struct utun_stats_param)) {
result = EINVAL;
break;
}
if (!pcb->utun_ext_ifdata_stats) {
result = EINVAL;
break;
}
if (opt == UTUN_OPT_INC_IFDATA_STATS_IN)
ifnet_stat_increment_in(pcb->utun_ifp, utsp->utsp_packets,
utsp->utsp_bytes, utsp->utsp_errors);
else
ifnet_stat_increment_out(pcb->utun_ifp, utsp->utsp_packets,
utsp->utsp_bytes, utsp->utsp_errors);
break;
}
default:
result = ENOPROTOOPT;
break;
}
return result;
}
static errno_t
utun_ctl_getopt(
__unused kern_ctl_ref kctlref,
__unused u_int32_t unit,
void *unitinfo,
int opt,
void *data,
size_t *len)
{
struct utun_pcb *pcb = unitinfo;
errno_t result = 0;
switch (opt) {
case UTUN_OPT_FLAGS:
if (*len != sizeof(u_int32_t))
result = EMSGSIZE;
else
*(u_int32_t *)data = pcb->utun_flags;
break;
case UTUN_OPT_EXT_IFDATA_STATS:
if (*len != sizeof(int))
result = EMSGSIZE;
else
*(int *)data = (pcb->utun_ext_ifdata_stats) ? 1 : 0;
break;
case UTUN_OPT_IFNAME:
*len = snprintf(data, *len, "%s%d", ifnet_name(pcb->utun_ifp), ifnet_unit(pcb->utun_ifp)) + 1;
break;
case UTUN_OPT_GENERATE_CRYPTO_KEYS_IDX:
result = utun_ctl_generate_crypto_keys_idx(kctlref, unit, unitinfo, opt, data, len);
break;
default:
result = ENOPROTOOPT;
break;
}
return result;
}
static errno_t
utun_output(
ifnet_t interface,
mbuf_t data)
{
struct utun_pcb *pcb = ifnet_softc(interface);
errno_t result;
bpf_tap_out(pcb->utun_ifp, DLT_NULL, data, 0, 0);
if (pcb->utun_flags & UTUN_FLAGS_NO_OUTPUT) {
mbuf_freem(data);
return 0;
}
if (pcb->utun_ctlref) {
int length;
if ((pcb->utun_flags & (UTUN_FLAGS_CRYPTO | UTUN_FLAGS_CRYPTO_STOP_DATA_TRAFFIC)) == UTUN_FLAGS_CRYPTO) {
if (utun_pkt_crypto_output(pcb, &data) == 0) {
return 0;
}
}
length = mbuf_pkthdr_len(data);
result = ctl_enqueuembuf(pcb->utun_ctlref, pcb->utun_unit, data, CTL_DATA_EOR);
if (result != 0) {
mbuf_freem(data);
printf("utun_output - ctl_enqueuembuf failed: %d\n", result);
ifnet_stat_increment_out(interface, 0, 0, 1);
}
else {
if (!pcb->utun_ext_ifdata_stats)
ifnet_stat_increment_out(interface, 1, length, 0);
}
}
else
mbuf_freem(data);
return 0;
}
static errno_t
utun_demux(
__unused ifnet_t interface,
mbuf_t data,
__unused char *frame_header,
protocol_family_t *protocol)
{
while (data != NULL && mbuf_len(data) < 1) {
data = mbuf_next(data);
}
if (data == NULL)
return ENOENT;
*protocol = ntohl(*(u_int32_t *)mbuf_data(data));
return 0;
}
static errno_t
utun_framer(
__unused ifnet_t interface,
mbuf_t *packet,
__unused const struct sockaddr *dest,
__unused const char *desk_linkaddr,
const char *frame_type
#if KPI_INTERFACE_EMBEDDED
,
u_int32_t *prepend_len,
u_int32_t *postpend_len
#endif
)
{
if (mbuf_prepend(packet, sizeof(protocol_family_t), MBUF_DONTWAIT) != 0) {
printf("utun_framer - ifnet_output prepend failed\n");
ifnet_stat_increment_out(interface, 0, 0, 1);
return EJUSTRETURN;
}
#if KPI_INTERFACE_EMBEDDED
*prepend_len = sizeof(protocol_family_t);
*postpend_len = 0;
#endif
*(protocol_family_t *)mbuf_data(*packet) = htonl(*(protocol_family_t *)(uintptr_t)(size_t)frame_type);
return 0;
}
static errno_t
utun_add_proto(
__unused ifnet_t interface,
protocol_family_t protocol,
__unused const struct ifnet_demux_desc *demux_array,
__unused u_int32_t demux_count)
{
switch(protocol) {
case PF_INET:
return 0;
case PF_INET6:
return 0;
default:
break;
}
return ENOPROTOOPT;
}
static errno_t
utun_del_proto(
__unused ifnet_t interface,
__unused protocol_family_t protocol)
{
return 0;
}
static errno_t
utun_ioctl(
ifnet_t interface,
u_long command,
void *data)
{
errno_t result = 0;
switch(command) {
case SIOCSIFMTU:
ifnet_set_mtu(interface, ((struct ifreq*)data)->ifr_mtu);
break;
case SIOCSIFFLAGS:
break;
default:
result = EOPNOTSUPP;
}
return result;
}
static void
utun_detached(
ifnet_t interface)
{
struct utun_pcb *pcb = ifnet_softc(interface);
utun_free(pcb);
OSDecrementAtomic(&utun_ifcount);
}
static errno_t
utun_proto_input(
__unused ifnet_t interface,
protocol_family_t protocol,
mbuf_t m,
__unused char *frame_header)
{
mbuf_adj(m, sizeof(u_int32_t));
if (proto_input(protocol, m) != 0)
m_freem(m);
return 0;
}
static errno_t
utun_proto_pre_output(
__unused ifnet_t interface,
protocol_family_t protocol,
__unused mbuf_t *packet,
__unused const struct sockaddr *dest,
__unused void *route,
__unused char *frame_type,
__unused char *link_layer_dest)
{
*(protocol_family_t *)(void *)frame_type = protocol;
return 0;
}
static errno_t
utun_attach_proto(
ifnet_t interface,
protocol_family_t protocol)
{
struct ifnet_attach_proto_param proto;
errno_t result;
bzero(&proto, sizeof(proto));
proto.input = utun_proto_input;
proto.pre_output = utun_proto_pre_output;
result = ifnet_attach_protocol(interface, protocol, &proto);
if (result != 0 && result != EEXIST) {
printf("utun_attach_inet - ifnet_attach_protocol %d failed: %d\n",
protocol, result);
}
return result;
}
errno_t
utun_pkt_input (struct utun_pcb *pcb, mbuf_t m)
{
errno_t result;
protocol_family_t protocol;
mbuf_pkthdr_setrcvif(m, pcb->utun_ifp);
bpf_tap_in(pcb->utun_ifp, DLT_NULL, m, 0, 0);
if (pcb->utun_flags & UTUN_FLAGS_NO_INPUT) {
mbuf_freem(m);
return 0;
}
protocol = ntohl(*(u_int32_t *)mbuf_data(m));
if (protocol == AF_UTUN && pcb->utun_flags & UTUN_FLAGS_CRYPTO) {
if (utun_pkt_crypto_output(pcb, &m) == 0) {
return 0;
}
printf("%s: utun_pkt_crypto_output failed, flags %x\n", __FUNCTION__, pcb->utun_flags);
return EINVAL;
}
if (!pcb->utun_ext_ifdata_stats) {
struct ifnet_stat_increment_param incs;
bzero(&incs, sizeof(incs));
incs.packets_in = 1;
incs.bytes_in = mbuf_pkthdr_len(m);
result = ifnet_input(pcb->utun_ifp, m, &incs);
} else {
result = ifnet_input(pcb->utun_ifp, m, NULL);
}
if (result != 0) {
ifnet_stat_increment_in(pcb->utun_ifp, 0, 0, 1);
printf("%s - ifnet_input failed: %d\n", __FUNCTION__, result);
mbuf_freem(m);
}
return 0;
}