diff options
author | Paul Beusterien <paulbeusterien@google.com> | 2017-05-15 12:27:07 -0700 |
---|---|---|
committer | Paul Beusterien <paulbeusterien@google.com> | 2017-05-15 12:27:07 -0700 |
commit | 98ba64449a632518bd2b86fe8d927f4a960d3ddc (patch) | |
tree | 131d9c4272fa6179fcda6c5a33fcb3b1bd57ad2e /Example/Messaging/Tests/FIRMessagingFakeSocket.m | |
parent | 32461366c9e204a527ca05e6e9b9404a2454ac51 (diff) |
Initial
Diffstat (limited to 'Example/Messaging/Tests/FIRMessagingFakeSocket.m')
-rw-r--r-- | Example/Messaging/Tests/FIRMessagingFakeSocket.m | 89 |
1 files changed, 89 insertions, 0 deletions
diff --git a/Example/Messaging/Tests/FIRMessagingFakeSocket.m b/Example/Messaging/Tests/FIRMessagingFakeSocket.m new file mode 100644 index 0000000..2b0b477 --- /dev/null +++ b/Example/Messaging/Tests/FIRMessagingFakeSocket.m @@ -0,0 +1,89 @@ +/* + * Copyright 2017 Google + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#import "FIRMessagingFakeSocket.h" + +#import "FIRMessagingConstants.h" +#import "FIRMessagingDefines.h" +#import <GoogleToolboxForMac/GTMDefines.h> + +@interface FIRMessagingSecureSocket() <NSStreamDelegate> + +@property(nonatomic, readwrite, assign) FIRMessagingSecureSocketState state; +@property(nonatomic, readwrite, strong) NSInputStream *inStream; +@property(nonatomic, readwrite, strong) NSOutputStream *outStream; + +@property(nonatomic, readwrite, assign) BOOL isInStreamOpen; +@property(nonatomic, readwrite, assign) BOOL isOutStreamOpen; + +@property(nonatomic, readwrite, strong) NSRunLoop *runLoop; + +@end + +@interface FIRMessagingFakeSocket () + +@property(nonatomic, readwrite, assign) int8_t bufferSize; + +@end + +@implementation FIRMessagingFakeSocket + +- (instancetype)initWithBufferSize:(uint8_t)bufferSize { + self = [super init]; + if (self) { + _bufferSize = bufferSize; + } + return self; +} + +- (void)connectToHost:(NSString *)host + port:(NSUInteger)port + onRunLoop:(NSRunLoop *)runLoop { + self.state = kFIRMessagingSecureSocketOpening; + self.runLoop = runLoop; + + CFReadStreamRef inputStreamRef = nil; + CFWriteStreamRef outputStreamRef = nil; + + CFStreamCreateBoundPair(NULL, + &inputStreamRef, + &outputStreamRef, + self.bufferSize); + + self.inStream = CFBridgingRelease(inputStreamRef); + self.outStream = CFBridgingRelease(outputStreamRef); + if (!self.inStream || !self.outStream) { + FIRMessaging_FAIL(@"cannot create a fake socket"); + return; + } + + self.isInStreamOpen = NO; + self.isOutStreamOpen = NO; + + [self openStream:self.outStream]; + [self openStream:self.inStream]; +} + +- (void)openStream:(NSStream *)stream { + _GTMDevAssert(stream, @"Cannot open nil stream"); + if (stream) { + stream.delegate = self; + [stream scheduleInRunLoop:self.runLoop forMode:NSDefaultRunLoopMode]; + [stream open]; + } +} + +@end |