Example how to reject a handshake as a server

Author: marci4

src/main/example/ServerRejectHandshakeExample.java

@@ -0,0 +1,90 @@
+/*
+ * Copyright (c) 2010-2017 Nathan Rajlich
+ *
+ * Permission is hereby granted, free of charge, to any person
+ * obtaining a copy of this software and associated documentation
+ * files (the "Software"), to deal in the Software without
+ * restriction, including without limitation the rights to use,
+ * copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following
+ * conditions:
+ *
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
+ * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+ * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
+ * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+ * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+ * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+ * OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+importorg.java_websocket.WebSocket;
+importorg.java_websocket.WebSocketImpl;
+importorg.java_websocket.drafts.Draft;
+importorg.java_websocket.exceptions.InvalidDataException;
+importorg.java_websocket.framing.CloseFrame;
+importorg.java_websocket.handshake.ClientHandshake;
+importorg.java_websocket.handshake.ServerHandshakeBuilder;
+
+importjava.io.BufferedReader;
+importjava.io.IOException;
+importjava.io.InputStreamReader;
+importjava.net.InetSocketAddress;
+
+/**
+ * This example shows how to reject a handshake as a server from a client.
+ *
+ * For this you have to override onWebsocketHandshakeReceivedAsServer in your WebSocketServer class
+ */
+publicclassServerRejectHandshakeExampleextendsChatServer{
+
+publicServerRejectHandshakeExample( intport ){
+super( newInetSocketAddress( port ));
+ }
+
+@Override
+publicServerHandshakeBuilderonWebsocketHandshakeReceivedAsServer( WebSocketconn, Draftdraft, ClientHandshakerequest) throwsInvalidDataException{
+ServerHandshakeBuilderbuilder = super.onWebsocketHandshakeReceivedAsServer( conn, draft, request );
+//In this example we don't allow any resource descriptor ( "ws://localhost:8887/?roomid=1 will be rejected but ws://localhost:8887 is fine)
+if (! request.getResourceDescriptor().equals( "/" )){
+thrownewInvalidDataException( CloseFrame.POLICY_VALIDATION, "Not accepted!");
+ }
+//If there are no cookies set reject it as well.
+if (!request.hasFieldValue( "Cookie" )){
+thrownewInvalidDataException( CloseFrame.POLICY_VALIDATION, "Not accepted!");
+ }
+//If the cookie does not contain a specific value
+if (!request.getFieldValue( "Cookie" ).equals( "username=nemo" )){
+thrownewInvalidDataException( CloseFrame.POLICY_VALIDATION, "Not accepted!");
+ }
+returnbuilder;
+ }
+
+
+publicstaticvoidmain( String[] args ) throwsInterruptedException , IOException{
+WebSocketImpl.DEBUG = true;
+intport = 8887; // 843 flash policy port
+try{
+port = Integer.parseInt( args[ 0 ] );
+ } catch ( Exceptionex ){
+ }
+ServerRejectHandshakeExamples = newServerRejectHandshakeExample( port );
+s.start();
+System.out.println( "Server started on port: " + s.getPort() );
+
+BufferedReadersysin = newBufferedReader( newInputStreamReader( System.in ) );
+while ( true ){
+Stringin = sysin.readLine();
+s.broadcast( in );
+if( in.equals( "exit" ) ){
+s.stop();
+break;
+ }
+ }
+ }
+}