Java HandshakeRequest类代码示例

import javax.websocket.server.HandshakeRequest; //导入依赖的package包/类
@Override
public void modifyHandshake(ServerEndpointConfig config,
        HandshakeRequest request, HandshakeResponse response) {

    HttpSession httpSession = (HttpSession) request.getHttpSession();

    super.modifyHandshake(config, request, response);

    if (httpSession == null) {
        LOGGER.info("httpSession == null after modifyHandshake");
        httpSession = (HttpSession) request.getHttpSession();
    }

    if (httpSession == null) {
        LOGGER.info("httpSession == null");
        return;
    }

    config.getUserProperties().put("httpSession", httpSession);

    httpSession = (HttpSession) request.getHttpSession();
    LOGGER.info("modifyHandshake " + httpSession.getId());

}
 

import javax.websocket.server.HandshakeRequest; //导入依赖的package包/类
@Override
public void modifyHandshake(ServerEndpointConfig sec, HandshakeRequest request, HandshakeResponse response) {
    super.modifyHandshake(sec, request, response);

    if ( token == null || token.isEmpty() ) {
        Log.log(Level.FINEST, this, "No token set for room, skipping validation");
    } else {
        Log.log(Level.FINEST, this, "Validating WS handshake");
        SignedRequestHmac wsHmac = new SignedRequestHmac("", token, "", request.getRequestURI().getRawPath());

        try {
            wsHmac.checkHeaders(new SignedRequestMap.MLS_StringMap(request.getHeaders()))
                    .verifyFullSignature()
                    .wsResignRequest(new SignedRequestMap.MLS_StringMap(response.getHeaders()));

            Log.log(Level.INFO, this, "validated and resigned", wsHmac);
        } catch(Exception e) {
            Log.log(Level.WARNING, this, "Failed to validate HMAC, unable to establish connection", e);

            response.getHeaders().replace(HandshakeResponse.SEC_WEBSOCKET_ACCEPT, Collections.emptyList());
        }
    }
}
 

import javax.websocket.server.HandshakeRequest; //导入依赖的package包/类
@Override
public void modifyHandshake(ServerEndpointConfig config,
        HandshakeRequest request, HandshakeResponse response) {

  Map<String, List<String>> headers = request.getHeaders();
  if (headers != null && headers.containsKey(WatcherSecurityKey.HTTP_HEADER)) {
    List<String> header = headers.get(WatcherSecurityKey.HTTP_HEADER);
    if (header.size() > 0) {
      config.getUserProperties().put(WatcherSecurityKey.HTTP_HEADER, header.
              get(0));
    }
  }
  HttpSession httpSession = (HttpSession) request.getHttpSession();
  String user = request.getUserPrincipal().getName();
  config.getUserProperties().put("httpSession", httpSession);
  config.getUserProperties().put("user", user);
  logger.log(Level.INFO, "Hand shake for upgrade to websocket by: {0}", user);
}
 

import javax.websocket.server.HandshakeRequest; //导入依赖的package包/类
/**
 * Intercept the handshake operation so that we can take a hold of the
 * ServletContext instance to be able to retrieve attributes stored to it
 * such as the database object and other similar class instances
 * <p/>
 * @param config
 * @param request
 * @param response
 */
@Override
public void modifyHandshake(ServerEndpointConfig config,
        HandshakeRequest request, HandshakeResponse response) {

  HttpSession httpSession = (HttpSession) request.getHttpSession();
  ServletContext context = (ServletContext) httpSession.getServletContext();

  config.getUserProperties().put("httpSession", httpSession);
  config.getUserProperties().put("user", request.getUserPrincipal().getName());

  /*
   * store these attributes to servletContext so that they are available to
   * every created user socket session
   */
  config.getUserProperties().put("protocol", context.getAttribute("protocol"));
}
 

import javax.websocket.server.HandshakeRequest; //导入依赖的package包/类
/**
 * Test of handleOpenConnexion method, of class IWSControllerMonitor.
 * @throws java.lang.Exception
 */
