FilesExpand file tree

 master

/

SSLServer.java

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

154 lines (130 loc) · 4.09 KB

 master

/

SSLServer.java

Copy path

Top

File metadata and controls

• Code
• Blame

154 lines (130 loc) · 4.09 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

// @formatter:off

import java.io.File;

import java.io.FileInputStream;

import java.net.InetAddress;

import java.net.InetSocketAddress;

import java.net.Socket;

import java.nio.channels.SocketChannel;

import java.security.KeyStore;

import java.util.List;

import javax.net.ssl.KeyManagerFactory;

import javax.net.ssl.SSLContext;

import javax.net.ssl.SSLEngine;

import javax.net.ssl.TrustManagerFactory;

import org.java_websocket.SSLSocketChannel;

import org.java_websocket.WebSocket;

import org.java_websocket.WebSocketAdapter;

import org.java_websocket.WebSocketImpl;

import org.java_websocket.drafts.Draft;

import org.java_websocket.handshake.ClientHandshake;

import org.java_websocket.server.WebSocketServer;

/*

* Create the appropriate websocket server.

*/

public class SSLServer implements WebSocketServer.WebSocketServerFactory

{

private static final String STORETYPE = "JKS";

private static final String KEYSTORE = "keystore.jks";

private static final String STOREPASSWORD = "storepassword";

private static final String KEYPASSWORD = "keypassword";

public static void main(String[] args) throws Exception

{

new SSLServer();

}

private SSLContext sslContext;

void loadFromFile() throws Exception

{

// load up the key store

KeyStore ks = KeyStore.getInstance(STORETYPE);

File kf = new File(KEYSTORE);

ks.load(new FileInputStream(kf), STOREPASSWORD.toCharArray());

KeyManagerFactory kmf = KeyManagerFactory.getInstance("SunX509");

kmf.init(ks, KEYPASSWORD.toCharArray());

TrustManagerFactory tmf = TrustManagerFactory.getInstance("SunX509");

tmf.init(ks);

sslContext = SSLContext.getInstance("TLS");

sslContext.init(kmf.getKeyManagers(), tmf.getTrustManagers(), null);

}

/*

* Keystore with certificate created like so (in JKS format):

*

keytool -genkey -validity 3650 -keystore "keystore.jks" -storepass "storepassword" -keypass "keypassword" -alias "default" -dname "CN=127.0.0.1, OU=MyOrgUnit, O=MyOrg, L=MyCity, S=MyRegion, C=MyCountry"

*/

SSLServer() throws Exception

{

sslContext = null;

loadFromFile();

// create the web socket server

WebSocketSource wsgateway = new WebSocketSource(8001, InetAddress.getByName("127.0.0.1"));

wsgateway.setWebSocketFactory(this);

wsgateway.start();

}

@Override

public WebSocketImpl createWebSocket( WebSocketAdapter a, Draft d, Socket c ) {

return new WebSocketImpl( a, d, c );

}

@Override

public WebSocketImpl createWebSocket( WebSocketAdapter a, List<Draft> d, Socket s ) {

if(sslContext != null) try{

SSLEngine e = sslContext.createSSLEngine();

e.setUseClientMode(false);

return new WebSocketImpl( a, d, s );

} catch ( Exception e1 ) {

}

return new WebSocketImpl( a, d, s );

}

@Override

public SocketChannel wrapChannel( SocketChannel c ) {

if( sslContext != null )

try {

SSLEngine e = sslContext.createSSLEngine();

e.setUseClientMode( false );

new SSLSocketChannel( c, e );

} catch ( Exception e1 ) {

}

return c;

}

class WebSocketSource extends WebSocketServer

{

private WebSocket handle;

WebSocketSource(int port, InetAddress addr)

{

super(new InetSocketAddress(addr, port));

handle = null;

}

@Override

public void onClose(WebSocket arg0, int arg1, String arg2, boolean arg3)

{

System.err.println("---------------------------->Closed");

if(arg0 == handle) handle = null;

}

@Override

public void onError(WebSocket arg0, Exception arg1) {

// TODO Auto-generated method stub

}

@Override

public void onMessage(WebSocket arg0, String arg1)

{

if(arg0 != handle){

arg0.close(org.java_websocket.framing.CloseFrame.NORMAL);

return;

}

System.out.println("--------->["+arg1+"]");

}

@Override

public void onOpen(WebSocket arg0, ClientHandshake arg1)

{

// nothing to see just yet

if(handle == null){

handle = arg0;

}else if(handle != arg0){

arg0.close(org.java_websocket.framing.CloseFrame.NORMAL);

}

}

void done()

{

if(handle != null) handle.close(org.java_websocket.framing.CloseFrame.NORMAL);

}

}

}

//@formatter:on