OLD | NEW |
| (Empty) |
1 /* | |
2 * libjingle | |
3 * Copyright 2014 Google Inc. | |
4 * | |
5 * Redistribution and use in source and binary forms, with or without | |
6 * modification, are permitted provided that the following conditions are met: | |
7 * | |
8 * 1. Redistributions of source code must retain the above copyright notice, | |
9 * this list of conditions and the following disclaimer. | |
10 * 2. Redistributions in binary form must reproduce the above copyright notice, | |
11 * this list of conditions and the following disclaimer in the documentation | |
12 * and/or other materials provided with the distribution. | |
13 * 3. The name of the author may not be used to endorse or promote products | |
14 * derived from this software without specific prior written permission. | |
15 * | |
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED | |
17 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF | |
18 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO | |
19 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | |
20 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, | |
21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; | |
22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, | |
23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR | |
24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF | |
25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
26 */ | |
27 | |
28 #import "ARDWebSocketChannel.h" | |
29 | |
30 #import "RTCLogging.h" | |
31 #import "SRWebSocket.h" | |
32 | |
33 #import "ARDUtilities.h" | |
34 | |
35 // TODO(tkchin): move these to a configuration object. | |
36 static NSString const *kARDWSSMessageErrorKey = @"error"; | |
37 static NSString const *kARDWSSMessagePayloadKey = @"msg"; | |
38 | |
39 @interface ARDWebSocketChannel () <SRWebSocketDelegate> | |
40 @end | |
41 | |
42 @implementation ARDWebSocketChannel { | |
43 NSURL *_url; | |
44 NSURL *_restURL; | |
45 SRWebSocket *_socket; | |
46 } | |
47 | |
48 @synthesize delegate = _delegate; | |
49 @synthesize state = _state; | |
50 @synthesize roomId = _roomId; | |
51 @synthesize clientId = _clientId; | |
52 | |
53 - (instancetype)initWithURL:(NSURL *)url | |
54 restURL:(NSURL *)restURL | |
55 delegate:(id<ARDSignalingChannelDelegate>)delegate { | |
56 if (self = [super init]) { | |
57 _url = url; | |
58 _restURL = restURL; | |
59 _delegate = delegate; | |
60 _socket = [[SRWebSocket alloc] initWithURL:url]; | |
61 _socket.delegate = self; | |
62 RTCLog(@"Opening WebSocket."); | |
63 [_socket open]; | |
64 } | |
65 return self; | |
66 } | |
67 | |
68 - (void)dealloc { | |
69 [self disconnect]; | |
70 } | |
71 | |
72 - (void)setState:(ARDSignalingChannelState)state { | |
73 if (_state == state) { | |
74 return; | |
75 } | |
76 _state = state; | |
77 [_delegate channel:self didChangeState:_state]; | |
78 } | |
79 | |
80 - (void)registerForRoomId:(NSString *)roomId | |
81 clientId:(NSString *)clientId { | |
82 NSParameterAssert(roomId.length); | |
83 NSParameterAssert(clientId.length); | |
84 _roomId = roomId; | |
85 _clientId = clientId; | |
86 if (_state == kARDSignalingChannelStateOpen) { | |
87 [self registerWithCollider]; | |
88 } | |
89 } | |
90 | |
91 - (void)sendMessage:(ARDSignalingMessage *)message { | |
92 NSParameterAssert(_clientId.length); | |
93 NSParameterAssert(_roomId.length); | |
94 NSData *data = [message JSONData]; | |
95 if (_state == kARDSignalingChannelStateRegistered) { | |
96 NSString *payload = | |
97 [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding]; | |
98 NSDictionary *message = @{ | |
99 @"cmd": @"send", | |
100 @"msg": payload, | |
101 }; | |
102 NSData *messageJSONObject = | |
103 [NSJSONSerialization dataWithJSONObject:message | |
104 options:NSJSONWritingPrettyPrinted | |
105 error:nil]; | |
106 NSString *messageString = | |
107 [[NSString alloc] initWithData:messageJSONObject | |
108 encoding:NSUTF8StringEncoding]; | |
109 RTCLog(@"C->WSS: %@", messageString); | |
110 [_socket send:messageString]; | |
111 } else { | |
112 NSString *dataString = | |
113 [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding]; | |
114 RTCLog(@"C->WSS POST: %@", dataString); | |
115 NSString *urlString = | |
116 [NSString stringWithFormat:@"%@/%@/%@", | |
117 [_restURL absoluteString], _roomId, _clientId]; | |
118 NSURL *url = [NSURL URLWithString:urlString]; | |
119 [NSURLConnection sendAsyncPostToURL:url | |
120 withData:data | |
121 completionHandler:nil]; | |
122 } | |
123 } | |
124 | |
125 - (void)disconnect { | |
126 if (_state == kARDSignalingChannelStateClosed || | |
127 _state == kARDSignalingChannelStateError) { | |
128 return; | |
129 } | |
130 [_socket close]; | |
131 RTCLog(@"C->WSS DELETE rid:%@ cid:%@", _roomId, _clientId); | |
132 NSString *urlString = | |
133 [NSString stringWithFormat:@"%@/%@/%@", | |
134 [_restURL absoluteString], _roomId, _clientId]; | |
135 NSURL *url = [NSURL URLWithString:urlString]; | |
136 NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:url]; | |
137 request.HTTPMethod = @"DELETE"; | |
138 request.HTTPBody = nil; | |
139 [NSURLConnection sendAsyncRequest:request completionHandler:nil]; | |
140 } | |
141 | |
142 #pragma mark - SRWebSocketDelegate | |
143 | |
144 - (void)webSocketDidOpen:(SRWebSocket *)webSocket { | |
145 RTCLog(@"WebSocket connection opened."); | |
146 self.state = kARDSignalingChannelStateOpen; | |
147 if (_roomId.length && _clientId.length) { | |
148 [self registerWithCollider]; | |
149 } | |
150 } | |
151 | |
152 - (void)webSocket:(SRWebSocket *)webSocket didReceiveMessage:(id)message { | |
153 NSString *messageString = message; | |
154 NSData *messageData = [messageString dataUsingEncoding:NSUTF8StringEncoding]; | |
155 id jsonObject = [NSJSONSerialization JSONObjectWithData:messageData | |
156 options:0 | |
157 error:nil]; | |
158 if (![jsonObject isKindOfClass:[NSDictionary class]]) { | |
159 RTCLogError(@"Unexpected message: %@", jsonObject); | |
160 return; | |
161 } | |
162 NSDictionary *wssMessage = jsonObject; | |
163 NSString *errorString = wssMessage[kARDWSSMessageErrorKey]; | |
164 if (errorString.length) { | |
165 RTCLogError(@"WSS error: %@", errorString); | |
166 return; | |
167 } | |
168 NSString *payload = wssMessage[kARDWSSMessagePayloadKey]; | |
169 ARDSignalingMessage *signalingMessage = | |
170 [ARDSignalingMessage messageFromJSONString:payload]; | |
171 RTCLog(@"WSS->C: %@", payload); | |
172 [_delegate channel:self didReceiveMessage:signalingMessage]; | |
173 } | |
174 | |
175 - (void)webSocket:(SRWebSocket *)webSocket didFailWithError:(NSError *)error { | |
176 RTCLogError(@"WebSocket error: %@", error); | |
177 self.state = kARDSignalingChannelStateError; | |
178 } | |
179 | |
180 - (void)webSocket:(SRWebSocket *)webSocket | |
181 didCloseWithCode:(NSInteger)code | |
182 reason:(NSString *)reason | |
183 wasClean:(BOOL)wasClean { | |
184 RTCLog(@"WebSocket closed with code: %ld reason:%@ wasClean:%d", | |
185 (long)code, reason, wasClean); | |
186 NSParameterAssert(_state != kARDSignalingChannelStateError); | |
187 self.state = kARDSignalingChannelStateClosed; | |
188 } | |
189 | |
190 #pragma mark - Private | |
191 | |
192 - (void)registerWithCollider { | |
193 if (_state == kARDSignalingChannelStateRegistered) { | |
194 return; | |
195 } | |
196 NSParameterAssert(_roomId.length); | |
197 NSParameterAssert(_clientId.length); | |
198 NSDictionary *registerMessage = @{ | |
199 @"cmd": @"register", | |
200 @"roomid" : _roomId, | |
201 @"clientid" : _clientId, | |
202 }; | |
203 NSData *message = | |
204 [NSJSONSerialization dataWithJSONObject:registerMessage | |
205 options:NSJSONWritingPrettyPrinted | |
206 error:nil]; | |
207 NSString *messageString = | |
208 [[NSString alloc] initWithData:message encoding:NSUTF8StringEncoding]; | |
209 RTCLog(@"Registering on WSS for rid:%@ cid:%@", _roomId, _clientId); | |
210 // Registration can fail if server rejects it. For example, if the room is | |
211 // full. | |
212 [_socket send:messageString]; | |
213 self.state = kARDSignalingChannelStateRegistered; | |
214 } | |
215 | |
216 @end | |
OLD | NEW |