mirror of
https://github.com/gnustep/libs-base.git
synced 2025-04-24 01:11:01 +00:00
31098241ae
git-svn-id: svn+ssh://svn.gna.org/svn/gnustep/libs/base/branches/stable@32970 72102866-910b-0410-8b05-ffd578937521
404 lines
12 KiB
Objective-C
404 lines
12 KiB
Objective-C
#if defined(GNUSTEP_BASE_LIBRARY)
|
|
/**
|
|
* This test tests client and server socket
|
|
*/
|
|
#import "ObjectTesting.h"
|
|
#import <Foundation/Foundation.h>
|
|
#import <Foundation/NSStream.h>
|
|
|
|
static GSServerStream *serverStream;
|
|
static NSOutputStream *serverOutput = nil;
|
|
static NSOutputStream *clientOutput = nil;
|
|
static NSInputStream *serverInput = nil;
|
|
static NSInputStream *clientInput = nil;
|
|
static NSData *goldData;
|
|
static NSMutableData *testData;
|
|
|
|
@interface ClientListener : NSObject
|
|
{
|
|
uint8_t buffer[4096];
|
|
int writePointer;
|
|
}
|
|
@end
|
|
|
|
@implementation ClientListener
|
|
|
|
- (void)stream: (NSStream *)theStream handleEvent: (NSStreamEvent)streamEvent
|
|
{
|
|
NSLog(@"Client %p %d", theStream, streamEvent);
|
|
switch (streamEvent)
|
|
{
|
|
case NSStreamEventOpenCompleted:
|
|
{
|
|
if (theStream==clientOutput)
|
|
writePointer = 0;
|
|
break;
|
|
}
|
|
case NSStreamEventHasSpaceAvailable:
|
|
{
|
|
NSAssert(theStream==clientOutput, @"Wrong stream for writing");
|
|
if (writePointer<[goldData length])
|
|
{
|
|
int writeReturn = [clientOutput write: [goldData bytes]+writePointer
|
|
maxLength: [goldData length]-writePointer];
|
|
NSLog(@"Client %p wrote %d", clientOutput, writeReturn);
|
|
if (writeReturn < 0)
|
|
NSLog(@"Error ... %@", [clientOutput streamError]);
|
|
writePointer += writeReturn;
|
|
}
|
|
else
|
|
{
|
|
writePointer = 0;
|
|
[clientOutput close];
|
|
[clientOutput removeFromRunLoop: [NSRunLoop currentRunLoop]
|
|
forMode: NSDefaultRunLoopMode];
|
|
NSLog(@"Client close %p", clientOutput);
|
|
}
|
|
break;
|
|
}
|
|
case NSStreamEventHasBytesAvailable:
|
|
{
|
|
int readSize;
|
|
NSAssert(theStream==clientInput, @"Wrong stream for reading");
|
|
readSize = [clientInput read: buffer maxLength: 4096];
|
|
NSLog(@"Client %p read %d", clientInput, readSize);
|
|
if (readSize < 0)
|
|
{
|
|
NSLog(@"Error ... %@", [clientInput streamError]);
|
|
// it is possible that readSize<0 but not an Error.
|
|
// For example would block
|
|
}
|
|
else if (readSize == 0)
|
|
{
|
|
[clientInput close];
|
|
[clientInput removeFromRunLoop: [NSRunLoop currentRunLoop]
|
|
forMode: NSDefaultRunLoopMode];
|
|
NSLog(@"Client close %p", clientInput);
|
|
}
|
|
else
|
|
{
|
|
[testData appendBytes: buffer length: readSize];
|
|
}
|
|
break;
|
|
}
|
|
case NSStreamEventEndEncountered:
|
|
{
|
|
[theStream close];
|
|
[theStream removeFromRunLoop: [NSRunLoop currentRunLoop]
|
|
forMode: NSDefaultRunLoopMode];
|
|
NSLog(@"Client close %p", theStream);
|
|
break;
|
|
}
|
|
case NSStreamEventErrorOccurred:
|
|
{
|
|
NSLog(@"Error code is %d ... %@",
|
|
[[theStream streamError] code], [theStream streamError]);
|
|
break;
|
|
}
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
@end
|
|
|
|
@interface ServerListener : NSObject
|
|
{
|
|
uint8_t buffer[4096];
|
|
int readSize;
|
|
int writeSize;
|
|
BOOL readable;
|
|
BOOL writable;
|
|
}
|
|
@end
|
|
|
|
@implementation ServerListener
|
|
|
|
- (void)stream: (NSStream *)theStream handleEvent: (NSStreamEvent)streamEvent
|
|
{
|
|
NSLog(@"Server %p %d", theStream, streamEvent);
|
|
switch (streamEvent)
|
|
{
|
|
case NSStreamEventHasBytesAvailable:
|
|
{
|
|
if (theStream==serverStream)
|
|
{
|
|
NSAssert(serverInput==nil, @"accept twice");
|
|
[serverStream acceptWithInputStream: &serverInput
|
|
outputStream: &serverOutput];
|
|
if (serverInput) // it is ok to accept nothing
|
|
{
|
|
NSRunLoop *rl = [NSRunLoop currentRunLoop];
|
|
[serverInput scheduleInRunLoop: rl
|
|
forMode: NSDefaultRunLoopMode];
|
|
[serverOutput scheduleInRunLoop: rl
|
|
forMode: NSDefaultRunLoopMode];
|
|
NSLog(@"Server input stream is %p", serverInput);
|
|
NSLog(@"Server output stream is %p", serverOutput);
|
|
[serverInput retain];
|
|
[serverOutput retain];
|
|
[serverInput setDelegate: self];
|
|
[serverOutput setDelegate: self];
|
|
[serverInput open];
|
|
[serverOutput open];
|
|
readSize = 0;
|
|
writeSize = 0;
|
|
[serverStream close];
|
|
[serverStream removeFromRunLoop: [NSRunLoop currentRunLoop]
|
|
forMode: NSDefaultRunLoopMode];
|
|
}
|
|
}
|
|
if (theStream == serverInput)
|
|
{
|
|
readable = YES;
|
|
}
|
|
break;
|
|
}
|
|
case NSStreamEventHasSpaceAvailable:
|
|
{
|
|
NSAssert(theStream==serverOutput, @"Wrong stream for writing");
|
|
writable = YES;
|
|
break;
|
|
}
|
|
case NSStreamEventEndEncountered:
|
|
{
|
|
[theStream close];
|
|
[theStream removeFromRunLoop: [NSRunLoop currentRunLoop]
|
|
forMode: NSDefaultRunLoopMode];
|
|
NSLog(@"Server close %p", theStream);
|
|
if (theStream == serverInput && writeSize == readSize)
|
|
{
|
|
[serverOutput close];
|
|
[serverOutput removeFromRunLoop: [NSRunLoop currentRunLoop]
|
|
forMode: NSDefaultRunLoopMode];
|
|
NSLog(@"Server output close %p", serverOutput);
|
|
}
|
|
break;
|
|
}
|
|
case NSStreamEventErrorOccurred:
|
|
{
|
|
NSLog(@"Error code is %d ... %@",
|
|
[[theStream streamError] code], [theStream streamError]);
|
|
break;
|
|
}
|
|
default:
|
|
break;
|
|
}
|
|
|
|
while ((readable == YES && writeSize == readSize)
|
|
|| (writable == YES && writeSize < readSize))
|
|
{
|
|
if (readable == YES && writeSize == readSize)
|
|
{
|
|
readSize = [serverInput read: buffer maxLength: 4096];
|
|
readable = NO;
|
|
NSLog(@"Server %p read %d", serverInput, readSize);
|
|
writeSize = 0;
|
|
if (readSize == 0)
|
|
{
|
|
[serverInput close];
|
|
[serverInput removeFromRunLoop: [NSRunLoop currentRunLoop]
|
|
forMode: NSDefaultRunLoopMode];
|
|
NSLog(@"Server input close %p", serverInput);
|
|
[serverOutput close];
|
|
[serverOutput removeFromRunLoop: [NSRunLoop currentRunLoop]
|
|
forMode: NSDefaultRunLoopMode];
|
|
NSLog(@"Server output close %p", serverOutput);
|
|
}
|
|
else if (readSize < 0)
|
|
{
|
|
NSLog(@"Error ... %@", [clientInput streamError]);
|
|
readSize = 0;
|
|
}
|
|
}
|
|
if (writable == YES && writeSize < readSize)
|
|
{
|
|
int writeReturn = [serverOutput write: buffer+writeSize
|
|
maxLength: readSize-writeSize];
|
|
NSLog(@"Server %p wrote %d", serverOutput, writeReturn);
|
|
writable = NO;
|
|
if (writeReturn == 0)
|
|
{
|
|
[serverOutput close];
|
|
[serverOutput removeFromRunLoop: [NSRunLoop currentRunLoop]
|
|
forMode: NSDefaultRunLoopMode];
|
|
NSLog(@"Server close %p", serverOutput);
|
|
[serverInput close];
|
|
[serverInput removeFromRunLoop: [NSRunLoop currentRunLoop]
|
|
forMode: NSDefaultRunLoopMode];
|
|
NSLog(@"Server input close %p", serverInput);
|
|
}
|
|
else if (writeReturn > 0)
|
|
{
|
|
writeSize += writeReturn;
|
|
}
|
|
else if (writeReturn < 0)
|
|
{
|
|
NSLog(@"Error ... %@", [serverOutput streamError]);
|
|
}
|
|
|
|
/* If we have finished writing and there is no more data coming,
|
|
* we can close the output stream.
|
|
*/
|
|
if (writeSize == readSize
|
|
&& [serverInput streamStatus] == NSStreamStatusClosed)
|
|
{
|
|
[serverOutput close];
|
|
[serverOutput removeFromRunLoop: [NSRunLoop currentRunLoop]
|
|
forMode: NSDefaultRunLoopMode];
|
|
NSLog(@"Server output close %p", serverOutput);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
@end
|
|
|
|
int main()
|
|
{
|
|
NSAutoreleasePool *arp = [NSAutoreleasePool new];
|
|
NSRunLoop *rl = [NSRunLoop currentRunLoop];
|
|
NSHost *host = [NSHost hostWithAddress: @"127.0.0.1"];
|
|
ServerListener *sli;
|
|
ClientListener *cli;
|
|
NSString *path = @"socket_cs.m";
|
|
NSString *socketPath = @"test-socket";
|
|
|
|
[[NSFileManager defaultManager] removeFileAtPath: socketPath handler: nil];
|
|
NSLog(@"sending and receiving on %@: %@", host, [host address]);
|
|
goldData = [NSData dataWithContentsOfFile: path];
|
|
testData = [NSMutableData dataWithCapacity: 4096];
|
|
|
|
sli = [ServerListener new];
|
|
cli = [ClientListener new];
|
|
serverStream
|
|
= [GSServerStream serverStreamToAddr: [host address] port: 54321];
|
|
[serverStream setDelegate: sli];
|
|
[serverStream scheduleInRunLoop: rl forMode: NSDefaultRunLoopMode];
|
|
[serverStream open];
|
|
[NSStream getStreamsToHost: host
|
|
port: 54321
|
|
inputStream: &clientInput
|
|
outputStream: &clientOutput];
|
|
NSLog(@"Client input stream is %p", clientInput);
|
|
NSLog(@"Client output stream is %p", clientOutput);
|
|
[clientInput setDelegate: cli];
|
|
[clientOutput setDelegate: cli];
|
|
[clientInput scheduleInRunLoop: rl forMode: NSDefaultRunLoopMode];
|
|
[clientOutput scheduleInRunLoop: rl forMode: NSDefaultRunLoopMode];
|
|
[clientInput open];
|
|
[clientOutput open];
|
|
|
|
[rl runUntilDate: [NSDate dateWithTimeIntervalSinceNow: 30]];
|
|
PASS([goldData isEqualToData: testData], "Local tcp");
|
|
|
|
DESTROY(serverInput);
|
|
DESTROY(serverOutput);
|
|
clientInput = nil;
|
|
clientOutput = nil;
|
|
DESTROY(sli);
|
|
DESTROY(cli);
|
|
[testData setLength: 0];
|
|
|
|
sli = [ServerListener new];
|
|
cli = [ClientListener new];
|
|
serverStream
|
|
= [GSServerStream serverStreamToAddr: [host address] port: 54321];
|
|
[serverStream setDelegate: sli];
|
|
[serverStream open];
|
|
[serverStream scheduleInRunLoop: rl forMode: NSDefaultRunLoopMode];
|
|
[NSStream getStreamsToHost: host
|
|
port: 54321
|
|
inputStream: &clientInput
|
|
outputStream: &clientOutput];
|
|
NSLog(@"Client input stream is %p", clientInput);
|
|
NSLog(@"Client output stream is %p", clientOutput);
|
|
[clientInput setDelegate: cli];
|
|
[clientOutput setDelegate: cli];
|
|
[clientInput open];
|
|
[clientOutput open];
|
|
[clientInput scheduleInRunLoop: rl forMode: NSDefaultRunLoopMode];
|
|
[clientOutput scheduleInRunLoop: rl forMode: NSDefaultRunLoopMode];
|
|
|
|
[rl runUntilDate: [NSDate dateWithTimeIntervalSinceNow: 30]];
|
|
PASS([goldData isEqualToData: testData], "Local tcp (blocking open)");
|
|
|
|
DESTROY(serverInput);
|
|
DESTROY(serverOutput);
|
|
clientInput = nil;
|
|
clientOutput = nil;
|
|
DESTROY(sli);
|
|
DESTROY(cli);
|
|
[testData setLength: 0];
|
|
|
|
sli = [ServerListener new];
|
|
cli = [ClientListener new];
|
|
serverStream = [GSServerStream serverStreamToAddr: socketPath];
|
|
[serverStream setDelegate: sli];
|
|
[serverStream scheduleInRunLoop: rl forMode: NSDefaultRunLoopMode];
|
|
[serverStream open];
|
|
[NSStream getLocalStreamsToPath: socketPath
|
|
inputStream: &clientInput
|
|
outputStream: &clientOutput];
|
|
NSLog(@"Client input stream is %p", clientInput);
|
|
NSLog(@"Client output stream is %p", clientOutput);
|
|
[clientInput setDelegate: cli];
|
|
[clientOutput setDelegate: cli];
|
|
[clientInput scheduleInRunLoop: rl forMode: NSDefaultRunLoopMode];
|
|
[clientOutput scheduleInRunLoop: rl forMode: NSDefaultRunLoopMode];
|
|
[clientInput open];
|
|
[clientOutput open];
|
|
|
|
[rl runUntilDate: [NSDate dateWithTimeIntervalSinceNow: 30]];
|
|
|
|
PASS([goldData isEqualToData: testData], "Local socket");
|
|
|
|
DESTROY(serverInput);
|
|
DESTROY(serverOutput);
|
|
clientInput = nil;
|
|
clientOutput = nil;
|
|
DESTROY(sli);
|
|
DESTROY(cli);
|
|
[testData setLength: 0];
|
|
[[NSFileManager defaultManager] removeFileAtPath: socketPath handler: nil];
|
|
|
|
sli = [ServerListener new];
|
|
cli = [ClientListener new];
|
|
serverStream = [GSServerStream serverStreamToAddr: socketPath];
|
|
[serverStream setDelegate: sli];
|
|
[serverStream open];
|
|
[serverStream scheduleInRunLoop: rl forMode: NSDefaultRunLoopMode];
|
|
[NSStream getLocalStreamsToPath: socketPath
|
|
inputStream: &clientInput
|
|
outputStream: &clientOutput];
|
|
NSLog(@"Client input stream is %p", clientInput);
|
|
NSLog(@"Client output stream is %p", clientOutput);
|
|
[clientInput setDelegate: cli];
|
|
[clientOutput setDelegate: cli];
|
|
[clientInput open];
|
|
[clientOutput open];
|
|
[clientInput scheduleInRunLoop: rl forMode: NSDefaultRunLoopMode];
|
|
[clientOutput scheduleInRunLoop: rl forMode: NSDefaultRunLoopMode];
|
|
|
|
[rl runUntilDate: [NSDate dateWithTimeIntervalSinceNow: 30]];
|
|
|
|
PASS([goldData isEqualToData: testData], "Local socket (blocking open)");
|
|
|
|
DESTROY(serverInput);
|
|
DESTROY(serverOutput);
|
|
clientInput = nil;
|
|
clientOutput = nil;
|
|
DESTROY(sli);
|
|
DESTROY(cli);
|
|
[testData setLength: 0];
|
|
[[NSFileManager defaultManager] removeFileAtPath: socketPath handler: nil];
|
|
|
|
[arp release];
|
|
return 0;
|
|
}
|
|
#else
|
|
int main()
|
|
{
|
|
return 0;
|
|
}
|
|
#endif
|