skarg
140 lines
4.3 KiB
C
140 lines
4.3 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 <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"
|
|
|
|
static int get_local_ifr_ioctl(char *ifname, struct ifreq *ifr, int request)
|
|
{
|
|
int fd;
|
|
int rv; // return value
|
|
|
|
strncpy(ifr->ifr_name, ifname, sizeof(ifr->ifr_name));
|
|
fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_IP);
|
|
if (fd < 0)
|
|
rv = fd;
|
|
else
|
|
rv = ioctl(fd, request, ifr);
|
|
|
|
return rv;
|
|
}
|
|
|
|
static int get_local_address_ioctl(
|
|
char *ifname,
|
|
struct in_addr *addr,
|
|
int request)
|
|
{
|
|
struct ifreq ifr = { {{0}} };
|
|
struct sockaddr_in *tcpip_address;
|
|
int rv; // return value
|
|
|
|
rv = get_local_ifr_ioctl(ifname,&ifr,request);
|
|
if (rv >= 0) {
|
|
tcpip_address = (struct sockaddr_in *) &ifr.ifr_addr;
|
|
memcpy(addr, &tcpip_address->sin_addr, sizeof(struct in_addr));
|
|
}
|
|
|
|
return rv;
|
|
}
|
|
|
|
void bip_set_interface(char *ifname)
|
|
{
|
|
struct in_addr local_address;
|
|
struct in_addr broadcast_address;
|
|
|
|
/* setup local address */
|
|
get_local_address_ioctl(ifname, &local_address, SIOCGIFADDR);
|
|
bip_set_addr(local_address.s_addr);
|
|
fprintf(stderr,"IP Address: %s\n",inet_ntoa(local_address));
|
|
/* setup local broadcast address */
|
|
get_local_address_ioctl(ifname, &broadcast_address, SIOCGIFBRDADDR);
|
|
bip_set_broadcast_addr(broadcast_address.s_addr);
|
|
fprintf(stderr,"Broadcast Address: %s\n",inet_ntoa(broadcast_address));
|
|
}
|
|
|
|
bool bip_init(void)
|
|
{
|
|
int status = 0; // return from socket lib calls
|
|
struct sockaddr_in sin;
|
|
int sockopt = 0;
|
|
int sock_fd = -1;
|
|
|
|
// 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)
|
|
return false;
|
|
// Allow us to use the same socket for sending and receiving
|
|
// This makes sure that the src port is correct when sending
|
|
sockopt = 1;
|
|
status = setsockopt(sock_fd, SOL_SOCKET, SO_REUSEADDR,
|
|
&sockopt, sizeof(sockopt));
|
|
if (status < 0)
|
|
{
|
|
close(sock_fd);
|
|
bip_set_socket(-1);
|
|
return status;
|
|
}
|
|
// allow us to send a broadcast
|
|
status = setsockopt(sock_fd, SOL_SOCKET, SO_BROADCAST,
|
|
&sockopt, sizeof(sockopt));
|
|
if (status < 0)
|
|
{
|
|
close(sock_fd);
|
|
bip_set_socket(-1);
|
|
return status;
|
|
}
|
|
|
|
// bind the socket to the local port number and IP address
|
|
sin.sin_family = AF_INET;
|
|
sin.sin_addr.s_addr = htonl(INADDR_ANY);
|
|
sin.sin_port = htons(bip_get_port());
|
|
memset(&(sin.sin_zero), '\0', 8);
|
|
status = bind(sock_fd,
|
|
(const struct sockaddr*)&sin, sizeof(struct sockaddr));
|
|
if (status < 0)
|
|
{
|
|
close(sock_fd);
|
|
bip_set_socket(-1);
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
|