Index: webrtc/examples/objc/AppRTCDemo/ARDWebSocketChannel.m |
diff --git a/webrtc/examples/objc/AppRTCDemo/ARDWebSocketChannel.m b/webrtc/examples/objc/AppRTCDemo/ARDWebSocketChannel.m |
new file mode 100644 |
index 0000000000000000000000000000000000000000..6f60380bff1988ccd1359328bb107e8e00a08111 |
--- /dev/null |
+++ b/webrtc/examples/objc/AppRTCDemo/ARDWebSocketChannel.m |
@@ -0,0 +1,251 @@ |
+/* |
+ * Copyright 2014 The WebRTC Project Authors. All rights reserved. |
+ * |
+ * Use of this source code is governed by a BSD-style license |
+ * that can be found in the LICENSE file in the root of the source |
+ * tree. An additional intellectual property rights grant can be found |
+ * in the file PATENTS. All contributing project authors may |
+ * be found in the AUTHORS file in the root of the source tree. |
+ */ |
+ |
+#import "ARDWebSocketChannel.h" |
+ |
+#import "WebRTC/RTCLogging.h" |
+#import "SRWebSocket.h" |
+ |
+#import "ARDSignalingMessage.h" |
+#import "ARDUtilities.h" |
+ |
+// TODO(tkchin): move these to a configuration object. |
+static NSString const *kARDWSSMessageErrorKey = @"error"; |
+static NSString const *kARDWSSMessagePayloadKey = @"msg"; |
+ |
+@interface ARDWebSocketChannel () <SRWebSocketDelegate> |
+@end |
+ |
+@implementation ARDWebSocketChannel { |
+ NSURL *_url; |
+ NSURL *_restURL; |
+ SRWebSocket *_socket; |
+} |
+ |
+@synthesize delegate = _delegate; |
+@synthesize state = _state; |
+@synthesize roomId = _roomId; |
+@synthesize clientId = _clientId; |
+ |
+- (instancetype)initWithURL:(NSURL *)url |
+ restURL:(NSURL *)restURL |
+ delegate:(id<ARDSignalingChannelDelegate>)delegate { |
+ if (self = [super init]) { |
+ _url = url; |
+ _restURL = restURL; |
+ _delegate = delegate; |
+ _socket = [[SRWebSocket alloc] initWithURL:url]; |
+ _socket.delegate = self; |
+ RTCLog(@"Opening WebSocket."); |
+ [_socket open]; |
+ } |
+ return self; |
+} |
+ |
+- (void)dealloc { |
+ [self disconnect]; |
+} |
+ |
+- (void)setState:(ARDSignalingChannelState)state { |
+ if (_state == state) { |
+ return; |
+ } |
+ _state = state; |
+ [_delegate channel:self didChangeState:_state]; |
+} |
+ |
+- (void)registerForRoomId:(NSString *)roomId |
+ clientId:(NSString *)clientId { |
+ NSParameterAssert(roomId.length); |
+ NSParameterAssert(clientId.length); |
+ _roomId = roomId; |
+ _clientId = clientId; |
+ if (_state == kARDSignalingChannelStateOpen) { |
+ [self registerWithCollider]; |
+ } |
+} |
+ |
+- (void)sendMessage:(ARDSignalingMessage *)message { |
+ NSParameterAssert(_clientId.length); |
+ NSParameterAssert(_roomId.length); |
+ NSData *data = [message JSONData]; |
+ if (_state == kARDSignalingChannelStateRegistered) { |
+ NSString *payload = |
+ [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding]; |
+ NSDictionary *message = @{ |
+ @"cmd": @"send", |
+ @"msg": payload, |
+ }; |
+ NSData *messageJSONObject = |
+ [NSJSONSerialization dataWithJSONObject:message |
+ options:NSJSONWritingPrettyPrinted |
+ error:nil]; |
+ NSString *messageString = |
+ [[NSString alloc] initWithData:messageJSONObject |
+ encoding:NSUTF8StringEncoding]; |
+ RTCLog(@"C->WSS: %@", messageString); |
+ [_socket send:messageString]; |
+ } else { |
+ NSString *dataString = |
+ [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding]; |
+ RTCLog(@"C->WSS POST: %@", dataString); |
+ NSString *urlString = |
+ [NSString stringWithFormat:@"%@/%@/%@", |
+ [_restURL absoluteString], _roomId, _clientId]; |
+ NSURL *url = [NSURL URLWithString:urlString]; |
+ [NSURLConnection sendAsyncPostToURL:url |
+ withData:data |
+ completionHandler:nil]; |
+ } |
+} |
+ |
+- (void)disconnect { |
+ if (_state == kARDSignalingChannelStateClosed || |
+ _state == kARDSignalingChannelStateError) { |
+ return; |
+ } |
+ [_socket close]; |
+ RTCLog(@"C->WSS DELETE rid:%@ cid:%@", _roomId, _clientId); |
+ NSString *urlString = |
+ [NSString stringWithFormat:@"%@/%@/%@", |
+ [_restURL absoluteString], _roomId, _clientId]; |
+ NSURL *url = [NSURL URLWithString:urlString]; |
+ NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:url]; |
+ request.HTTPMethod = @"DELETE"; |
+ request.HTTPBody = nil; |
+ [NSURLConnection sendAsyncRequest:request completionHandler:nil]; |
+} |
+ |
+#pragma mark - SRWebSocketDelegate |
+ |
+- (void)webSocketDidOpen:(SRWebSocket *)webSocket { |
+ RTCLog(@"WebSocket connection opened."); |
+ self.state = kARDSignalingChannelStateOpen; |
+ if (_roomId.length && _clientId.length) { |
+ [self registerWithCollider]; |
+ } |
+} |
+ |
+- (void)webSocket:(SRWebSocket *)webSocket didReceiveMessage:(id)message { |
+ NSString *messageString = message; |
+ NSData *messageData = [messageString dataUsingEncoding:NSUTF8StringEncoding]; |
+ id jsonObject = [NSJSONSerialization JSONObjectWithData:messageData |
+ options:0 |
+ error:nil]; |
+ if (![jsonObject isKindOfClass:[NSDictionary class]]) { |
+ RTCLogError(@"Unexpected message: %@", jsonObject); |
+ return; |
+ } |
+ NSDictionary *wssMessage = jsonObject; |
+ NSString *errorString = wssMessage[kARDWSSMessageErrorKey]; |
+ if (errorString.length) { |
+ RTCLogError(@"WSS error: %@", errorString); |
+ return; |
+ } |
+ NSString *payload = wssMessage[kARDWSSMessagePayloadKey]; |
+ ARDSignalingMessage *signalingMessage = |
+ [ARDSignalingMessage messageFromJSONString:payload]; |
+ RTCLog(@"WSS->C: %@", payload); |
+ [_delegate channel:self didReceiveMessage:signalingMessage]; |
+} |
+ |
+- (void)webSocket:(SRWebSocket *)webSocket didFailWithError:(NSError *)error { |
+ RTCLogError(@"WebSocket error: %@", error); |
+ self.state = kARDSignalingChannelStateError; |
+} |
+ |
+- (void)webSocket:(SRWebSocket *)webSocket |
+ didCloseWithCode:(NSInteger)code |
+ reason:(NSString *)reason |
+ wasClean:(BOOL)wasClean { |
+ RTCLog(@"WebSocket closed with code: %ld reason:%@ wasClean:%d", |
+ (long)code, reason, wasClean); |
+ NSParameterAssert(_state != kARDSignalingChannelStateError); |
+ self.state = kARDSignalingChannelStateClosed; |
+} |
+ |
+#pragma mark - Private |
+ |
+- (void)registerWithCollider { |
+ if (_state == kARDSignalingChannelStateRegistered) { |
+ return; |
+ } |
+ NSParameterAssert(_roomId.length); |
+ NSParameterAssert(_clientId.length); |
+ NSDictionary *registerMessage = @{ |
+ @"cmd": @"register", |
+ @"roomid" : _roomId, |
+ @"clientid" : _clientId, |
+ }; |
+ NSData *message = |
+ [NSJSONSerialization dataWithJSONObject:registerMessage |
+ options:NSJSONWritingPrettyPrinted |
+ error:nil]; |
+ NSString *messageString = |
+ [[NSString alloc] initWithData:message encoding:NSUTF8StringEncoding]; |
+ RTCLog(@"Registering on WSS for rid:%@ cid:%@", _roomId, _clientId); |
+ // Registration can fail if server rejects it. For example, if the room is |
+ // full. |
+ [_socket send:messageString]; |
+ self.state = kARDSignalingChannelStateRegistered; |
+} |
+ |
+@end |
+ |
+@interface ARDLoopbackWebSocketChannel () <ARDSignalingChannelDelegate> |
+@end |
+ |
+@implementation ARDLoopbackWebSocketChannel |
+ |
+- (instancetype)initWithURL:(NSURL *)url restURL:(NSURL *)restURL { |
+ return [super initWithURL:url restURL:restURL delegate:self]; |
+} |
+ |
+#pragma mark - ARDSignalingChannelDelegate |
+ |
+- (void)channel:(id<ARDSignalingChannel>)channel |
+ didReceiveMessage:(ARDSignalingMessage *)message { |
+ switch (message.type) { |
+ case kARDSignalingMessageTypeOffer: { |
+ // Change message to answer, send back to server. |
+ ARDSessionDescriptionMessage *sdpMessage = |
+ (ARDSessionDescriptionMessage *)message; |
+ RTCSessionDescription *description = sdpMessage.sessionDescription; |
+ NSString *dsc = description.sdp; |
+ dsc = [dsc stringByReplacingOccurrencesOfString:@"offer" |
+ withString:@"answer"]; |
+ RTCSessionDescription *answerDescription = |
+ [[RTCSessionDescription alloc] initWithType:RTCSdpTypeAnswer sdp:dsc]; |
+ ARDSignalingMessage *answer = |
+ [[ARDSessionDescriptionMessage alloc] |
+ initWithDescription:answerDescription]; |
+ [self sendMessage:answer]; |
+ break; |
+ } |
+ case kARDSignalingMessageTypeAnswer: |
+ // Should not receive answer in loopback scenario. |
+ break; |
+ case kARDSignalingMessageTypeCandidate: |
+ case kARDSignalingMessageTypeCandidateRemoval: |
+ // Send back to server. |
+ [self sendMessage:message]; |
+ break; |
+ case kARDSignalingMessageTypeBye: |
+ // Nothing to do. |
+ return; |
+ } |
+} |
+ |
+- (void)channel:(id<ARDSignalingChannel>)channel |
+ didChangeState:(ARDSignalingChannelState)state { |
+} |
+ |
+@end |
+ |