1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
/*
* Copyright (c) 2008-2025 Jonathan Schleifer <js@nil.im>
*
* All rights reserved.
*
* This program is free software: you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License version 3.0 only,
* 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 Lesser General Public License
* version 3.0 for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* version 3.0 along with this program. If not, see
* <https://www.gnu.org/licenses/>.
*/
#include "config.h"
#include <errno.h>
#include <string.h>
#include "unistd_wrapper.h"
#import "ObjFW.h"
#import "ObjFWTest.h"
@interface OFUNIXStreamSocketTests: OTTestCase
@end
@implementation OFUNIXStreamSocketTests
- (void)testUNIXStreamSocketWithPath: (OFString *)path
{
OFUNIXStreamSocket *sockClient, *sockServer, *sockAccepted;
char buffer[5];
#ifdef OF_WINDOWS
if ([OFSystemInfo wineVersion] != nil)
OTSkip(@"UNIX stream sockets are broken on Wine");
#endif
sockClient = [OFUNIXStreamSocket socket];
sockServer = [OFUNIXStreamSocket socket];
@try {
[sockServer bindToPath: path];
} @catch (OFBindSocketFailedException *e) {
switch (e.errNo) {
case EAFNOSUPPORT:
case EPERM:
OTSkip(@"UNIX stream sockets unsupported");
default:
@throw e;
}
}
@try {
#ifndef OF_WINDOWS
OFUNIXSocketCredentials peerCredentials;
OFNumber *number;
#endif
[sockServer listen];
[sockClient connectToPath: path];
sockAccepted = [sockServer accept];
[sockAccepted writeBuffer: "Hello" length: 5];
OTAssertEqual([sockClient readIntoBuffer: buffer length: 5], 5);
OTAssertEqual(memcmp(buffer, "Hello", 5), 0);
OTAssertEqual(OFSocketAddressUNIXPath(
sockAccepted.remoteAddress).length, 0);
#ifndef OF_WINDOWS
peerCredentials = sockAccepted.peerCredentials;
number = [peerCredentials objectForKey:
OFUNIXSocketCredentialsUserID];
if (number != nil)
OTAssertEqualObjects(number,
[OFNumber numberWithUnsignedLong: getuid()]);
number = [peerCredentials objectForKey:
OFUNIXSocketCredentialsGroupID];
if (number != nil)
OTAssertEqualObjects(number,
[OFNumber numberWithUnsignedLong: getgid()]);
number = [peerCredentials objectForKey:
OFUNIXSocketCredentialsProcessID];
if (number != nil)
OTAssertEqualObjects(number,
[OFNumber numberWithUnsignedLong: getpid()]);
#endif
} @finally {
#ifdef OF_HAVE_FILES
if (![path hasPrefix: @"@"])
[[OFFileManager defaultManager] removeItemAtPath: path];
#endif
}
}
- (void)testUNIXStreamSocket
{
#if defined(OF_HAVE_FILES) && !defined(OF_IOS)
OFString *path = [[OFSystemInfo temporaryDirectoryIRI]
IRIByAppendingPathComponent: [[OFUUID UUID] UUIDString]]
.fileSystemRepresentation;
#else
/*
* We can have sockets, including UNIX sockets, while file support is
* disabled.
*
* We also use this code path for iOS, as the temporaryDirectory:RI is
* too long on the iOS simulator.
*/
OFString *path = [OFString
stringWithFormat: @"/tmp/%@", [[OFUUID UUID] UUIDString]];
#endif
OTAssertNotNil(path);
[self testUNIXStreamSocketWithPath: path];
}
#ifdef OF_LINUX
- (void)testAbstractUNIXStreamSocket
{
[self testUNIXStreamSocketWithPath: [OFString stringWithFormat:
@"@/tmp/%@", [[OFUUID UUID] UUIDString]]];
}
#endif
@end
|
