Skip to main content
aboutsummaryrefslogtreecommitdiffstats
blob: c24a553027502423ca3aa4a1001929a4025417f5 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
/*
 * Copyright (c) 2004 - 2012 Eike Stepper (Berlin, Germany) and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *    Eike Stepper - initial API and implementation
 */
package org.eclipse.net4j.tests.defs;

import org.eclipse.net4j.Net4jUtil;
import org.eclipse.net4j.buffer.IBufferPool;
import org.eclipse.net4j.internal.tcp.TCPAcceptor;
import org.eclipse.net4j.internal.tcp.TCPClientConnector;
import org.eclipse.net4j.internal.tcp.TCPConnector;
import org.eclipse.net4j.internal.tcp.TCPSelector;
import org.eclipse.net4j.util.lifecycle.LifecycleUtil;
import org.eclipse.net4j.util.security.ChallengeNegotiator;
import org.eclipse.net4j.util.security.INegotiator;
import org.eclipse.net4j.util.security.IRandomizer;
import org.eclipse.net4j.util.security.PasswordCredentials;
import org.eclipse.net4j.util.security.PasswordCredentialsProvider;
import org.eclipse.net4j.util.security.Randomizer;
import org.eclipse.net4j.util.security.ResponseNegotiator;
import org.eclipse.net4j.util.security.UserManager;

import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

/**
 * @author Eike Stepper
 */
class Util
{
  public static TCPAcceptor createTCPAcceptor(String host, int port, long timeout)
  {
    TCPAcceptor acceptor = new TCPAcceptor();
    acceptor.setStartSynchronously(true);
    acceptor.setSynchronousStartTimeout(timeout);
    acceptor.getConfig().setBufferProvider(createBufferPool());
    acceptor.getConfig().setReceiveExecutor(createThreadPool());
    acceptor.setSelector(createTCPSelector());
    acceptor.setAddress(host);
    acceptor.setPort(port);

    LifecycleUtil.activate(acceptor);

    return acceptor;
  }

  private static IRandomizer createRandomizer()
  {
    Randomizer randomizer = new Randomizer();
    LifecycleUtil.activate(randomizer);

    return randomizer;
  }

  public static void addNegotiator(String userId, String password, TCPAcceptor tcpAcceptor)
  {
    UserManager userManager = new UserManager();
    LifecycleUtil.activate(userManager);
    userManager.addUser(userId, password.toCharArray());

    ChallengeNegotiator challengeNegotiator = new ChallengeNegotiator();
    challengeNegotiator.setRandomizer(createRandomizer());
    challengeNegotiator.setUserManager(userManager);
    LifecycleUtil.activate(challengeNegotiator);

    tcpAcceptor.getConfig().setNegotiator(challengeNegotiator);
  }

  public static void removeNegotiator(TCPAcceptor tcpAcceptor)
  {
    INegotiator negotiator = tcpAcceptor.getConfig().getNegotiator();
    tcpAcceptor.getConfig().setNegotiator(null);
    LifecycleUtil.deactivate(negotiator);
  }

  public static TCPConnector createTCPClientConnector(String host, int port, TCPSelector tcpSelector)
  {
    TCPClientConnector tcpConnector = new TCPClientConnector();
    tcpConnector.getConfig().setBufferProvider(createBufferPool());
    tcpConnector.getConfig().setReceiveExecutor(createThreadPool());
    tcpConnector.setSelector(tcpSelector);
    tcpConnector.setHost(host);
    tcpConnector.setPort(port);

    return tcpConnector;
  }

  public static void addNegotiator(String userId, String password, TCPConnector tcpConnector)
  {
    PasswordCredentials passwordCredentials = new PasswordCredentials(userId, password.toCharArray());
    PasswordCredentialsProvider credentialsProvider = new PasswordCredentialsProvider(passwordCredentials);
    LifecycleUtil.activate(credentialsProvider);

    tcpConnector.getConfig().setNegotiator(createResponseNegotiator(credentialsProvider));
  }

  public static ResponseNegotiator createResponseNegotiator(PasswordCredentialsProvider credentialsProvider)
  {
    ResponseNegotiator responseNegotiator = new ResponseNegotiator();
    responseNegotiator.setCredentialsProvider(credentialsProvider);
    LifecycleUtil.activate(responseNegotiator);
    return responseNegotiator;
  }

  public static void removeNegotiator(TCPConnector tcpConnector)
  {
    INegotiator negotiator = tcpConnector.getConfig().getNegotiator();
    tcpConnector.setNegotiator(null);
    LifecycleUtil.deactivate(negotiator);
  }

  public static TCPSelector createTCPSelector()
  {
    TCPSelector selector = new TCPSelector();
    LifecycleUtil.activate(selector);
    return selector;
  }

  public static IBufferPool createBufferPool()
  {
    IBufferPool bufferPool = Net4jUtil.createBufferPool();
    LifecycleUtil.activate(bufferPool);
    return bufferPool;
  }

  public static ExecutorService createThreadPool()
  {
    ExecutorService threadPool = Executors.newCachedThreadPool();
    LifecycleUtil.activate(threadPool);
    return threadPool;
  }
}

Back to the top