mirror of
https://github.com/solemnwarning/ipxwrapper
synced 2024-12-30 16:45:37 +01:00
157 lines
3.8 KiB
C
157 lines
3.8 KiB
C
/* IPXWrapper - Unit tests
|
|
* Copyright (C) 2014 Daniel Collins <solemnwarning@solemnwarning.net>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU General Public License version 2 as published by
|
|
* the Free Software Foundation.
|
|
*
|
|
* 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., 51
|
|
* Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#include <windows.h>
|
|
#include <winsock2.h>
|
|
#include <assert.h>
|
|
#include <wsipx.h>
|
|
#include <wsnwlink.h>
|
|
|
|
#include "test.h"
|
|
|
|
static struct sockaddr_ipx get_iface_addr(int ifnum)
|
|
{
|
|
int sock = socket(AF_IPX, SOCK_DGRAM, NSPROTO_IPX);
|
|
assert(sock != -1);
|
|
|
|
IPX_ADDRESS_DATA iface_info;
|
|
iface_info.adapternum = ifnum;
|
|
|
|
int optlen = sizeof(iface_info);
|
|
assert(getsockopt(sock, NSPROTO_IPX, IPX_ADDRESS, (char*)(&iface_info), &optlen) == 0);
|
|
|
|
closesocket(sock);
|
|
|
|
struct sockaddr_ipx addr;
|
|
|
|
addr.sa_family = AF_IPX;
|
|
memcpy(addr.sa_netnum, iface_info.netnum, 4);
|
|
memcpy(addr.sa_nodenum, iface_info.nodenum, 6);
|
|
|
|
return addr;
|
|
}
|
|
|
|
int main()
|
|
{
|
|
{
|
|
WSADATA wsaData;
|
|
assert(WSAStartup(MAKEWORD(1,1), &wsaData) == 0);
|
|
}
|
|
|
|
struct sockaddr_ipx iface1_addr = get_iface_addr(0);
|
|
struct sockaddr_ipx iface2_addr = get_iface_addr(1);
|
|
|
|
/* Bind lsock1 to socket 2000 on the first interface */
|
|
|
|
int lsock1 = socket(AF_IPX, SOCK_STREAM, NSPROTO_SPX);
|
|
assert(lsock1 != -1);
|
|
|
|
{
|
|
unsigned long nonblock = 1;
|
|
assert(ioctlsocket(lsock1, FIONBIO, &nonblock) == 0);
|
|
}
|
|
|
|
struct sockaddr_ipx ls1_addr = iface1_addr;
|
|
ls1_addr.sa_socket = htons(2000);
|
|
|
|
assert(bind(lsock1, (struct sockaddr*)(&ls1_addr), sizeof(ls1_addr)) == 0);
|
|
|
|
assert(listen(lsock1, 8) == 0);
|
|
|
|
/* Bind lsock2 to a random socket on the second interface */
|
|
|
|
int lsock2 = socket(AF_IPX, SOCK_STREAM, NSPROTO_SPX);
|
|
assert(lsock2 != -1);
|
|
|
|
{
|
|
unsigned long nonblock = 1;
|
|
assert(ioctlsocket(lsock2, FIONBIO, &nonblock) == 0);
|
|
}
|
|
|
|
struct sockaddr_ipx ls2_addr = iface2_addr;
|
|
ls2_addr.sa_socket = 0;
|
|
|
|
assert(bind(lsock2, (struct sockaddr*)(&ls2_addr), sizeof(ls2_addr)) == 0);
|
|
assert(listen(lsock2, 8) == 0);
|
|
|
|
{
|
|
int addrlen = sizeof(ls2_addr);
|
|
assert(getsockname(lsock2, (struct sockaddr*)(&ls2_addr), &addrlen) == 0);
|
|
}
|
|
|
|
/* Unbound client should implicitly be bound to a random address. */
|
|
|
|
{
|
|
EXPECT_NO_ACCEPT(lsock1);
|
|
|
|
int sock = socket(AF_IPX, SOCK_STREAM, NSPROTO_SPX);
|
|
assert(sock != -1);
|
|
|
|
struct sockaddr_ipx my_addr = iface1_addr;
|
|
my_addr.sa_socket = htons(84);
|
|
|
|
assert(bind(sock, (struct sockaddr*)(&my_addr), sizeof(my_addr)) == 0);
|
|
EXPECT_LOCAL_ADDR(sock, my_addr);
|
|
|
|
MUST_CONNECT_TO(sock, ls1_addr);
|
|
|
|
/* Work around race condition; connect returns before listening
|
|
* socket has necessarily been notified.
|
|
*/
|
|
|
|
Sleep(100);
|
|
|
|
closesocket(EXPECT_ACCEPT(lsock1));
|
|
|
|
closesocket(sock);
|
|
}
|
|
|
|
/* Explicitly bound client should retain address. */
|
|
|
|
{
|
|
EXPECT_NO_ACCEPT(lsock1);
|
|
|
|
int sock = socket(AF_IPX, SOCK_STREAM, NSPROTO_SPX);
|
|
assert(sock != -1);
|
|
|
|
struct sockaddr_ipx my_addr = iface1_addr;
|
|
my_addr.sa_socket = htons(84);
|
|
|
|
assert(bind(sock, (struct sockaddr*)(&my_addr), sizeof(my_addr)) == 0);
|
|
EXPECT_LOCAL_ADDR(sock, my_addr);
|
|
|
|
MUST_CONNECT_TO(sock, ls1_addr);
|
|
|
|
/* Work around race condition; connect returns before listening
|
|
* socket has necessarily been notified.
|
|
*/
|
|
|
|
Sleep(100);
|
|
|
|
closesocket(EXPECT_ACCEPT_FROM(lsock1, my_addr));
|
|
|
|
closesocket(sock);
|
|
}
|
|
|
|
closesocket(lsock2);
|
|
closesocket(lsock1);
|
|
|
|
WSACleanup();
|
|
|
|
return 0;
|
|
}
|