@Test
public void testHandleOpenConnexion() throws Exception {
	System.out.println("handleOpenConnexion");
	Session session = mock(Session.class);
	EndpointConfig config = mock(EndpointConfig.class);
	HandshakeRequest handshakeRequest = mock(HandshakeRequest.class);
	HttpSession httpSession = mock(HttpSession.class);
	Map<String, Object> configProperties = mock(Map.class);
	
	when(config.getUserProperties()).thenReturn(configProperties);
	when(handshakeRequest.getHttpSession()).thenReturn(httpSession);
	when(httpSession.getId()).thenReturn("SESSIONID");
	when(configProperties.get(eq(Constants.HANDSHAKEREQUEST))).thenReturn(handshakeRequest);
	
	instance.handleOpenConnexion(session, config);
	
	verify(httpSessionManager).addSession(eq(session), eq("SESSIONID"));
	verify(iwse).handleOpenConnexion(eq(session), eq(config));
}
 

import javax.websocket.server.HandshakeRequest; //导入依赖的package包/类
/**
 * Return locale of client
 *
 * @param request
 * @return
 */
Locale getLocale(HandshakeRequest request) {
	if(null != request) {
		Map<String, List<String>> headers = request.getHeaders();
		if(null != headers) {
			List<String> accepts = headers.get(HttpHeaders.ACCEPT_LANGUAGE);
			logger.debug("Get accept-language from client headers : {}", accepts);
			if (null != accepts) {
				for (String accept : accepts) {
					try {
						return localeExtractor.extractFromAccept(accept);
					} catch (LocaleNotFoundException ex) {
					}
				}
			}
		}
	}
	return Locale.US;
}
 

import javax.websocket.server.HandshakeRequest; //导入依赖的package包/类
/**
 * Test of handleOpenConnexion method, of class WSEndpoint.
 *
 * @throws java.io.IOException
 */
@Test
public void testHandleOpenConnexionFromBrowser() throws IOException {
	System.out.println("handleOpenConnexion");
	HandshakeRequest request = mock(HandshakeRequest.class);
	EndpointConfig config = mock(EndpointConfig.class);
	Map<String, Object> map = new HashMap<>();
	map.put(Constants.HANDSHAKEREQUEST, request);
	Session session = mock(Session.class);

	when(config.getUserProperties()).thenReturn(map);
	when(session.getId()).thenReturn("WSSESSIONID");
	doNothing().when(instance).setContext(eq(request));

	instance.handleOpenConnexion(session, config);

	verify(userContextFactory).createUserContext(any(HandshakeRequest.class), eq("WSSESSIONID"));
}
 

import javax.websocket.server.HandshakeRequest; //导入依赖的package包/类
/**
 * Test of handleOpenConnexion method, of class IWSDecorator.
 * @throws java.lang.Exception
 */
@Test
public void testHandleOpenConnexion() throws Exception {
	System.out.println("handleOpenConnexion");
	Session session = mock(Session.class);
	EndpointConfig config = mock(EndpointConfig.class);
	Map map = mock(Map.class);
	HandshakeRequest request = mock(HandshakeRequest.class);
	HttpSession httpSession = mock(HttpSession.class);
	when(session.getId()).thenReturn("WSID");
	when(config.getUserProperties()).thenReturn(map);
	when(map.get(eq(Constants.HANDSHAKEREQUEST))).thenReturn(request);
	when(request.getHttpSession()).thenReturn(httpSession);
	when(httpSession.getId()).thenReturn("HTTPID");
	instance.handleOpenConnexion(session, config);
	verify(sessionManager).linkWsToHttp(eq(session), eq("HTTPID"));
	verify(iwse).handleOpenConnexion(eq(session), eq(config));
}
 

import javax.websocket.server.HandshakeRequest; //导入依赖的package包/类
/**
 * Test of getPrincipal method, of class PrincipalTools.
 */
@Test
public void testGetPrincipal_HandshakeRequest() {
	System.out.println("getPrincipal");
	HandshakeRequest handshakeRequest = mock(HandshakeRequest.class);
	Principal principal = mock(Principal.class);
	when(principal.getName()).thenReturn("FOO");
	when(handshakeRequest.getUserPrincipal()).thenReturn(null).thenReturn(principal);
	Principal result = instance.getPrincipal(handshakeRequest);
	assertThat(result).isNotNull();
	assertThat(result.getName()).isEqualTo(Constants.ANONYMOUS);

	result = instance.getPrincipal(handshakeRequest);
	assertThat(result).isEqualTo(principal);
	assertThat(result.getName()).isEqualTo("FOO");
	
	handshakeRequest = null;
	result = instance.getPrincipal(handshakeRequest);
	assertThat(result).isNotNull();
	assertThat(result.getName()).isEqualTo(Constants.ANONYMOUS);
}
 

import javax.websocket.server.HandshakeRequest; //导入依赖的package包/类
/**
 * Test of getPrincipal method, of class WsUserContext.
 */
