/*
* Copyright (c) 2008-2021 Jonathan Schleifer <js@nil.im>
*
* All rights reserved.
*
* This file is part of ObjFW. It may be distributed under the terms of the
* Q Public License 1.0, which can be found in the file LICENSE.QPL included in
* the packaging of this file.
*
* Alternatively, it may be distributed under the terms of the GNU General
* Public License, either version 2 or 3, which can be found in the file
* LICENSE.GPLv2 or LICENSE.GPLv3 respectively included in the packaging of this
* file.
*/
#include "config.h"
#include <errno.h>
#import "TestsAppDelegate.h"
static OFString *const module = @"OFSPXSocket";
@interface SPXSocketDelegate: OFObject <OFSPXSocketDelegate>
{
@public
OFSequencedPacketSocket *_expectedServerSocket;
OFSPXSocket *_expectedClientSocket;
unsigned char _expectedNode[IPX_NODE_LEN];
uint32_t _expectedNetwork;
uint16_t _expectedPort;
bool _accepted;
bool _connected;
}
@end
@implementation SPXSocketDelegate
- (bool)socket: (OFSequencedPacketSocket *)sock
didAcceptSocket: (OFSequencedPacketSocket *)accepted
exception: (id)exception
{
OFEnsure(!_accepted);
_accepted = (sock == _expectedServerSocket && accepted != nil &&
exception == nil);
if (_accepted && _connected)
[[OFRunLoop mainRunLoop] stop];
return false;
}
- (void)socket: (OFSPXSocket *)sock
didConnectToNode: (unsigned char [IPX_NODE_LEN])node
network: (uint32_t)network
port: (uint16_t)port
exception: (id)exception
{
OFEnsure(!_connected);
_connected = (sock == _expectedClientSocket &&
memcmp(node, _expectedNode, IPX_NODE_LEN) == 0 &&
network == _expectedNetwork && port == _expectedPort &&
exception == nil);
if (_accepted && _connected)
[[OFRunLoop mainRunLoop] stop];
}
@end
@implementation TestsAppDelegate (OFSPXSocketTests)
- (void)SPXSocketTests
{
void *pool = objc_autoreleasePoolPush();
OFSPXSocket *sockClient, *sockServer, *sockAccepted;;
OFSocketAddress address1;
const OFSocketAddress *address2;
unsigned char node[IPX_NODE_LEN], node2[IPX_NODE_LEN];
uint32_t network;
uint16_t port;
char buffer[5];
SPXSocketDelegate *delegate;
TEST(@"+[socket]", (sockClient = [OFSPXSocket socket]) &&
(sockServer = [OFSPXSocket socket]))
@try {
TEST(@"-[bindToPort:]",
R(address1 = [sockServer bindToPort: 0]))
} @catch (OFBindFailedException *e) {
switch (e.errNo) {
case EAFNOSUPPORT:
[OFStdOut setForegroundColor: [OFColor lime]];
[OFStdOut writeLine:
@"\r[OFSPXSocket] -[bindToPort:]: "
@"IPX unsupported, skipping tests"];
break;
case ESOCKTNOSUPPORT:
[OFStdOut setForegroundColor: [OFColor lime]];
[OFStdOut writeLine:
@"\r[OFSPXSocket] -[bindToPort:]: "
@"SPX unsupported, skipping tests"];
break;
case EADDRNOTAVAIL:
[OFStdOut setForegroundColor: [OFColor lime]];
[OFStdOut writeLine:
@"\r[OFSPXSocket] -[bindToPort:]: "
@"IPX not configured, skipping tests"];
break;
default:
@throw e;
}
objc_autoreleasePoolPop(pool);
return;
}
OFSocketAddressIPXNode(&address1, node);
network = OFSocketAddressIPXNetwork(&address1);
port = OFSocketAddressPort(&address1);
TEST(@"-[listen]", R([sockServer listen]))
TEST(@"-[connectToNode:network:port:]",
R([sockClient connectToNode: node network: network port: port]))
TEST(@"-[accept]", (sockAccepted = [sockServer accept]))
TEST(@"-[sendBuffer:length:]",
R([sockAccepted sendBuffer: "Hello" length: 5]))
TEST(@"-[receiveIntoBuffer:length:]",
[sockClient receiveIntoBuffer: buffer length: 5] == 5 &&
memcmp(buffer, "Hello", 5) == 0)
TEST(@"-[remoteAddress]",
(address2 = sockAccepted.remoteAddress) &&
R(OFSocketAddressIPXNode(address2, node2)) &&
memcmp(node, node2, IPX_NODE_LEN) == 0 &&
OFSocketAddressIPXNetwork(address2) == network)
delegate = [[[SPXSocketDelegate alloc] init] autorelease];
sockServer = [OFSPXSocket socket];
delegate->_expectedServerSocket = sockServer;
sockServer.delegate = delegate;
sockClient = [OFSPXSocket socket];
delegate->_expectedClientSocket = sockClient;
sockClient.delegate = delegate;
address1 = [sockServer bindToPort: 0];
[sockServer listen];
[sockServer asyncAccept];
OFSocketAddressIPXNode(&address1, node);
memcpy(delegate->_expectedNode, node, IPX_NODE_LEN);
delegate->_expectedNetwork = network =
OFSocketAddressIPXNetwork(&address1);
delegate->_expectedPort = port = OFSocketAddressPort(&address1);
@try {
[sockClient asyncConnectToNode: node
network: network
port: port];
[[OFRunLoop mainRunLoop] runUntilDate:
[OFDate dateWithTimeIntervalSinceNow: 2]];
TEST(@"-[asyncAccept] & -[asyncConnectToNode:network:port:]",
delegate->_accepted && delegate->_connected)
} @catch (OFObserveFailedException *e) {
switch (e.errNo) {
case ENOTSOCK:
[OFStdOut setForegroundColor: [OFColor lime]];
[OFStdOut writeLine:
@"\r[OFSPXSocket] -[asyncAccept] & "
@"-[asyncConnectToNode:network:port:]: select() "
@"not supported for SPX, skipping test"];
break;
default:
@throw e;
}
}
objc_autoreleasePoolPop(pool);
}
@end