1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.eclipse.jetty.websocket.client.io;
20
21 import java.net.InetSocketAddress;
22 import java.nio.ByteBuffer;
23 import java.util.concurrent.Executor;
24 import java.util.concurrent.atomic.AtomicBoolean;
25
26 import org.eclipse.jetty.io.EndPoint;
27 import org.eclipse.jetty.util.log.Log;
28 import org.eclipse.jetty.util.log.Logger;
29 import org.eclipse.jetty.websocket.api.ProtocolException;
30 import org.eclipse.jetty.websocket.api.WebSocketPolicy;
31 import org.eclipse.jetty.websocket.api.WriteCallback;
32 import org.eclipse.jetty.websocket.api.extensions.Frame;
33 import org.eclipse.jetty.websocket.api.extensions.IncomingFrames;
34 import org.eclipse.jetty.websocket.client.masks.Masker;
35 import org.eclipse.jetty.websocket.common.WebSocketFrame;
36 import org.eclipse.jetty.websocket.common.WebSocketSession;
37 import org.eclipse.jetty.websocket.common.io.AbstractWebSocketConnection;
38
39
40
41
42 public class WebSocketClientConnection extends AbstractWebSocketConnection
43 {
44 private static final Logger LOG = Log.getLogger(WebSocketClientConnection.class);
45 private final ConnectPromise connectPromise;
46 private final Masker masker;
47 private final AtomicBoolean opened = new AtomicBoolean(false);
48
49 public WebSocketClientConnection(EndPoint endp, Executor executor, ConnectPromise connectPromise, WebSocketPolicy policy)
50 {
51 super(endp,executor,connectPromise.getClient().getScheduler(),policy,connectPromise.getClient().getBufferPool());
52 this.connectPromise = connectPromise;
53 this.masker = connectPromise.getMasker();
54 assert (this.masker != null);
55 }
56
57 @Override
58 public InetSocketAddress getLocalAddress()
59 {
60 return getEndPoint().getLocalAddress();
61 }
62
63 @Override
64 public InetSocketAddress getRemoteAddress()
65 {
66 return getEndPoint().getRemoteAddress();
67 }
68
69 @Override
70 public void onClose()
71 {
72 super.onClose();
73 ConnectionManager connectionManager = connectPromise.getClient().getConnectionManager();
74 connectionManager.removeSession(getSession());
75 }
76
77 @Override
78 public void onOpen()
79 {
80 boolean beenOpened = opened.getAndSet(true);
81 if (!beenOpened)
82 {
83 WebSocketSession session = getSession();
84 ConnectionManager connectionManager = connectPromise.getClient().getConnectionManager();
85 connectionManager.addSession(session);
86 connectPromise.succeeded(session);
87
88 ByteBuffer extraBuf = connectPromise.getResponse().getRemainingBuffer();
89 if (extraBuf.hasRemaining())
90 {
91 LOG.debug("Parsing extra remaining buffer from UpgradeConnection");
92 getParser().parse(extraBuf);
93 }
94 }
95 super.onOpen();
96 }
97
98
99
100
101 @Override
102 public void outgoingFrame(Frame frame, WriteCallback callback)
103 {
104 if (frame instanceof WebSocketFrame)
105 {
106 if (masker == null)
107 {
108 ProtocolException ex = new ProtocolException("Must set a Masker");
109 LOG.warn(ex);
110 if (callback != null)
111 {
112 callback.writeFailed(ex);
113 }
114 return;
115 }
116 masker.setMask((WebSocketFrame)frame);
117 }
118 super.outgoingFrame(frame,callback);
119 }
120
121 @Override
122 public void setNextIncomingFrames(IncomingFrames incoming)
123 {
124 getParser().setIncomingFramesHandler(incoming);
125 }
126 }