summaryrefslogtreecommitdiffstats
path: root/libdnet-stripped/src/ip-win32.c
blob: 8f2ae843755abbab85797167030867b9de440925 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
/*
 * ip-win32.c
 *
 * Copyright (c) 2002 Dug Song <dugsong@monkey.org>
 *
 * $Id: ip-win32.c 547 2005-01-25 21:30:40Z dugsong $
 */

#ifdef _WIN32
#include "dnet_winconfig.h"
#else
#include "config.h"
#endif

#include <ws2tcpip.h>

#include <errno.h>
#include <stdlib.h>

#include "dnet.h"

struct ip_handle {
	WSADATA			wsdata;
	SOCKET			fd;
	struct sockaddr_in	sin;
};

ip_t *
ip_open(void)
{
	BOOL on;
	ip_t *ip;

	if ((ip = calloc(1, sizeof(*ip))) != NULL) {
		if (WSAStartup(MAKEWORD(2, 2), &ip->wsdata) != 0) {
			free(ip);
			return (NULL);
		}
		if ((ip->fd = socket(AF_INET, SOCK_RAW, IPPROTO_RAW)) ==
		    INVALID_SOCKET)
			return (ip_close(ip));
		
		on = TRUE;
		if (setsockopt(ip->fd, IPPROTO_IP, IP_HDRINCL,
			(const char *)&on, sizeof(on)) == SOCKET_ERROR) {
			SetLastError(ERROR_NETWORK_ACCESS_DENIED);
			return (ip_close(ip));
		}
		ip->sin.sin_family = AF_INET;
		ip->sin.sin_port = htons(666);
	}
	return (ip);
}

ssize_t
ip_send(ip_t *ip, const void *buf, size_t len)
{
	struct ip_hdr *hdr = (struct ip_hdr *)buf;
	
	ip->sin.sin_addr.s_addr = hdr->ip_src;
	
	if ((len = sendto(ip->fd, (const char *)buf, (int)len, 0,
	    (struct sockaddr *)&ip->sin, sizeof(ip->sin))) != SOCKET_ERROR)
		return (ssize_t)(len);
	
	return (-1);
}

ip_t *
ip_close(ip_t *ip)
{
	if (ip != NULL) {
		WSACleanup();
		if (ip->fd != INVALID_SOCKET)
			closesocket(ip->fd);
		free(ip);
	}
	return (NULL);
}