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
|
// Tags: not-a-test
// Copyright (C) 2004 Sascha Brawer <brawer@dandelis.ch>
// This file is part of Mauve.
// Mauve is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2, or (at your option)
// any later version.
// Mauve is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
// You should have received a copy of the GNU General Public License
// along with Mauve; see the file COPYING. If not, write to
// the Free Software Foundation, 59 Temple Place - Suite 330,
// Boston, MA 02111-1307, USA.
package gnu.testlet.java.util.logging.SocketHandler;
import java.io.*;
import java.net.*;
/**
* A helper class that opens a TCP/IP server socket and starts
* a separate thread which listens for a connection request on
* that socket, reads in everything until the client closes the
* connection, and returns the transmitted content.
*
* @author <a href="mailto:brawer@dandelis.ch">Sascha Brawer</a>
*/
public class SocketCapturer
{
Server server;
Thread thread;
ServerSocket socket;
public SocketCapturer()
throws java.io.IOException
{
this(0);
}
public SocketCapturer(int port)
throws java.io.IOException
{
socket = new ServerSocket(port);
server = new Server(socket);
thread = new Thread(server);
thread.start();
}
public int getLocalPort()
{
return socket.getLocalPort();
}
/**
* Returns the transmitted content. Blocks the current thread
* until a client has transmitted all its content and closed the
* connection.
*/
public byte[] getCaptured()
{
try
{
thread.join();
}
catch (InterruptedException _)
{
}
return server.getCaptured();
}
private class Server
implements Runnable
{
ServerSocket serverSocket;
Socket conn;
private volatile Throwable thrown;
private volatile byte[] captured;
public Server(ServerSocket serverSocket)
{
this.serverSocket = serverSocket;
}
public byte[] getCaptured()
{
return captured;
}
public void run()
{
try
{
conn = serverSocket.accept();
InputStream is = conn.getInputStream();
ByteArrayOutputStream bos = new ByteArrayOutputStream();
int r = 0;
byte[] b = new byte[2000];
while ((r = is.read(b)) >= 0)
{
if (r > 0)
bos.write(b, 0, r);
}
conn.close();
captured = bos.toByteArray();
bos.close();
}
catch (Exception ex)
{
this.thrown = ex;
ex.printStackTrace();
}
}
}
}
|
