Mon 21 Jul 22:43:21 CEST 2025
This commit is contained in:
parent
6f55d53b70
commit
f864e27e47
|
|
@ -0,0 +1,735 @@
|
|||
// Copyright (c) 2014 The Chromium Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style license that can be
|
||||
// found in the LICENSE file.
|
||||
|
||||
#import "ChromeSocketsTcp.h"
|
||||
#import "GCDAsyncSocket.h"
|
||||
|
||||
#ifndef CHROME_SOCKETS_TCP_VERBOSE_LOGGING
|
||||
#define CHROME_SOCKETS_TCP_VERBOSE_LOGGING 0
|
||||
#endif
|
||||
|
||||
#if CHROME_SOCKETS_TCP_VERBOSE_LOGGING
|
||||
#define VERBOSE_LOG NSLog
|
||||
#else
|
||||
#define VERBOSE_LOG(args...) do {} while (false)
|
||||
#endif
|
||||
|
||||
#if CHROME_SOCKETS_TCP_VERBOSE_LOGGING
|
||||
static NSString* stringFromData(NSData* data) {
|
||||
NSUInteger len = [data length];
|
||||
if (len > 200) {
|
||||
len = 200;
|
||||
}
|
||||
char* buf = (char*)malloc(len + 1);
|
||||
memcpy(buf, [data bytes], len);
|
||||
buf[len] = 0;
|
||||
NSString* ret = [NSString stringWithUTF8String:buf];
|
||||
free(buf);
|
||||
return ret;
|
||||
}
|
||||
#endif // CHROME_SOCKETS_TCP_VERBOSE_LOGGING
|
||||
|
||||
#pragma mark ChromeSocketsTcpSocket interface
|
||||
|
||||
@interface ChromeSocketsTcpSocket : NSObject {
|
||||
@public
|
||||
__weak ChromeSocketsTcp* _plugin;
|
||||
|
||||
NSUInteger _socketId;
|
||||
NSNumber* _persistent;
|
||||
NSString* _name;
|
||||
NSNumber* _bufferSize;
|
||||
|
||||
// pipeToFile options
|
||||
NSNumber* _append;
|
||||
NSString* _uri;
|
||||
NSFileHandle* _uriFileHandle;
|
||||
NSInteger _numBytes;
|
||||
id _pipeToFileCompleteCallback;
|
||||
long _bytesReadNotSend;
|
||||
NSTimeInterval _lastProgressTimestamp;
|
||||
|
||||
NSUInteger _readTag;
|
||||
NSUInteger _receivedTag;
|
||||
|
||||
NSNumber* _paused;
|
||||
NSMutableArray* _pausedBuffers;
|
||||
|
||||
GCDAsyncSocket* _socket;
|
||||
|
||||
NSMutableArray* _sendCallbacks;
|
||||
|
||||
id _connectCallback;
|
||||
id _disconnectCallback;
|
||||
id _secureCallback;
|
||||
}
|
||||
|
||||
@end
|
||||
|
||||
#pragma mark ChromeSocketsTcp interface
|
||||
|
||||
@interface ChromeSocketsTcp() {
|
||||
NSMutableDictionary* _sockets;
|
||||
NSUInteger _nextSocketId;
|
||||
NSString* _receiveEventsCallbackId;
|
||||
|
||||
@public
|
||||
NSInteger _pendingReceive;
|
||||
}
|
||||
|
||||
- (void)create:(CDVInvokedUrlCommand*)command;
|
||||
- (void)update:(CDVInvokedUrlCommand*)command;
|
||||
- (void)setPaused:(CDVInvokedUrlCommand*)command;
|
||||
// - (void)setKeepAlive:(CDVInvokedUrlCommand*)command;
|
||||
// - (void)setNoDelay:(CDVInvokedUrlCommand*)command;
|
||||
- (void)connect:(CDVInvokedUrlCommand*)command;
|
||||
- (void)disconnect:(CDVInvokedUrlCommand*)command;
|
||||
- (void)secure:(CDVInvokedUrlCommand*)command;
|
||||
- (void)send:(CDVInvokedUrlCommand*)command;
|
||||
- (void)close:(CDVInvokedUrlCommand*)command;
|
||||
- (void)getInfo:(CDVInvokedUrlCommand*)command;
|
||||
- (void)getSockets:(CDVInvokedUrlCommand*)command;
|
||||
- (void)pipeToFile:(CDVInvokedUrlCommand*)command;
|
||||
- (void)registerReceiveEvents:(CDVInvokedUrlCommand*)command;
|
||||
- (void)readyToRead:(CDVInvokedUrlCommand*)command;
|
||||
- (void)fireReceiveEventsWithSocketId:(NSUInteger)theSocketId data:(NSData*)theData;
|
||||
- (void)fireReceiveEventsWithInfo:(NSDictionary*)theInfo waitReadyToRead:(BOOL)waitReadyToRead;
|
||||
- (void)fireReceiveErrorEventsWithSocketId:(NSUInteger)theSocketId errorCode:(NSUInteger)theErrorCode message:(NSString*)theMessage;
|
||||
@end
|
||||
|
||||
@implementation ChromeSocketsTcpSocket
|
||||
|
||||
NSTimeInterval const PIPE_TO_FILE_PROGRESS_INTERVAL = 0.1;
|
||||
|
||||
- (ChromeSocketsTcpSocket*)initWithId:(NSUInteger)theSocketId plugin:(ChromeSocketsTcp*)thePlugin properties:(NSDictionary*)theProperties
|
||||
{
|
||||
self = [super init];
|
||||
if (self) {
|
||||
_socketId = theSocketId;
|
||||
_plugin = thePlugin;
|
||||
_paused = [NSNumber numberWithBool:NO];
|
||||
_socket = [[GCDAsyncSocket alloc] initWithDelegate:self delegateQueue:dispatch_get_main_queue()];
|
||||
[self resetSocket];
|
||||
[self setProperties:theProperties];
|
||||
}
|
||||
return self;
|
||||
}
|
||||
|
||||
- (ChromeSocketsTcpSocket*)initAcceptedSocketWithId:(NSUInteger)theSocketId plugin:(ChromeSocketsTcp*)thePlugin socket:(GCDAsyncSocket*)theSocket
|
||||
{
|
||||
self = [super init];
|
||||
if (self) {
|
||||
_socketId = theSocketId;
|
||||
_plugin = thePlugin;
|
||||
_socket = theSocket;
|
||||
[_socket setDelegate:self];
|
||||
_paused = [NSNumber numberWithBool:YES];
|
||||
[self resetSocket];
|
||||
[self setProperties:nil];
|
||||
}
|
||||
return self;
|
||||
}
|
||||
|
||||
- (void)resetSocket
|
||||
{
|
||||
_readTag = 0;
|
||||
_receivedTag = 0;
|
||||
_pausedBuffers = [NSMutableArray array];
|
||||
_sendCallbacks = [NSMutableArray array];
|
||||
_connectCallback = nil;
|
||||
_disconnectCallback = nil;
|
||||
_secureCallback = nil;
|
||||
[self resetPipeToFileProperties];
|
||||
}
|
||||
|
||||
- (void)resetPipeToFileProperties
|
||||
{
|
||||
if (_uriFileHandle) {
|
||||
[_uriFileHandle closeFile];
|
||||
_uriFileHandle = nil;
|
||||
_uri = nil;
|
||||
}
|
||||
_append = nil;
|
||||
_pipeToFileCompleteCallback = nil;
|
||||
_numBytes = 0;
|
||||
_bytesReadNotSend = 0;
|
||||
}
|
||||
|
||||
- (NSDictionary*)getInfo
|
||||
{
|
||||
NSString* localAddress = [_socket localHost];
|
||||
NSNumber* localPort = [NSNumber numberWithUnsignedInt:[_socket localPort]];
|
||||
NSString* peerAddress = [_socket connectedHost];
|
||||
NSNumber* peerPort = [NSNumber numberWithUnsignedInt:[_socket connectedPort]];
|
||||
|
||||
NSMutableDictionary* socketInfo = [@{
|
||||
@"socketId": [NSNumber numberWithUnsignedInteger:_socketId],
|
||||
@"persistent": _persistent,
|
||||
@"name": _name,
|
||||
@"bufferSize": _bufferSize,
|
||||
@"connected": [NSNumber numberWithBool:[_socket isConnected]],
|
||||
@"paused": _paused,
|
||||
} mutableCopy];
|
||||
|
||||
if (localAddress) {
|
||||
socketInfo[@"localAddress"] = localAddress;
|
||||
socketInfo[@"localPort"] = localPort;
|
||||
}
|
||||
|
||||
if (peerAddress) {
|
||||
socketInfo[@"peerAddress"] = peerAddress;
|
||||
socketInfo[@"peerPort"] = peerPort;
|
||||
}
|
||||
|
||||
|
||||
return [socketInfo copy];
|
||||
}
|
||||
|
||||
- (void)setProperties:(NSDictionary*)theProperties
|
||||
{
|
||||
NSNumber* persistent = theProperties[@"persistent"];
|
||||
NSString* name = theProperties[@"name"];
|
||||
NSNumber* bufferSize = theProperties[@"bufferSize"];
|
||||
|
||||
if (persistent)
|
||||
_persistent = persistent;
|
||||
|
||||
if (name)
|
||||
_name = name;
|
||||
|
||||
if (bufferSize && _bufferSize == 0 && ![_paused boolValue]) // read delegate method won't be called when _bufferSize == 0
|
||||
[_socket readDataWithTimeout:-1 buffer:nil bufferOffset:0 maxLength:[bufferSize unsignedIntegerValue] tag:++_readTag];
|
||||
|
||||
if (bufferSize)
|
||||
_bufferSize = bufferSize;
|
||||
|
||||
// Set undefined properties to default value
|
||||
if (_persistent == nil)
|
||||
_persistent = [NSNumber numberWithBool:NO];
|
||||
|
||||
if (_name == nil)
|
||||
_name = @"";
|
||||
|
||||
if (_bufferSize == nil)
|
||||
_bufferSize = [NSNumber numberWithUnsignedInteger:4096];
|
||||
}
|
||||
|
||||
- (BOOL)setPipeToFileProperties:(NSDictionary*)theProperties
|
||||
{
|
||||
NSNumber* append = theProperties[@"append"];
|
||||
NSNumber* numBytes = theProperties[@"numBytes"];
|
||||
NSString* uri = theProperties[@"uri"];
|
||||
|
||||
[self resetPipeToFileProperties];
|
||||
|
||||
if (append) {
|
||||
_append = append;
|
||||
} else {
|
||||
_append = [NSNumber numberWithBool:NO];
|
||||
}
|
||||
|
||||
if (numBytes && [numBytes integerValue] > 0) {
|
||||
_numBytes = [numBytes integerValue];
|
||||
} else {
|
||||
return NO;
|
||||
}
|
||||
|
||||
if (uri && uri.length > 0) {
|
||||
NSString* filePath = [[NSURL URLWithString:uri] path];
|
||||
|
||||
if(![NSFileManager.defaultManager fileExistsAtPath:filePath]) {
|
||||
[[NSFileManager defaultManager] createFileAtPath:filePath contents:nil attributes:nil];
|
||||
}
|
||||
|
||||
_uriFileHandle = [NSFileHandle fileHandleForWritingAtPath:filePath];
|
||||
|
||||
if (append.boolValue) {
|
||||
[_uriFileHandle seekToEndOfFile];
|
||||
}
|
||||
|
||||
if (_uriFileHandle) {
|
||||
_uri = uri;
|
||||
} else {
|
||||
return NO;
|
||||
}
|
||||
} else {
|
||||
return NO;
|
||||
}
|
||||
|
||||
_lastProgressTimestamp = [[NSDate date] timeIntervalSince1970];
|
||||
return YES;
|
||||
}
|
||||
|
||||
- (void)resumeReadIfNotReading
|
||||
{
|
||||
if (_readTag == _receivedTag && _plugin->_pendingReceive == 0 && [_socket isConnected] && ![_paused boolValue]) {
|
||||
[_socket readDataWithTimeout:-1 buffer:nil bufferOffset:0 maxLength:[_bufferSize unsignedIntegerValue] tag:++_readTag];
|
||||
}
|
||||
}
|
||||
|
||||
- (void)setPaused:(NSNumber*)paused
|
||||
{
|
||||
if (![_paused isEqualToNumber:paused]) {
|
||||
_paused = paused;
|
||||
if (![_paused boolValue]) {
|
||||
for (NSData* data in _pausedBuffers) {
|
||||
[self sendReceivedData:data];
|
||||
}
|
||||
[_pausedBuffers removeAllObjects];
|
||||
[self resumeReadIfNotReading];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
- (void)sendReceivedData:(NSData*)data
|
||||
{
|
||||
if (_uriFileHandle) {
|
||||
|
||||
NSUInteger bytesRead = 0;
|
||||
|
||||
// extra data that need to send to the webveiw
|
||||
NSData* extraData = nil;
|
||||
|
||||
if (_numBytes >= [data length]) {
|
||||
[_uriFileHandle writeData:data];
|
||||
bytesRead = data.length;
|
||||
} else {
|
||||
NSData* writeData = [data subdataWithRange:NSMakeRange(0, _numBytes)];
|
||||
extraData = [data subdataWithRange:NSMakeRange(_numBytes, data.length - _numBytes)];
|
||||
[_uriFileHandle writeData:writeData];
|
||||
bytesRead = writeData.length;
|
||||
}
|
||||
|
||||
_numBytes -= bytesRead;
|
||||
|
||||
NSTimeInterval timestamp = [[NSDate date] timeIntervalSince1970];
|
||||
|
||||
_bytesReadNotSend += bytesRead;
|
||||
if (_numBytes == 0 || timestamp - _lastProgressTimestamp > PIPE_TO_FILE_PROGRESS_INTERVAL) {
|
||||
NSDictionary *info = @{
|
||||
@"socketId": [NSNumber numberWithUnsignedInteger:_socketId],
|
||||
@"uri": _uri,
|
||||
@"bytesRead": [NSNumber numberWithUnsignedInteger:_bytesReadNotSend],
|
||||
};
|
||||
|
||||
[_plugin fireReceiveEventsWithInfo:info waitReadyToRead:NO];
|
||||
_bytesReadNotSend = 0;
|
||||
_lastProgressTimestamp = timestamp;
|
||||
}
|
||||
|
||||
if (_numBytes == 0) {
|
||||
void (^callback)() = _pipeToFileCompleteCallback;
|
||||
[self resetPipeToFileProperties];
|
||||
assert(callback != nil);
|
||||
callback();
|
||||
}
|
||||
|
||||
if (extraData) {
|
||||
[_plugin fireReceiveEventsWithSocketId:_socketId data:extraData];
|
||||
} else {
|
||||
[self resumeReadIfNotReading];
|
||||
}
|
||||
|
||||
} else {
|
||||
[_plugin fireReceiveEventsWithSocketId:_socketId data:data];
|
||||
}
|
||||
}
|
||||
|
||||
- (void)socket:(GCDAsyncSocket *)sock didConnectToHost:(NSString *)host port:(UInt16)port
|
||||
{
|
||||
VERBOSE_LOG(@"socket:didConnectToHost socketId: %u", _socketId);
|
||||
|
||||
void (^callback)(BOOL, NSError*) = _connectCallback;
|
||||
assert(callback != nil);
|
||||
_connectCallback = nil;
|
||||
|
||||
callback(YES, nil);
|
||||
|
||||
if (![_paused boolValue])
|
||||
[_socket readDataWithTimeout:-1 buffer:nil bufferOffset:0 maxLength:[_bufferSize unsignedIntegerValue] tag:++_readTag];
|
||||
}
|
||||
|
||||
- (void)socket:(GCDAsyncSocket *)sock didReadData:(NSData *)data withTag:(long)tag
|
||||
{
|
||||
VERBOSE_LOG(@"socket:didReadDataWithTag socketId: %u", _socketId);
|
||||
|
||||
_receivedTag = tag;
|
||||
|
||||
if ([_paused boolValue]) {
|
||||
[_pausedBuffers addObject:data];
|
||||
} else {
|
||||
[self sendReceivedData:data];
|
||||
}
|
||||
}
|
||||
|
||||
- (void)socket:(GCDAsyncSocket *)sock didWriteDataWithTag:(long)tag
|
||||
{
|
||||
assert([_sendCallbacks count] != 0);
|
||||
void (^callback)() = _sendCallbacks[0];
|
||||
assert(callback != nil);
|
||||
[_sendCallbacks removeObjectAtIndex:0];
|
||||
|
||||
callback();
|
||||
|
||||
[_socket readDataWithTimeout:-1 buffer:nil bufferOffset:0 maxLength:[_bufferSize unsignedIntegerValue] tag:-1];
|
||||
}
|
||||
|
||||
- (void)socketDidDisconnect:(GCDAsyncSocket *)sock withError:(NSError *)err
|
||||
{
|
||||
if (_disconnectCallback) {
|
||||
void (^callback)() = _disconnectCallback;
|
||||
assert(_disconnectCallback != nil);
|
||||
_disconnectCallback = nil;
|
||||
callback();
|
||||
} else if (_connectCallback) {
|
||||
void(^callback)(BOOL, NSError*) = _connectCallback;
|
||||
callback(NO, err);
|
||||
_connectCallback = nil;
|
||||
} else if (err) {
|
||||
[_plugin fireReceiveErrorEventsWithSocketId:_socketId errorCode:[err code] message:[err localizedDescription]];
|
||||
}
|
||||
|
||||
[self resetSocket];
|
||||
}
|
||||
|
||||
- (void)socketDidSecure:(GCDAsyncSocket *)sock
|
||||
{
|
||||
VERBOSE_LOG(@"socketDidSecure socketId: %u", _socketId);
|
||||
|
||||
assert(_secureCallback != nil);
|
||||
void (^callback)() = _secureCallback;
|
||||
_secureCallback = nil;
|
||||
callback();
|
||||
}
|
||||
|
||||
@end
|
||||
|
||||
@implementation ChromeSocketsTcp
|
||||
|
||||
- (void)pluginInitialize
|
||||
{
|
||||
_sockets = [NSMutableDictionary dictionary];
|
||||
_nextSocketId = 1;
|
||||
_receiveEventsCallbackId = nil;
|
||||
_pendingReceive = 0;
|
||||
}
|
||||
|
||||
- (void)onReset
|
||||
{
|
||||
for (NSNumber* socketId in _sockets) {
|
||||
[self closeSocketWithId:socketId callbackId:nil];
|
||||
}
|
||||
}
|
||||
|
||||
- (NSDictionary*)buildErrorInfoWithErrorCode:(NSInteger)theErrorCode message:(NSString*)message
|
||||
{
|
||||
return @{
|
||||
@"resultCode": [NSNumber numberWithInteger:theErrorCode],
|
||||
@"message": message,
|
||||
};
|
||||
}
|
||||
|
||||
- (void)create:(CDVInvokedUrlCommand*)command
|
||||
{
|
||||
NSDictionary* properties = [command argumentAtIndex:0];
|
||||
|
||||
ChromeSocketsTcpSocket* socket = [[ChromeSocketsTcpSocket alloc] initWithId:_nextSocketId++ plugin:self properties:properties];
|
||||
|
||||
_sockets[[NSNumber numberWithUnsignedInteger:socket->_socketId]] = socket;
|
||||
|
||||
[self.commandDelegate sendPluginResult:[CDVPluginResult resultWithStatus:CDVCommandStatus_OK messageAsInt:(int)socket->_socketId] callbackId:command.callbackId];
|
||||
}
|
||||
|
||||
- (void)update:(CDVInvokedUrlCommand*)command
|
||||
{
|
||||
NSNumber* socketId = [command argumentAtIndex:0];
|
||||
NSDictionary* properties = [command argumentAtIndex:1];
|
||||
|
||||
ChromeSocketsTcpSocket *socket = _sockets[socketId];
|
||||
|
||||
if (socket == nil)
|
||||
return;
|
||||
|
||||
[socket setProperties:properties];
|
||||
[self.commandDelegate sendPluginResult:[CDVPluginResult resultWithStatus:CDVCommandStatus_OK] callbackId:command.callbackId];
|
||||
}
|
||||
|
||||
- (void)setPaused:(CDVInvokedUrlCommand*)command
|
||||
{
|
||||
NSNumber* socketId = [command argumentAtIndex:0];
|
||||
NSNumber* paused = [command argumentAtIndex:1];
|
||||
|
||||
ChromeSocketsTcpSocket* socket = _sockets[socketId];
|
||||
|
||||
if (socket == nil)
|
||||
return;
|
||||
|
||||
[socket setPaused:paused];
|
||||
[self.commandDelegate sendPluginResult:[CDVPluginResult resultWithStatus:CDVCommandStatus_OK] callbackId:command.callbackId];
|
||||
}
|
||||
|
||||
- (void)connect:(CDVInvokedUrlCommand*)command
|
||||
{
|
||||
NSNumber* socketId = [command argumentAtIndex:0];
|
||||
NSString* peerAddress = [command argumentAtIndex:1];
|
||||
|
||||
NSUInteger peerPort = [[command argumentAtIndex:2] unsignedIntegerValue];
|
||||
|
||||
ChromeSocketsTcpSocket* socket = _sockets[socketId];
|
||||
|
||||
if (socket == nil) {
|
||||
[self.commandDelegate sendPluginResult:[CDVPluginResult resultWithStatus:CDVCommandStatus_ERROR messageAsDictionary:[self buildErrorInfoWithErrorCode:ENOTSOCK message:@"Invalid Argument"]] callbackId:command.callbackId];
|
||||
return;
|
||||
}
|
||||
|
||||
id<CDVCommandDelegate> commandDelegate = self.commandDelegate;
|
||||
socket->_connectCallback = [^(BOOL success, NSError* error) {
|
||||
if (success) {
|
||||
[commandDelegate sendPluginResult:[CDVPluginResult resultWithStatus:CDVCommandStatus_OK] callbackId:command.callbackId];
|
||||
} else {
|
||||
[commandDelegate sendPluginResult:[CDVPluginResult resultWithStatus:CDVCommandStatus_ERROR messageAsDictionary:[self buildErrorInfoWithErrorCode:[error code] message:[error localizedDescription]]] callbackId:command.callbackId];
|
||||
}
|
||||
} copy];
|
||||
|
||||
NSError* err = nil;
|
||||
BOOL success = [socket->_socket connectToHost:peerAddress onPort:peerPort withTimeout:2 error:&err];
|
||||
|
||||
if (!success) {
|
||||
void(^callback)(BOOL, NSError*) = socket->_connectCallback;
|
||||
callback(NO, err);
|
||||
socket->_connectCallback = nil;
|
||||
}
|
||||
}
|
||||
|
||||
- (void)disconnectSocketWithId:(NSNumber*)socketId callbackId:(NSString*)theCallbackId close:(BOOL)close
|
||||
{
|
||||
ChromeSocketsTcpSocket* socket = _sockets[socketId];
|
||||
|
||||
if (socket == nil)
|
||||
return;
|
||||
|
||||
id<CDVCommandDelegate> commandDelegate = self.commandDelegate;
|
||||
socket->_disconnectCallback = [^(){
|
||||
if (theCallbackId)
|
||||
[commandDelegate sendPluginResult:[CDVPluginResult resultWithStatus:CDVCommandStatus_OK] callbackId:theCallbackId];
|
||||
if (close)
|
||||
[_sockets removeObjectForKey:socketId];
|
||||
} copy];
|
||||
|
||||
if ([socket->_socket isDisconnected]) {
|
||||
void (^callback)() = socket->_disconnectCallback;
|
||||
socket->_disconnectCallback = nil;
|
||||
callback();
|
||||
} else {
|
||||
[socket->_socket disconnect];
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
- (void)disconnect:(CDVInvokedUrlCommand*)command
|
||||
{
|
||||
NSNumber* socketId = [command argumentAtIndex:0];
|
||||
[self disconnectSocketWithId:socketId callbackId:command.callbackId close:NO];
|
||||
}
|
||||
|
||||
- (NSNumber*)getSecureVersion:(NSString*)versionString
|
||||
{
|
||||
if ([versionString isEqualToString:@"ssl3"]) {
|
||||
return [NSNumber numberWithInt:kSSLProtocol3];
|
||||
} else if ([versionString isEqualToString:@"tls1"]) {
|
||||
return [NSNumber numberWithInt:kTLSProtocol1];
|
||||
} else if ([versionString isEqualToString:@"tls1.1"]) {
|
||||
return [NSNumber numberWithInt:kTLSProtocol11];
|
||||
} else if ([versionString isEqualToString:@"tls1.2"]) {
|
||||
return [NSNumber numberWithInt:kTLSProtocol12];
|
||||
} else {
|
||||
return nil;
|
||||
}
|
||||
}
|
||||
|
||||
- (void)secure:(CDVInvokedUrlCommand *)command
|
||||
{
|
||||
NSNumber* socketId = [command argumentAtIndex:0];
|
||||
NSDictionary* options = [command argumentAtIndex:1];
|
||||
|
||||
ChromeSocketsTcpSocket* socket = _sockets[socketId];
|
||||
|
||||
if (socket == nil) {
|
||||
[self.commandDelegate sendPluginResult:[CDVPluginResult resultWithStatus:CDVCommandStatus_ERROR messageAsDictionary:[self buildErrorInfoWithErrorCode:ENOTSOCK message:@"Invalid Argument"]] callbackId:command.callbackId];
|
||||
return;
|
||||
}
|
||||
|
||||
id<CDVCommandDelegate> commandDelegate = self.commandDelegate;
|
||||
socket->_secureCallback = [^() {
|
||||
[commandDelegate sendPluginResult:[CDVPluginResult resultWithStatus:CDVCommandStatus_OK] callbackId:command.callbackId];
|
||||
} copy];
|
||||
|
||||
NSMutableDictionary* settings = [NSMutableDictionary dictionaryWithCapacity:2];
|
||||
|
||||
NSDictionary* tlsVersion = options[@"tlsVersion"];
|
||||
if (tlsVersion) {
|
||||
NSNumber* minVersion = [self getSecureVersion:tlsVersion[@"min"]];
|
||||
NSNumber* maxVersion = [self getSecureVersion:tlsVersion[@"max"]];
|
||||
|
||||
if (minVersion) {
|
||||
settings[GCDAsyncSocketSSLProtocolVersionMin] = minVersion;
|
||||
}
|
||||
|
||||
if (maxVersion) {
|
||||
settings[GCDAsyncSocketSSLProtocolVersionMax] = maxVersion;
|
||||
}
|
||||
}
|
||||
|
||||
[socket->_socket startTLS:settings];
|
||||
}
|
||||
|
||||
- (void)send:(CDVInvokedUrlCommand*)command
|
||||
{
|
||||
NSNumber* socketId = [command argumentAtIndex:0];
|
||||
NSData* data = [command argumentAtIndex:1];
|
||||
|
||||
ChromeSocketsTcpSocket* socket = _sockets[socketId];
|
||||
|
||||
if (socket == nil) {
|
||||
[self.commandDelegate sendPluginResult:[CDVPluginResult resultWithStatus:CDVCommandStatus_ERROR messageAsDictionary:[self buildErrorInfoWithErrorCode:ENOTSOCK message:@"Invalid Argument"]] callbackId:command.callbackId];
|
||||
return;
|
||||
}
|
||||
|
||||
if ([socket->_socket isConnected]) {
|
||||
id<CDVCommandDelegate> commandDelegate = self.commandDelegate;
|
||||
[socket->_sendCallbacks addObject:[^() {
|
||||
VERBOSE_LOG(@"ACK %@.%@ Write", socketId, command.callbackId);
|
||||
[commandDelegate sendPluginResult:[CDVPluginResult resultWithStatus:CDVCommandStatus_OK messageAsInt:(int)[data length]] callbackId:command.callbackId];
|
||||
} copy]];
|
||||
|
||||
[socket->_socket writeData:data withTimeout:-1 tag:-1];
|
||||
|
||||
} else {
|
||||
[self.commandDelegate sendPluginResult:[CDVPluginResult resultWithStatus:CDVCommandStatus_ERROR messageAsDictionary:[self buildErrorInfoWithErrorCode:ENOTCONN message:@"Socket Not Connected"]] callbackId:command.callbackId];
|
||||
}
|
||||
}
|
||||
|
||||
- (void)closeSocketWithId:(NSNumber*)socketId callbackId:(NSString*)theCallbackId
|
||||
{
|
||||
[self disconnectSocketWithId:socketId callbackId:theCallbackId close:YES];
|
||||
}
|
||||
|
||||
- (void)close:(CDVInvokedUrlCommand*)command
|
||||
{
|
||||
NSNumber* socketId = [command argumentAtIndex:0];
|
||||
[self closeSocketWithId:socketId callbackId:command.callbackId];
|
||||
}
|
||||
|
||||
- (void)getInfo:(CDVInvokedUrlCommand*)command
|
||||
{
|
||||
NSNumber* socketId = [command argumentAtIndex:0];
|
||||
|
||||
ChromeSocketsTcpSocket* socket = _sockets[socketId];
|
||||
|
||||
if (socket == nil)
|
||||
return;
|
||||
|
||||
[self.commandDelegate sendPluginResult:[CDVPluginResult resultWithStatus:CDVCommandStatus_OK messageAsDictionary:[socket getInfo]] callbackId:command.callbackId];
|
||||
}
|
||||
|
||||
- (void)getSockets:(CDVInvokedUrlCommand*)command
|
||||
{
|
||||
NSArray* sockets = [_sockets allValues];
|
||||
NSMutableArray* socketsInfo = [NSMutableArray arrayWithCapacity:[sockets count]];
|
||||
|
||||
for (ChromeSocketsTcpSocket* socket in sockets) {
|
||||
[socketsInfo addObject:[socket getInfo]];
|
||||
}
|
||||
|
||||
[self.commandDelegate sendPluginResult:[CDVPluginResult resultWithStatus:CDVCommandStatus_OK messageAsArray:socketsInfo] callbackId:command.callbackId];
|
||||
}
|
||||
|
||||
- (void)pipeToFile:(CDVInvokedUrlCommand *)command
|
||||
{
|
||||
NSNumber* socketId = [command argumentAtIndex:0];
|
||||
NSDictionary* options = [command argumentAtIndex:1];
|
||||
|
||||
ChromeSocketsTcpSocket* socket = _sockets[socketId];
|
||||
|
||||
if (socket == nil)
|
||||
return;
|
||||
|
||||
if ([socket setPipeToFileProperties:options]) {
|
||||
id<CDVCommandDelegate> commandDelegate = self.commandDelegate;
|
||||
socket->_pipeToFileCompleteCallback = [^() {
|
||||
[commandDelegate sendPluginResult:[CDVPluginResult resultWithStatus:CDVCommandStatus_OK] callbackId:command.callbackId];
|
||||
} copy];
|
||||
} else {
|
||||
[self fireReceiveErrorEventsWithSocketId:socket->_socketId errorCode:-1000 message:@"Failed to start pipeToFile"];
|
||||
}
|
||||
}
|
||||
|
||||
- (void)registerReceiveEvents:(CDVInvokedUrlCommand*)command
|
||||
{
|
||||
VERBOSE_LOG(@"registerReceiveEvents: ");
|
||||
_receiveEventsCallbackId = command.callbackId;
|
||||
}
|
||||
|
||||
- (void)fireReceiveEventsWithSocketId:(NSUInteger)theSocketId data:(NSData*)theData
|
||||
{
|
||||
NSArray* multipart = @[
|
||||
@{@"socketId":[NSNumber numberWithUnsignedInteger:theSocketId]},
|
||||
theData,
|
||||
];
|
||||
[self fireReceiveEventsWithMultiPart:multipart waitReadyToRead:YES];
|
||||
}
|
||||
|
||||
- (void)fireReceiveEventsWithInfo:(NSDictionary*)theInfo waitReadyToRead:(BOOL)waitReadyToRead
|
||||
{
|
||||
[self fireReceiveEventsWithMultiPart:@[theInfo] waitReadyToRead:waitReadyToRead];
|
||||
}
|
||||
|
||||
- (void)fireReceiveEventsWithMultiPart:(NSArray*)theMultiPart waitReadyToRead:(BOOL)waitReadyToRead
|
||||
{
|
||||
assert(_receiveEventsCallbackId != nil);
|
||||
CDVPluginResult* result = [CDVPluginResult resultWithStatus:CDVCommandStatus_OK messageAsMultipart:theMultiPart];
|
||||
[result setKeepCallbackAsBool:YES];
|
||||
if (waitReadyToRead) {
|
||||
_pendingReceive++;
|
||||
}
|
||||
[self.commandDelegate sendPluginResult:result callbackId:_receiveEventsCallbackId];
|
||||
}
|
||||
|
||||
- (void)fireReceiveErrorEventsWithSocketId:(NSUInteger)theSocketId errorCode:(NSUInteger)theErrorCode message:(NSString*)theMessage
|
||||
{
|
||||
assert(_receiveEventsCallbackId != nil);
|
||||
|
||||
NSDictionary* info = @{
|
||||
@"socketId": [NSNumber numberWithUnsignedInteger:theSocketId],
|
||||
@"resultCode": [NSNumber numberWithUnsignedInteger:theErrorCode],
|
||||
@"message": theMessage,
|
||||
};
|
||||
|
||||
CDVPluginResult *result = [CDVPluginResult resultWithStatus:CDVCommandStatus_ERROR messageAsDictionary:info];
|
||||
[result setKeepCallbackAsBool:YES];
|
||||
|
||||
[self.commandDelegate sendPluginResult:result callbackId:_receiveEventsCallbackId];
|
||||
}
|
||||
|
||||
- (NSUInteger)registerAcceptedSocket:(GCDAsyncSocket*)theSocket
|
||||
{
|
||||
ChromeSocketsTcpSocket* socket = [[ChromeSocketsTcpSocket alloc] initAcceptedSocketWithId:_nextSocketId++ plugin:self socket:theSocket];
|
||||
_sockets[[NSNumber numberWithUnsignedInteger:socket->_socketId]] = socket;
|
||||
|
||||
[socket setPaused:[NSNumber numberWithBool:YES]];
|
||||
|
||||
return socket->_socketId;
|
||||
}
|
||||
|
||||
- (void)readyToRead:(CDVInvokedUrlCommand*)command
|
||||
{
|
||||
_pendingReceive--;
|
||||
if (_pendingReceive == 0) {
|
||||
for (ChromeSocketsTcpSocket* socket in [_sockets allValues]) {
|
||||
[socket resumeReadIfNotReading];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@end
|
||||
Loading…
Reference in New Issue
Block a user