View Javadoc
1   package com.ericsson.research.transport.ws.spi;
2   
3   /*
4    * ##_BEGIN_LICENSE_##
5    * Transport Abstraction Package (trap)
6    * ----------
7    * Copyright (C) 2014 Ericsson AB
8    * ----------
9    * Redistribution and use in source and binary forms, with or without modification,
10   * are permitted provided that the following conditions are met:
11   * 
12   * 1. Redistributions of source code must retain the above copyright notice, this
13   *    list of conditions and the following disclaimer.
14   * 
15   * 2. Redistributions in binary form must reproduce the above copyright notice,
16   *    this list of conditions and the following disclaimer in the documentation
17   *    and/or other materials provided with the distribution.
18   * 
19   * 3. Neither the name of the Ericsson AB nor the names of its contributors
20   *    may be used to endorse or promote products derived from this software without
21   *    specific prior written permission.
22   * 
23   * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
24   * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
25   * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
26   * IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
27   * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
28   * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
29   * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
30   * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
31   * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
32   * OF THE POSSIBILITY OF SUCH DAMAGE.
33   * ##_END_LICENSE_##
34   */
35  
36  import java.io.IOException;
37  import java.io.OutputStream;
38  
39  import com.ericsson.research.transport.ws.WSException;
40  import com.ericsson.research.transport.ws.WSSecurityContext;
41  
42  public class WSPrefetcherHandshake extends WSAbstractHandshake implements WSConstants {
43  
44  	private final WSSecurityContext securityContext;
45  	
46  	public WSPrefetcherHandshake(WSSecurityContext securityContext) {
47  		super(null);
48  		this.securityContext = securityContext;
49  	}
50  	
51  	public void preambleRead() throws WSException {}
52  
53  	public void headersRead() throws WSException {
54  		if(getHeader(SEC_WEBSOCKET_VERSION_HEADER)!=null) {
55  			if("13".equals(getHeader(SEC_WEBSOCKET_VERSION_HEADER)))
56  				protocol = new WSRfc6455(securityContext);
57  			else
58  				if("8".equals(getHeader(SEC_WEBSOCKET_VERSION_HEADER)))
59  					protocol = new WSHybi10(securityContext);
60  		} else {
61  			if(getHeader(SEC_WEBSOCKET_KEY1_HEADER) != null && getHeader(SEC_WEBSOCKET_KEY2_HEADER) != null)
62  				protocol = new WSHixie76(securityContext);
63  			else
64  				if(getHeader(SEC_WEBSOCKET_KEY1_HEADER) == null && getHeader(SEC_WEBSOCKET_KEY2_HEADER) == null && getHeader(SEC_WEBSOCKET_KEY_HEADER) == null)
65  					protocol = new WSHixie75(securityContext);
66  		}
67  		if(protocol == null)
68  			throw new WSException("Could not determine WebSocket client version");
69  		WSAbstractHandshake handshake = protocol.getHandshake();
70  		expectedHandshakeBodyLength = handshake.expectedHandshakeBodyLength;
71  		handshake.preamble = preamble;
72  		handshake.headers = headers;
73  		handshake.preambleRead();
74  		handshake.headersRead();
75  	}
76  	
77  	public void bodyRead() throws WSException {
78  		protocol.getHandshake().body = body;
79  		protocol.getHandshake().bodyRead();
80  	}
81  
82  	public void sendResponse() throws IOException {
83  		protocol.getHandshake().sendResponse();
84  	}
85  
86  	public void sendHandshake(OutputStream os) throws IOException {
87  		protocol.getHandshake().sendHandshake(os);
88  	}
89  
90  }