skarg
355 lines
13 KiB
C
355 lines
13 KiB
C
/*####COPYRIGHTBEGIN####
|
|
-------------------------------------------
|
|
Copyright (C) 2005 Steve Karg
|
|
|
|
This program is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU General Public License
|
|
as published by the Free Software Foundation; either version 2
|
|
of the License, or (at your option) any later version.
|
|
|
|
This program 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 General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to:
|
|
The Free Software Foundation, Inc.
|
|
59 Temple Place - Suite 330
|
|
Boston, MA 02111-1307, USA.
|
|
|
|
As a special exception, if other files instantiate templates or
|
|
use macros or inline functions from this file, or you compile
|
|
this file and link it with other works to produce a work based
|
|
on this file, this file does not by itself cause the resulting
|
|
work to be covered by the GNU General Public License. However
|
|
the source code for this file must still be made available in
|
|
accordance with section (3) of the GNU General Public License.
|
|
|
|
This exception does not invalidate any other reasons why a work
|
|
based on this file might be covered by the GNU General Public
|
|
License.
|
|
-------------------------------------------
|
|
####COPYRIGHTEND####*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <stdint.h> /* for standard integer types uint8_t etc. */
|
|
#include <stdbool.h> /* for the standard bool type. */
|
|
#include "bacdcode.h"
|
|
#include "bip.h"
|
|
#include "net.h"
|
|
|
|
/* To fill a need, we invent the gethostaddr() function. */
|
|
static long gethostaddr(void)
|
|
{
|
|
struct hostent *host_ent;
|
|
char host_name[255];
|
|
|
|
if (gethostname(host_name, sizeof(host_name)) != 0)
|
|
return -1;
|
|
|
|
if ((host_ent = gethostbyname(host_name)) == NULL)
|
|
return -1;
|
|
#ifdef BIP_DEBUG
|
|
printf("host: %s at %u.%u.%u.%u\n", host_name,
|
|
((uint8_t*)host_ent->h_addr)[0],
|
|
((uint8_t*)host_ent->h_addr)[1],
|
|
((uint8_t*)host_ent->h_addr)[2],
|
|
((uint8_t*)host_ent->h_addr)[3]);
|
|
#endif
|
|
/* note: network byte order */
|
|
return *(long *) host_ent->h_addr;
|
|
}
|
|
|
|
static void set_broadcast_address(uint32_t net_address)
|
|
{
|
|
#if USE_INADDR
|
|
/* Note: sometimes INADDR_BROADCAST does not let me get
|
|
any unicast messages. Not sure why... */
|
|
(void) net_address;
|
|
bip_set_broadcast_addr(INADDR_BROADCAST);
|
|
#else
|
|
long broadcast_address = 0;
|
|
|
|
if (IN_CLASSA(ntohl(net_address)))
|
|
broadcast_address =
|
|
(ntohl(net_address) & ~IN_CLASSA_HOST) | IN_CLASSA_HOST;
|
|
else if (IN_CLASSB(ntohl(net_address)))
|
|
broadcast_address =
|
|
(ntohl(net_address) & ~IN_CLASSB_HOST) | IN_CLASSB_HOST;
|
|
else if (IN_CLASSC(ntohl(net_address)))
|
|
broadcast_address =
|
|
(ntohl(net_address) & ~IN_CLASSC_HOST) | IN_CLASSC_HOST;
|
|
else if (IN_CLASSD(ntohl(net_address)))
|
|
broadcast_address =
|
|
(ntohl(net_address) & ~IN_CLASSD_HOST) | IN_CLASSD_HOST;
|
|
else
|
|
broadcast_address = INADDR_BROADCAST;
|
|
bip_set_broadcast_addr(htonl(broadcast_address));
|
|
#endif
|
|
}
|
|
|
|
static void cleanup(void)
|
|
{
|
|
WSACleanup();
|
|
}
|
|
|
|
|
|
/* on Windows, ifname is the dotted ip address of the interface */
|
|
void bip_set_interface(char *ifname)
|
|
{
|
|
struct in_addr address;
|
|
|
|
/* setup local address */
|
|
if (bip_get_addr() == 0) {
|
|
bip_set_addr(inet_addr(ifname));
|
|
}
|
|
#ifdef BIP_DEBUG
|
|
address.s_addr = htonl(bip_get_addr());
|
|
fprintf(stderr, "Interface: %s\n", ifname);
|
|
#endif
|
|
/* setup local broadcast address */
|
|
if (bip_get_broadcast_addr() == 0) {
|
|
address.s_addr = htonl(bip_get_addr());
|
|
set_broadcast_address(address.s_addr);
|
|
}
|
|
}
|
|
|
|
static char *winsock_error_code_text(int code)
|
|
{
|
|
switch (code) {
|
|
case WSAEACCES:
|
|
return "Permission denied.";
|
|
case WSAEINTR:
|
|
return "Interrupted system call.";
|
|
case WSAEBADF:
|
|
return "Bad file number.";
|
|
case WSAEFAULT:
|
|
return "Bad address.";
|
|
case WSAEINVAL:
|
|
return "Invalid argument.";
|
|
case WSAEMFILE:
|
|
return "Too many open files.";
|
|
case WSAEWOULDBLOCK:
|
|
return "Operation would block.";
|
|
case WSAEINPROGRESS:
|
|
return "Operation now in progress. This error is returned if any Windows Sockets API function is called while a blocking function is in progress.";
|
|
case WSAENOTSOCK:
|
|
return "Socket operation on nonsocket.";
|
|
case WSAEDESTADDRREQ:
|
|
return "Destination address required.";
|
|
case WSAEMSGSIZE:
|
|
return "Message too long.";
|
|
case WSAEPROTOTYPE:
|
|
return "Protocol wrong type for socket.";
|
|
case WSAENOPROTOOPT:
|
|
return "Protocol not available.";
|
|
case WSAEPROTONOSUPPORT:
|
|
return "Protocol not supported.";
|
|
case WSAESOCKTNOSUPPORT:
|
|
return "Socket type not supported.";
|
|
case WSAEOPNOTSUPP:
|
|
return "Operation not supported on socket.";
|
|
case WSAEPFNOSUPPORT:
|
|
return "Protocol family not supported.";
|
|
case WSAEAFNOSUPPORT:
|
|
return "Address family not supported by protocol family.";
|
|
case WSAEADDRINUSE:
|
|
return "Address already in use.";
|
|
case WSAEADDRNOTAVAIL:
|
|
return "Cannot assign requested address.";
|
|
case WSAENETDOWN:
|
|
return "Network is down. This error may be reported at any time if the Windows Sockets implementation detects an underlying failure.";
|
|
case WSAENETUNREACH:
|
|
return "Network is unreachable.";
|
|
case WSAENETRESET:
|
|
return "Network dropped connection on reset.";
|
|
case WSAECONNABORTED:
|
|
return "Software caused connection abort.";
|
|
case WSAECONNRESET:
|
|
return "Connection reset by peer.";
|
|
case WSAENOBUFS:
|
|
return "No buffer space available.";
|
|
case WSAEISCONN:
|
|
return "Socket is already connected.";
|
|
case WSAENOTCONN:
|
|
return "Socket is not connected.";
|
|
case WSAESHUTDOWN:
|
|
return "Cannot send after socket shutdown.";
|
|
case WSAETOOMANYREFS:
|
|
return "Too many references: cannot splice.";
|
|
case WSAETIMEDOUT:
|
|
return "Connection timed out.";
|
|
case WSAECONNREFUSED:
|
|
return "Connection refused.";
|
|
case WSAELOOP:
|
|
return "Too many levels of symbolic links.";
|
|
case WSAENAMETOOLONG:
|
|
return "File name too long.";
|
|
case WSAEHOSTDOWN:
|
|
return "Host is down.";
|
|
case WSAEHOSTUNREACH:
|
|
return "No route to host.";
|
|
case WSASYSNOTREADY:
|
|
return "Returned by WSAStartup(), "
|
|
"indicating that the network subsystem is unusable.";
|
|
case WSAVERNOTSUPPORTED:
|
|
return "Returned by WSAStartup(), "
|
|
"indicating that the Windows Sockets DLL cannot support "
|
|
"this application.";
|
|
case WSANOTINITIALISED:
|
|
return "Winsock not initialized. "
|
|
"This message is returned by any function except WSAStartup(), "
|
|
"indicating that a successful WSAStartup() has not yet "
|
|
"been performed.";
|
|
case WSAEDISCON:
|
|
return "Disconnect.";
|
|
case WSAHOST_NOT_FOUND:
|
|
return "Host not found. "
|
|
"This message indicates that the key "
|
|
"(name, address, and so on) was not found.";
|
|
case WSATRY_AGAIN:
|
|
return "Nonauthoritative host not found. "
|
|
"This error may suggest that the name service itself "
|
|
"is not functioning.";
|
|
case WSANO_RECOVERY:
|
|
return "Nonrecoverable error. "
|
|
"This error may suggest that the name service itself "
|
|
"is not functioning.";
|
|
case WSANO_DATA:
|
|
return "Valid name, no data record of requested type. "
|
|
"This error indicates that the key "
|
|
"(name, address, and so on) was not found.";
|
|
default: return "unknown";
|
|
}
|
|
}
|
|
|
|
bool bip_init(char *ifname)
|
|
{
|
|
int rv = 0; /* return from socket lib calls */
|
|
struct sockaddr_in sin = { -1 };
|
|
int value = 1;
|
|
int sock_fd = -1;
|
|
int Result;
|
|
int Code;
|
|
WSADATA wd;
|
|
struct in_addr address;
|
|
struct in_addr broadcast_address;
|
|
|
|
Result = WSAStartup((1 << 8) | 1, &wd);
|
|
/*Result = WSAStartup(MAKEWORD(2,2), &wd); */
|
|
if (Result != 0) {
|
|
Code = WSAGetLastError();
|
|
printf("TCP/IP stack initialization failed\n"
|
|
" error code: %i %s\n",
|
|
Code, winsock_error_code_text(Code));
|
|
exit(1);
|
|
}
|
|
atexit(cleanup);
|
|
|
|
if (ifname)
|
|
bip_set_interface(ifname);
|
|
/* has address been set? */
|
|
address.s_addr = htonl(bip_get_addr());
|
|
if (address.s_addr == 0) {
|
|
address.s_addr = gethostaddr();
|
|
if (address.s_addr == (unsigned) -1) {
|
|
Code = WSAGetLastError();
|
|
printf("Get host address failed\n"
|
|
" error code: %i %s\n",
|
|
Code, winsock_error_code_text(Code));
|
|
exit(1);
|
|
}
|
|
bip_set_addr(address.s_addr);
|
|
}
|
|
#ifdef BIP_DEBUG
|
|
fprintf(stderr, "IP Address: %s\n", inet_ntoa(address));
|
|
#endif
|
|
/* has broadcast address been set? */
|
|
if (bip_get_broadcast_addr() == 0) {
|
|
set_broadcast_address(address.s_addr);
|
|
}
|
|
#ifdef BIP_DEBUG
|
|
broadcast_address.s_addr = htonl(bip_get_broadcast_addr());
|
|
fprintf(stderr, "IP Broadcast Address: %s\n",
|
|
inet_ntoa(broadcast_address));
|
|
fprintf(stderr, "UDP Port: 0x%04X [%hu]\n",
|
|
bip_get_port(),
|
|
bip_get_port());
|
|
#endif
|
|
/* assumes that the driver has already been initialized */
|
|
sock_fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
|
bip_set_socket(sock_fd);
|
|
if (sock_fd < 0) {
|
|
fprintf(stderr, "bip: failed to allocate a socket.\n");
|
|
return false;
|
|
}
|
|
/* Allow us to use the same socket for sending and receiving */
|
|
/* This makes sure that the src port is correct when sending */
|
|
rv = setsockopt(sock_fd, SOL_SOCKET, SO_REUSEADDR,
|
|
(char *) &value, sizeof(value));
|
|
if (rv < 0) {
|
|
fprintf(stderr, "bip: failed to set REUSEADDR socket option.\n");
|
|
close(sock_fd);
|
|
bip_set_socket(-1);
|
|
return false;
|
|
}
|
|
/* allow us to send a broadcast */
|
|
rv = setsockopt(sock_fd, SOL_SOCKET, SO_BROADCAST,
|
|
(char *) &value, sizeof(value));
|
|
if (rv < 0) {
|
|
fprintf(stderr, "bip: failed to set BROADCAST socket option.\n");
|
|
close(sock_fd);
|
|
bip_set_socket(-1);
|
|
return false;
|
|
}
|
|
#if 0
|
|
/* probably only for Apple... */
|
|
/* rebind a port that is already in use.
|
|
Note: all users of the port must specify this flag */
|
|
rv = setsockopt(sock_fd, SOL_SOCKET, SO_REUSEPORT,
|
|
(char *) &value, sizeof(value));
|
|
if (rv < 0) {
|
|
fprintf(stderr, "bip: failed to set REUSEPORT socket option.\n");
|
|
close(sock_fd);
|
|
bip_set_socket(-1);
|
|
return false;
|
|
}
|
|
#endif
|
|
/* bind the socket to the local port number and IP address */
|
|
sin.sin_family = AF_INET;
|
|
#if USE_INADDR
|
|
/* by setting sin.sin_addr.s_addr to INADDR_ANY,
|
|
I am telling the IP stack to automatically fill
|
|
in the IP address of the machine the process
|
|
is running on.
|
|
|
|
Some server computers have multiple IP addresses.
|
|
A socket bound to one of these will not accept
|
|
connections to another address. Frequently you prefer
|
|
to allow any one of the computer's IP addresses
|
|
to be used for connections. Use INADDR_ANY (0L) to
|
|
allow clients to connect using any one of the host's
|
|
IP addresses. */
|
|
sin.sin_addr.s_addr = htonl(INADDR_ANY);
|
|
#else
|
|
/* or we could use the specific adapter address
|
|
note: already in network byte order */
|
|
sin.sin_addr.s_addr = address.s_addr;
|
|
#endif
|
|
sin.sin_port = htons(bip_get_port());
|
|
memset(&(sin.sin_zero), '\0', sizeof(sin.sin_zero));
|
|
rv = bind(sock_fd,
|
|
(const struct sockaddr *) &sin, sizeof(struct sockaddr));
|
|
if (rv < 0) {
|
|
fprintf(stderr, "bip: failed to bind to %s port %hd\n",
|
|
inet_ntoa(sin.sin_addr), bip_get_port());
|
|
close(sock_fd);
|
|
bip_set_socket(-1);
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|