@Test
public void testGetPrincipal() {
	System.out.println("getPrincipal");
	Principal principal = mock(Principal.class);
	PrincipalTools principalTools = mock(PrincipalTools.class);
	HandshakeRequest handshakeRequest = mock(HandshakeRequest.class);
	when(principalTools.getPrincipal(eq(handshakeRequest))).thenReturn(principal);

	WsUserContext instance = new WsUserContext(handshakeRequest);
	instance.principalTools = principalTools;

	Principal result = instance.getPrincipal();

	assertThat(result).isNotNull();
	assertThat(result).isEqualTo(principal);
}
 

import javax.websocket.server.HandshakeRequest; //导入依赖的package包/类
@Override
public Set<ServerEndpointConfig> getEndpointConfigs(Set<Class<? extends Endpoint>> set) {
    return new HashSet<ServerEndpointConfig>() {{
        add(ServerEndpointConfig.Builder
                .create(MyEndpoint.class, "/websocket")
                .configurator(new ServerEndpointConfig.Configurator() {

                    @Override
                    public void modifyHandshake(ServerEndpointConfig sec, HandshakeRequest request, HandshakeResponse response) {
                        HttpSession session = (HttpSession)request.getHttpSession();
                        System.out.println("HttpSession id: " + session.getId());
                        System.out.println("HttpSession creation time: " + session.getCreationTime());
                        super.modifyHandshake(sec, request, response);
                    }

                })
                .build());
    }};
}
 

import javax.websocket.server.HandshakeRequest; //导入依赖的package包/类
public void modifyHandshake(ServerEndpointConfig config, HandshakeRequest request, HandshakeResponse response) {
	HttpServletRequest _request = (HttpServletRequest) GenericReflection.NoThrow.getValue(_HandshakeRequest, request);
	Request _requestFaced = (Request) GenericReflection.NoThrow.getValue(Core.requestField, _request);
	
	MimeHeaders mime = new MimeHeaders();
	
	Enumeration<String> enuns = _requestFaced.getHeaderNames();
	while(enuns.hasMoreElements()){
		String param = (String) enuns.nextElement();		
		mime.addValue(param).setString(_requestFaced.getHeader(param));
	}
	
	Map<String, Object> properties = config.getUserProperties();
	
	properties.put("httpRequest", _request);
	properties.put("httpResponse", _request.getAttribute("httpResponse"));
	properties.put("httpSession", _request.getSession());
	properties.put("context", _requestFaced.getContext());
	properties.put("headers", mime);
	properties.put("remoteHost", _request.getRemoteHost());
	properties.put("localPort", _request.getLocalPort());		
	properties.put("remoteAddr", _request.getRemoteAddr());
}
 

import javax.websocket.server.HandshakeRequest; //导入依赖的package包/类
@Override
public String authenticate(HandshakeRequest request) throws AuthenticationException {
    SecurityContext context = SecurityContextHolder.getContext();
    if (context == null) {
        throw new AuthenticationException("User not authenticateded", "Anonymous");
    }

    Authentication authentication = context.getAuthentication();
    if (authentication == null) {
        throw new AuthenticationException("User not authenticateded", "Anonymous");
    }
    if (authentication.isAuthenticated()) {
        return authentication.getName();
    } else {
        throw new AuthenticationException("User not authenticateded", authentication.getName());
    }
}
 

import javax.websocket.server.HandshakeRequest; //导入依赖的package包/类
@Override
public String authenticate(AuthenticationProvider provider, HandshakeRequest request) throws AuthenticationException {
    AuthenticationProvider auth = resolveAuthenticationProvider(provider);
    if (auth != null) {
        try {
            String username = auth.authenticate(request);
            fireAuthentication(username, request, null);
            return username;
        } catch(AuthenticationException ae) {
            LOG.warn("Unauthorized access by " + ae.getUsername() + " : " +ae.getMessage());
            fireAuthentication(ae.getUsername(), request, ae);
            throw ae;
        }    
        
    }
    return null;
   
}
 

import javax.websocket.server.HandshakeRequest; //导入依赖的package包/类
@Override
public String getId(HandshakeRequest request,
                    ClientChannel manager) {
    String clientId = getIdValue(request);
   
    if (manager.hasClient(clientId)) {
        clientId = clientId + "-" + getUuid();
    }
    if (clientId == null) {
        clientId = getUuid();
    }
    
   
    
    return clientId;
}