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
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
|
// SDLTCPTransport.m
//
#import "SDLTCPTransport.h"
#import "SDLDebugTool.h"
#import "SDLHexUtility.h"
#import <errno.h>
#import <netdb.h>
#import <netinet/in.h>
#import <signal.h>
#import <stdio.h>
#import <sys/socket.h>
#import <sys/types.h>
#import <sys/wait.h>
#import <unistd.h>
// C function forward declarations.
int call_socket(const char *hostname, const char *port);
static void TCPCallback(CFSocketRef socket, CFSocketCallBackType type, CFDataRef address, const void *data, void *info);
@interface SDLTCPTransport () {
BOOL _alreadyDestructed;
dispatch_queue_t _sendQueue;
}
@end
@implementation SDLTCPTransport
- (instancetype)init {
if (self = [super init]) {
_alreadyDestructed = NO;
_sendQueue = dispatch_queue_create("com.sdl.transport.tcp.transmit", DISPATCH_QUEUE_SERIAL);
[SDLDebugTool logInfo:@"SDLTCPTransport Init"
withType:SDLDebugType_Transport_iAP
toOutput:SDLDebugOutput_All
toGroup:self.debugConsoleGroupName];
}
return self;
}
- (void)connect {
[SDLDebugTool logInfo:@"TCP Transport attempt connect" withType:SDLDebugType_Transport_TCP];
int sock_fd = call_socket([self.hostName UTF8String], [self.portNumber UTF8String]);
if (sock_fd < 0) {
[SDLDebugTool logInfo:@"Server Not Ready, Connection Failed" withType:SDLDebugType_Transport_TCP];
return;
}
CFSocketContext socketCtxt = {0, (__bridge void *)(self), NULL, NULL, NULL};
socket = CFSocketCreateWithNative(kCFAllocatorDefault, sock_fd, kCFSocketDataCallBack | kCFSocketConnectCallBack, (CFSocketCallBack)&TCPCallback, &socketCtxt);
CFRunLoopSourceRef source = CFSocketCreateRunLoopSource(kCFAllocatorDefault, socket, 0);
CFRunLoopRef loop = CFRunLoopGetCurrent();
CFRunLoopAddSource(loop, source, kCFRunLoopDefaultMode);
CFRelease(source);
}
- (void)sendData:(NSData *)msgBytes {
dispatch_async(_sendQueue, ^{
@autoreleasepool {
NSString *byteStr = [SDLHexUtility getHexString:msgBytes];
[SDLDebugTool logInfo:[NSString stringWithFormat:@"Sent %lu bytes: %@", (unsigned long)msgBytes.length, byteStr] withType:SDLDebugType_Transport_TCP toOutput:SDLDebugOutput_DeviceConsole];
CFSocketError e = CFSocketSendData(socket, NULL, (__bridge CFDataRef)msgBytes, 10000);
if (e != kCFSocketSuccess) {
NSString *errorCause = nil;
switch (e) {
case kCFSocketTimeout:
errorCause = @"Socket Timeout Error.";
break;
case kCFSocketError:
default:
errorCause = @"Socket Error.";
break;
}
[SDLDebugTool logInfo:[NSString stringWithFormat:@"Socket sendData error: %@", errorCause] withType:SDLDebugType_Transport_TCP toOutput:SDLDebugOutput_DeviceConsole];
}
}
});
}
- (void)destructObjects {
[SDLDebugTool logInfo:@"SDLTCPTransport invalidate and dispose"];
if (!_alreadyDestructed) {
_alreadyDestructed = YES;
if (socket != nil) {
CFSocketInvalidate(socket);
CFRelease(socket);
}
}
}
- (void)disconnect {
[self dispose];
}
- (void)dispose {
[self destructObjects];
}
- (void)dealloc {
[self destructObjects];
}
@end
// C functions
int call_socket(const char *hostname, const char *port) {
int status, sock;
struct addrinfo hints;
struct addrinfo *servinfo;
memset(&hints, 0, sizeof hints);
hints.ai_family = AF_UNSPEC;
hints.ai_socktype = SOCK_STREAM;
//no host name?, no problem, get local host
if (hostname == nil) {
char localhost[128];
gethostname(localhost, sizeof localhost);
hostname = (const char *)&localhost;
}
//getaddrinfo setup
if ((status = getaddrinfo(hostname, port, &hints, &servinfo)) != 0) {
fprintf(stderr, "getaddrinfo error: %s\n", gai_strerror(status));
return (-1);
}
//get socket
if ((sock = socket(servinfo->ai_family, servinfo->ai_socktype, servinfo->ai_protocol)) < 0)
return (-1);
//connect
if (connect(sock, servinfo->ai_addr, servinfo->ai_addrlen) < 0) {
close(sock);
return (-1);
}
freeaddrinfo(servinfo); // free the linked-list
return (sock);
}
static void TCPCallback(CFSocketRef socket, CFSocketCallBackType type, CFDataRef address, const void *data, void *info) {
if (kCFSocketConnectCallBack == type) {
SDLTCPTransport *transport = (__bridge SDLTCPTransport *)info;
[transport.delegate onTransportConnected];
} else if (kCFSocketDataCallBack == type) {
SDLTCPTransport *transport = (__bridge SDLTCPTransport *)info;
// Check if Core disconnected from us
if (CFDataGetLength((CFDataRef)data) <= 0) {
[SDLDebugTool logInfo:@"TCPCallback Got a data packet with length 0, the connection was terminated on the other side"];
[transport.delegate onTransportDisconnected];
return;
}
// Handle the data we received
NSMutableString *byteStr = [NSMutableString stringWithCapacity:((int)CFDataGetLength((CFDataRef)data) * 2)];
for (int i = 0; i < (int)CFDataGetLength((CFDataRef)data); i++) {
[byteStr appendFormat:@"%02X", ((Byte *)(UInt8 *)CFDataGetBytePtr((CFDataRef)data))[i]];
}
[SDLDebugTool logInfo:[NSString stringWithFormat:@"Read %d bytes: %@", (int)CFDataGetLength((CFDataRef)data), byteStr] withType:SDLDebugType_Transport_TCP toOutput:SDLDebugOutput_DeviceConsole];
[transport.delegate onDataReceived:[NSData dataWithBytes:(UInt8 *)CFDataGetBytePtr((CFDataRef)data) length:(int)CFDataGetLength((CFDataRef)data)]];
} else {
NSString *logMessage = [NSString stringWithFormat:@"unhandled TCPCallback: %lu", type];
[SDLDebugTool logInfo:logMessage withType:SDLDebugType_Transport_TCP toOutput:SDLDebugOutput_DeviceConsole];
}
}
|
