Exemplo n.º 1
0
  @Test
  public void testAuthentication() throws Exception {
    final AtomicReference<String> sessionId = new AtomicReference<>();
    class A extends DefaultSecurityPolicy implements ServerSession.RemoveListener {
      @Override
      public boolean canHandshake(
          BayeuxServer server, ServerSession session, ServerMessage message) {
        Map<String, Object> ext = message.getExt();
        if (ext == null) return false;

        Object authn = ext.get("authentication");
        if (!(authn instanceof Map)) return false;

        @SuppressWarnings("unchecked")
        Map<String, Object> authentication = (Map<String, Object>) authn;

        String token = (String) authentication.get("token");
        if (token == null) return false;

        sessionId.set(session.getId());
        session.addListener(this);

        return true;
      }

      @Override
      public void removed(ServerSession session, boolean timeout) {
        sessionId.set(null);
      }
    }
    A authenticator = new A();

    bayeux.setSecurityPolicy(authenticator);
    BayeuxClient client = newBayeuxClient();

    Map<String, Object> authentication = new HashMap<>();
    authentication.put("token", "1234567890");
    Message.Mutable fields = new HashMapMessage();
    fields.getExt(true).put("authentication", authentication);
    client.handshake(fields);

    Assert.assertTrue(client.waitFor(5000, State.CONNECTED));

    Assert.assertEquals(client.getId(), sessionId.get());

    disconnectBayeuxClient(client);

    Assert.assertNull(sessionId.get());
  }
Exemplo n.º 2
0
  @Test
  public void testIdleLongPollDoesNotCauseMultipleClientsAdvice() throws Exception {
    startServer(null);

    final long timeout = 2000;
    final long sleep = 500;
    bayeux.setTransports(
        new JSONTransport(bayeux) {
          {
            setOption(TIMEOUT_OPTION, timeout);
            init();
          }

          @Override
          protected LongPollScheduler newLongPollScheduler(
              ServerSessionImpl session,
              AsyncContext asyncContext,
              ServerMessage.Mutable metaConnectReply,
              String browserId) {
            return new LongPollScheduler(session, asyncContext, metaConnectReply, browserId) {
              private final AtomicInteger decrements = new AtomicInteger();

              @Override
              public void onComplete(final AsyncEvent asyncEvent) throws IOException {
                if (decrements.incrementAndGet() == 1) {
                  // Simulate that onComplete() is delayed without blocking
                  // this thread, to cause a race condition
                  new Thread() {
                    @Override
                    public void run() {
                      try {
                        Thread.sleep(sleep);
                        superOnComplete(asyncEvent);
                      } catch (Exception x) {
                        x.printStackTrace();
                      }
                    }
                  }.start();
                } else {
                  superOnComplete(asyncEvent);
                }
              }

              private void superOnComplete(AsyncEvent asyncEvent) throws IOException {
                super.onComplete(asyncEvent);
              }
            };
          }
        });

    Request handshake =
        newBayeuxRequest(
            "[{"
                + "\"channel\": \"/meta/handshake\","
                + "\"version\": \"1.0\","
                + "\"minimumVersion\": \"1.0\","
                + "\"supportedConnectionTypes\": [\"long-polling\"]"
                + "}]");
    ContentResponse response = handshake.send();
    Assert.assertEquals(200, response.getStatus());

    String clientId = extractClientId(response);

    Request connect1 =
        newBayeuxRequest(
            "[{"
                + "\"channel\": \"/meta/connect\","
                + "\"clientId\": \""
                + clientId
                + "\","
                + "\"connectionType\": \"long-polling\""
                + "}]");
    response = connect1.send();
    Assert.assertEquals(200, response.getStatus());

    ServerSession serverSession = bayeux.getSession(clientId);
    Assert.assertNotNull(serverSession);

    Request connect2 =
        newBayeuxRequest(
            "[{"
                + "\"channel\": \"/meta/connect\","
                + "\"clientId\": \""
                + clientId
                + "\","
                + "\"connectionType\": \"long-polling\""
                + "}]");
    response = connect2.send();
    Assert.assertEquals(200, response.getStatus());

    Request connect3 =
        newBayeuxRequest(
            "[{"
                + "\"channel\": \"/meta/connect\","
                + "\"clientId\": \""
                + clientId
                + "\","
                + "\"connectionType\": \"long-polling\""
                + "}]");
    response = connect3.send();
    Assert.assertEquals(200, response.getStatus());

    JSONContext.Client jsonContext = new JettyJSONContextClient();
    Message.Mutable[] messages = jsonContext.parse(response.getContentAsString());
    Assert.assertEquals(1, messages.length);
    Message.Mutable message = messages[0];
    Map<String, Object> advice = message.getAdvice();
    Assert.assertNull(advice);
  }