|
|
1
2
3
4
5
|
package pad.prac2;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
|
|
|
6
|
import java.io.OutputStream;
|
|
|
7
8
9
10
11
12
13
|
import java.net.InetSocketAddress;
import java.net.Socket;
import java.net.SocketAddress;
public class MySocket extends Socket
{
private BufferedReader input;
|
|
|
14
|
private OutputStream output;
|
|
|
15
16
17
18
|
public MySocket()
{
super();
|
|
|
19
20
21
22
23
|
}
public void connect(String host, int port)
{
SocketAddress addr = new InetSocketAddress(host,port);
|
|
|
24
25
|
try
{
|
|
|
26
27
|
super.connect(addr);
initializeStreams();
|
|
|
28
29
30
|
}
catch(IOException ioExc)
{
|
|
|
31
|
System.out.println("TCP: Error occured while connecting to remote host");
|
|
|
32
33
34
|
}
}
|
|
|
35
|
public void initializeStreams()
|
|
|
36
37
38
|
{
try
{
|
|
|
39
|
output = super.getOutputStream();
|
|
|
40
|
input = new BufferedReader(new InputStreamReader(super.getInputStream()));
|
|
|
41
42
43
|
}
catch(IOException ioExc)
{
|
|
|
44
|
System.out.println("TCP: Error initializing socket");
|
|
|
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
}
}
public void close()
{
try
{
super.shutdownInput();
super.shutdownOutput();
super.close();
}
catch(IOException ioExc)
{
System.out.println("TCP: Error while closing socket");
}
}
|
|
|
62
|
public void sendMsg(String msg) throws IOException
|
|
|
63
64
|
{
String length = new Integer(msg.length()).toString();
|
|
|
65
66
|
this.write(length + '\0');
this.write(msg);
|
|
|
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
}
public String recvMsg()
{
String len,str;
int length;
len = this.readLine();
if(len == null)
{
return len;
}
length = Integer.parseInt(len);
str = this.read(length);
return str;
}
|
|
|
83
|
public void write(String str) throws IOException
|
|
|
84
|
{
|
|
|
85
|
output.write(str.getBytes());
|
|
|
86
87
88
89
90
91
92
93
94
95
96
97
98
|
}
public String read(int bytes)
{
char[] buffer = new char[bytes];
try
{
input.read(buffer, 0, bytes);
return new String(buffer);
}
catch(IOException ioExc)
{
System.out.println("TCP: Error retrieving data from remote endpoint");
|
|
|
99
100
101
102
103
104
105
106
|
return null;
}
}
public String readLine()
{
try
{
|
|
|
107
108
109
110
111
112
113
114
115
116
117
118
|
String line = "";
char[] c = new char[1];
while(true)
{
input.read(c,0,1);
if(c[0] == '\0')
{
break;
}
line += c[0];
}
return line;
|
|
|
119
120
121
122
123
|
}
catch(IOException ioExc)
{
System.out.println("TCP: Error retrieving data from remote endpoint");
return null;
|
|
|
124
125
126
127
|
}
}
}
|
