annotate src/main/java/ac/ryukyu/treevnc/client/MyRfbProtoClient.java @ 7:30bd62abf424

add files
author one
date Wed, 08 Aug 2012 18:58:05 +0900
parents b32668b8e83c
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
3
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
1 package ac.ryukyu.treevnc.client;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
2
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
3 import java.io.BufferedInputStream;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
4 import java.io.DataInputStream;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
5 import java.io.DataOutputStream;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
6 import java.io.IOException;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
7 import java.io.InputStream;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
8 import java.io.OutputStream;
4
b32668b8e83c create multicast function
one
parents: 3
diff changeset
9 import java.net.BindException;
3
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
10 import java.net.InetAddress;
4
b32668b8e83c create multicast function
one
parents: 3
diff changeset
11 import java.net.ServerSocket;
3
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
12 import java.net.Socket;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
13 import java.nio.ByteBuffer;
4
b32668b8e83c create multicast function
one
parents: 3
diff changeset
14 import java.util.LinkedList;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
15 import java.util.concurrent.atomic.AtomicInteger;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
16 import java.util.zip.DataFormatException;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
17 import java.util.zip.Deflater;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
18 import java.util.zip.Inflater;
3
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
19
4
b32668b8e83c create multicast function
one
parents: 3
diff changeset
20 import ac.ryukyu.*;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
21 import ac.ryukyu.treevnc.MulticastQueue;
3
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
22
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
23 import com.glavsoft.exceptions.TransportException;
4
b32668b8e83c create multicast function
one
parents: 3
diff changeset
24 import com.glavsoft.rfb.encoding.EncodingType;
3
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
25 import com.glavsoft.transport.Reader;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
26 import com.glavsoft.viewer.ContainerManager;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
27 import com.glavsoft.viewer.swing.ParametersHandler.ConnectionParams;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
28
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
29 public class MyRfbProtoClient {
4
b32668b8e83c create multicast function
one
parents: 3
diff changeset
30 final static int FramebufferUpdate = 0;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
31 final static int CheckDelay = 11;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
32 final static String versionMsg_3_855 = "RFB 003.855\n";
b32668b8e83c create multicast function
one
parents: 3
diff changeset
33 private static final int INFLATE_BUFSIZE = 1024*100;
3
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
34 private Reader reader;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
35 private EchoClient echoValue = new EchoClient();
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
36 private String host,treenum,parent,pHost,leaderflag;
4
b32668b8e83c create multicast function
one
parents: 3
diff changeset
37 private int echoPort,port,acceptPort;
3
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
38 Socket clientSocket,sock;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
39 DataInputStream is;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
40 OutputStream os;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
41 private ContainerManager containerManager;
4
b32668b8e83c create multicast function
one
parents: 3
diff changeset
42 private ServerSocket servSock;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
43 private byte initData[];
b32668b8e83c create multicast function
one
parents: 3
diff changeset
44 private int clients = 0;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
45 private MulticastQueue<LinkedList<ByteBuffer>> multicastqueue = new MulticastQueue<LinkedList<ByteBuffer>>();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
46 boolean proxyFlag = false;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
47 int serverMajor, serverMinor;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
48 int clientMajor, clientMinor;
3
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
49
4
b32668b8e83c create multicast function
one
parents: 3
diff changeset
50 private Inflater inflater = new Inflater();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
51 private Deflater deflater = new Deflater();
3
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
52
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
53 public MyRfbProtoClient(Reader reader,String host,String port) {
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
54 this.reader = reader;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
55 }
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
56
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
57 public void setParam(ConnectionParams connectionParams) {
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
58 pHost = connectionParams.hostName;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
59 echoPort = connectionParams.portNumber;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
60 }
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
61
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
62 public boolean readProxyFlag() throws TransportException {
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
63 int flag = reader.readUInt8();
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
64 if(flag == 1)
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
65 return true;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
66 else
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
67 return false;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
68 }
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
69
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
70 public byte[] readEchoPort() throws Exception {
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
71 byte[] b = new byte[4];
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
72 reader.readBytes(b, 0, b.length);
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
73 //readFully(b);
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
74 return b;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
75 }
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
76
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
77 public void getParentName() {
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
78 if (echoValue == null) {
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
79
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
80 if (clientSocket == null) {
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
81
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
82 // echo = new EchoClient(pHost, this);
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
83 echoValue = new EchoClient(pHost, echoPort);
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
84 echoValue.openport();
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
85
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
86 echoValue = echoValue.requestHostName("1");
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
87 } else {
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
88 echoValue = new EchoClient();
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
89 echoValue = echoValue.Interruption(clientSocket);
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
90 }
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
91 }
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
92 // proxyからの返信で接続先を決定する
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
93 host = echoValue.responseLine;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
94 parent = echoValue.parent;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
95 if (echoValue.treenum != null) {
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
96 treenum = echoValue.treenum;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
97 } else {
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
98 treenum = echoValue.treenum;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
99 }
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
100
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
101 if (echoValue.leaderflag != null) {
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
102 leaderflag = echoValue.leaderflag;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
103 } else {
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
104 leaderflag = echoValue.leaderflag;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
105 }
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
106 }
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
107
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
108 int castByteInt(byte[] b) {
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
109 ByteBuffer bb = ByteBuffer.wrap(b);
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
110 int value = bb.getInt();
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
111 return value;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
112 }
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
113
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
114 Socket changeParent(String host, int port) throws IOException {
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
115 sock = new Socket(host, port);
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
116 return sock;
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
117 }
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
118
4
b32668b8e83c create multicast function
one
parents: 3
diff changeset
119 public Socket accept() throws IOException {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
120 return servSock.accept();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
121 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
122
b32668b8e83c create multicast function
one
parents: 3
diff changeset
123 void initServSock(int port) throws IOException {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
124 servSock = new ServerSocket(port);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
125 acceptPort = port;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
126 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
127
b32668b8e83c create multicast function
one
parents: 3
diff changeset
128 public void selectPort(int p) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
129 int port = p;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
130 while (true) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
131 try {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
132 initServSock(port);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
133 break;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
134 } catch (BindException e) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
135 port++;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
136 continue;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
137 } catch (IOException e) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
138
b32668b8e83c create multicast function
one
parents: 3
diff changeset
139 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
140 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
141 System.out.println("accept port = " + port);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
142 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
143
b32668b8e83c create multicast function
one
parents: 3
diff changeset
144 public void newClient(AcceptThread acceptThread, final Socket newCli,
b32668b8e83c create multicast function
one
parents: 3
diff changeset
145 final OutputStream os, final InputStream is) throws IOException {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
146 // createBimgFlag = true;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
147 // rfb.addSockTmp(newCli);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
148 // addSock(newCli);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
149 final int myId = clients;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
150 final MulticastQueue.Client <LinkedList<ByteBuffer>> c = multicastqueue.newClient();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
151 final AtomicInteger writerRunning = new AtomicInteger();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
152 writerRunning.set(1);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
153 /**
b32668b8e83c create multicast function
one
parents: 3
diff changeset
154 * Timeout thread. If a client is suspended, it has top of queue indefinitely, which caused memory
b32668b8e83c create multicast function
one
parents: 3
diff changeset
155 * overflow. After the timeout, we poll the queue and discard it. Start long wait if writer is running.
b32668b8e83c create multicast function
one
parents: 3
diff changeset
156 */
b32668b8e83c create multicast function
one
parents: 3
diff changeset
157 final Runnable timer = new Runnable() {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
158 public void run() {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
159 int count = 0;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
160 for(;;) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
161 long timeout = 40000/8;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
162 try {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
163 synchronized(this) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
164 int state,flag;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
165 writerRunning.set(0);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
166 wait(timeout);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
167 flag = 0;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
168 while((state=writerRunning.get())==0) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
169 c.poll(); // discard, should be timeout
b32668b8e83c create multicast function
one
parents: 3
diff changeset
170 count++;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
171 if (flag==0) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
172 System.out.println("Discarding "+myId + " count="+ count); flag = 1;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
173 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
174 wait(10); // if this is too short, writer cannot take the poll, if this is too long, memory will overflow...
b32668b8e83c create multicast function
one
parents: 3
diff changeset
175 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
176 if (flag==1) System.out.println("Resuming "+myId + " count="+count);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
177 if (state!=1) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
178 System.out.println("Client died "+myId);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
179 break;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
180 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
181 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
182 } catch (InterruptedException e) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
183 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
184 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
185 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
186 };
b32668b8e83c create multicast function
one
parents: 3
diff changeset
187 new Thread(timer).start();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
188 /**
b32668b8e83c create multicast function
one
parents: 3
diff changeset
189 * discard all incoming from clients
b32668b8e83c create multicast function
one
parents: 3
diff changeset
190 */
b32668b8e83c create multicast function
one
parents: 3
diff changeset
191 final Runnable reader = new Runnable() {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
192 public void run() {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
193 byte b[] = new byte[4096];
b32668b8e83c create multicast function
one
parents: 3
diff changeset
194 for(;;) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
195 try {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
196 int c = is.read(b);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
197 if (c<=0) throw new IOException();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
198 // System.out.println("client read "+c);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
199 } catch (IOException e) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
200 try {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
201 writerRunning.set(2);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
202 os.close();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
203 is.close();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
204 } catch (IOException e1) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
205 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
206 return;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
207 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
208 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
209 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
210 };
b32668b8e83c create multicast function
one
parents: 3
diff changeset
211 /**
b32668b8e83c create multicast function
one
parents: 3
diff changeset
212 * send packets to a client
b32668b8e83c create multicast function
one
parents: 3
diff changeset
213 */
b32668b8e83c create multicast function
one
parents: 3
diff changeset
214 Runnable sender = new Runnable() {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
215 public void run() {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
216 writerRunning.set(1);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
217 try {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
218 /**
b32668b8e83c create multicast function
one
parents: 3
diff changeset
219 * initial connection of RFB protocol
b32668b8e83c create multicast function
one
parents: 3
diff changeset
220 */
b32668b8e83c create multicast function
one
parents: 3
diff changeset
221 sendRfbVersion(os);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
222 // readVersionMsg(is);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
223 int rfbMinor = readVersionMsg(is,os);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
224 sendSecurityType(os);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
225 readSecType(is);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
226 sendSecResult(os);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
227 readClientInit(is);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
228 sendInitData(os);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
229 new Thread(reader).start(); // discard incoming packet here after.
b32668b8e83c create multicast function
one
parents: 3
diff changeset
230 if(rfbMinor == 855){
b32668b8e83c create multicast function
one
parents: 3
diff changeset
231 //checkDilay(os);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
232 // send jpeg data of full screen.
b32668b8e83c create multicast function
one
parents: 3
diff changeset
233 // sendFullScreen("jpeg" ,os);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
234 } else {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
235 // send raw data of full screen.
b32668b8e83c create multicast function
one
parents: 3
diff changeset
236
b32668b8e83c create multicast function
one
parents: 3
diff changeset
237 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
238 for (;;) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
239 LinkedList<ByteBuffer> bufs = c.poll();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
240 int inputIndex = 0;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
241 ByteBuffer header = bufs.get(inputIndex);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
242 if (header==null) continue;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
243 if (header.get(0)==CheckDelay) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
244 System.out.println("--------------------");
b32668b8e83c create multicast function
one
parents: 3
diff changeset
245 //writeToClient(os, bufs, inputIndex);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
246 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
247 if (header.get(0)==FramebufferUpdate) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
248 // System.out.println("client "+ myId);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
249 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
250 writeToClient(os, bufs, inputIndex);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
251 writerRunning.set(1); // yes my client is awaking.
b32668b8e83c create multicast function
one
parents: 3
diff changeset
252 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
253 } catch (IOException e) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
254 try {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
255 writerRunning.set(2);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
256 os.close();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
257 } catch (IOException e1) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
258 /* if socket closed cliList.remove(newCli); */
b32668b8e83c create multicast function
one
parents: 3
diff changeset
259 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
260 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
261 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
262
b32668b8e83c create multicast function
one
parents: 3
diff changeset
263 public void writeToClient(final OutputStream os,
b32668b8e83c create multicast function
one
parents: 3
diff changeset
264 LinkedList<ByteBuffer> bufs, int inputIndex)
b32668b8e83c create multicast function
one
parents: 3
diff changeset
265 throws IOException {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
266 while(inputIndex < bufs.size()) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
267 ByteBuffer b = bufs.get(inputIndex++);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
268 os.write(b.array(), b.position(), b.limit());
b32668b8e83c create multicast function
one
parents: 3
diff changeset
269 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
270 os.flush();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
271 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
272 };
b32668b8e83c create multicast function
one
parents: 3
diff changeset
273 clients++;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
274 new Thread(sender).start();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
275
b32668b8e83c create multicast function
one
parents: 3
diff changeset
276 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
277
b32668b8e83c create multicast function
one
parents: 3
diff changeset
278 void sendRfbVersion(OutputStream os) throws IOException {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
279 os.write(versionMsg_3_855.getBytes());
b32668b8e83c create multicast function
one
parents: 3
diff changeset
280 // os.write(versionMsg_3_8.getBytes());
b32668b8e83c create multicast function
one
parents: 3
diff changeset
281 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
282 int readVersionMsg(InputStream is, OutputStream os) throws IOException {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
283
b32668b8e83c create multicast function
one
parents: 3
diff changeset
284 byte[] b = new byte[12];
b32668b8e83c create multicast function
one
parents: 3
diff changeset
285
b32668b8e83c create multicast function
one
parents: 3
diff changeset
286 is.read(b);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
287
b32668b8e83c create multicast function
one
parents: 3
diff changeset
288 if ((b[0] != 'R') || (b[1] != 'F') || (b[2] != 'B') || (b[3] != ' ')
b32668b8e83c create multicast function
one
parents: 3
diff changeset
289 || (b[4] < '0') || (b[4] > '9') || (b[5] < '0') || (b[5] > '9')
b32668b8e83c create multicast function
one
parents: 3
diff changeset
290 || (b[6] < '0') || (b[6] > '9') || (b[7] != '.')
b32668b8e83c create multicast function
one
parents: 3
diff changeset
291 || (b[8] < '0') || (b[8] > '9') || (b[9] < '0') || (b[9] > '9')
b32668b8e83c create multicast function
one
parents: 3
diff changeset
292 || (b[10] < '0') || (b[10] > '9') || (b[11] != '\n')) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
293 throw new IOException("Host " + host + " port " + port
b32668b8e83c create multicast function
one
parents: 3
diff changeset
294 + " is not an RFB server");
b32668b8e83c create multicast function
one
parents: 3
diff changeset
295 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
296
b32668b8e83c create multicast function
one
parents: 3
diff changeset
297 int rfbMajor = (b[4] - '0') * 100 + (b[5] - '0') * 10 + (b[6] - '0');
b32668b8e83c create multicast function
one
parents: 3
diff changeset
298 int rfbMinor = (b[8] - '0') * 100 + (b[9] - '0') * 10 + (b[10] - '0');
b32668b8e83c create multicast function
one
parents: 3
diff changeset
299
b32668b8e83c create multicast function
one
parents: 3
diff changeset
300 if (rfbMajor < 3) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
301 throw new IOException(
b32668b8e83c create multicast function
one
parents: 3
diff changeset
302 "RFB server does not support protocol version 3");
b32668b8e83c create multicast function
one
parents: 3
diff changeset
303 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
304
b32668b8e83c create multicast function
one
parents: 3
diff changeset
305 if (rfbMinor == 855) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
306 sendProxyFlag(os);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
307 // if(proxyFlag)sendPortNumber(os);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
308 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
309 return rfbMinor;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
310
b32668b8e83c create multicast function
one
parents: 3
diff changeset
311 } void readVersionMsg(InputStream is) throws IOException {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
312
b32668b8e83c create multicast function
one
parents: 3
diff changeset
313 byte[] b = new byte[12];
b32668b8e83c create multicast function
one
parents: 3
diff changeset
314
b32668b8e83c create multicast function
one
parents: 3
diff changeset
315 is.read(b);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
316
b32668b8e83c create multicast function
one
parents: 3
diff changeset
317 if ((b[0] != 'R') || (b[1] != 'F') || (b[2] != 'B') || (b[3] != ' ')
b32668b8e83c create multicast function
one
parents: 3
diff changeset
318 || (b[4] < '0') || (b[4] > '9') || (b[5] < '0') || (b[5] > '9')
b32668b8e83c create multicast function
one
parents: 3
diff changeset
319 || (b[6] < '0') || (b[6] > '9') || (b[7] != '.')
b32668b8e83c create multicast function
one
parents: 3
diff changeset
320 || (b[8] < '0') || (b[8] > '9') || (b[9] < '0') || (b[9] > '9')
b32668b8e83c create multicast function
one
parents: 3
diff changeset
321 || (b[10] < '0') || (b[10] > '9') || (b[11] != '\n')) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
322 throw new IOException("Host " + host + " port " + port
b32668b8e83c create multicast function
one
parents: 3
diff changeset
323 + " is not an RFB server");
b32668b8e83c create multicast function
one
parents: 3
diff changeset
324 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
325
b32668b8e83c create multicast function
one
parents: 3
diff changeset
326 serverMajor = (b[4] - '0') * 100 + (b[5] - '0') * 10 + (b[6] - '0');
b32668b8e83c create multicast function
one
parents: 3
diff changeset
327 serverMinor = (b[8] - '0') * 100 + (b[9] - '0') * 10 + (b[10] - '0');
b32668b8e83c create multicast function
one
parents: 3
diff changeset
328
b32668b8e83c create multicast function
one
parents: 3
diff changeset
329 if (serverMajor < 3) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
330 throw new IOException(
b32668b8e83c create multicast function
one
parents: 3
diff changeset
331 "RFB server does not support protocol version 3");
b32668b8e83c create multicast function
one
parents: 3
diff changeset
332 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
333
b32668b8e83c create multicast function
one
parents: 3
diff changeset
334 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
335
b32668b8e83c create multicast function
one
parents: 3
diff changeset
336 void sendSecurityType(OutputStream os) throws IOException {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
337 // number-of-security-types
b32668b8e83c create multicast function
one
parents: 3
diff changeset
338 os.write(1);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
339 // security-types
b32668b8e83c create multicast function
one
parents: 3
diff changeset
340 // 1:None
b32668b8e83c create multicast function
one
parents: 3
diff changeset
341 os.write(1);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
342 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
343 void readSecType(InputStream is) throws IOException {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
344 byte[] b = new byte[1];
b32668b8e83c create multicast function
one
parents: 3
diff changeset
345 is.read(b);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
346
b32668b8e83c create multicast function
one
parents: 3
diff changeset
347 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
348
b32668b8e83c create multicast function
one
parents: 3
diff changeset
349 void sendSecResult(OutputStream os) throws IOException {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
350 byte[] b = castIntByte(0);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
351 os.write(b);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
352 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
353
b32668b8e83c create multicast function
one
parents: 3
diff changeset
354 void readClientInit(InputStream in) throws IOException {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
355 byte[] b = new byte[0];
b32668b8e83c create multicast function
one
parents: 3
diff changeset
356 in.read(b);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
357 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
358
b32668b8e83c create multicast function
one
parents: 3
diff changeset
359 void sendInitData(OutputStream os) throws IOException {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
360 os.write(initData);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
361 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
362
b32668b8e83c create multicast function
one
parents: 3
diff changeset
363 void sendProxyFlag(OutputStream os) throws IOException {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
364 if(proxyFlag) os.write(1);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
365 else os.write(0);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
366 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
367
b32668b8e83c create multicast function
one
parents: 3
diff changeset
368 byte[] castIntByte(int len) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
369 byte[] b = new byte[4];
b32668b8e83c create multicast function
one
parents: 3
diff changeset
370 b[0] = (byte) ((len >>> 24) & 0xFF);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
371 b[1] = (byte) ((len >>> 16) & 0xFF);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
372 b[2] = (byte) ((len >>> 8) & 0xFF);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
373 b[3] = (byte) ((len >>> 0) & 0xFF);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
374 return b;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
375 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
376
b32668b8e83c create multicast function
one
parents: 3
diff changeset
377 /**
b32668b8e83c create multicast function
one
parents: 3
diff changeset
378 * gzip byte arrays
b32668b8e83c create multicast function
one
parents: 3
diff changeset
379 * @param deflater
b32668b8e83c create multicast function
one
parents: 3
diff changeset
380 * @param inputs
b32668b8e83c create multicast function
one
parents: 3
diff changeset
381 * byte data[]
b32668b8e83c create multicast function
one
parents: 3
diff changeset
382 * @param inputIndex
b32668b8e83c create multicast function
one
parents: 3
diff changeset
383 * @param outputs
b32668b8e83c create multicast function
one
parents: 3
diff changeset
384 * byte data[]
b32668b8e83c create multicast function
one
parents: 3
diff changeset
385 * @return byte length in last byte array
b32668b8e83c create multicast function
one
parents: 3
diff changeset
386 * @throws IOException
b32668b8e83c create multicast function
one
parents: 3
diff changeset
387 */
b32668b8e83c create multicast function
one
parents: 3
diff changeset
388 public int zip(Deflater deflater,LinkedList<ByteBuffer> inputs, int inputIndex, LinkedList<ByteBuffer> outputs) throws IOException {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
389 int len = 0;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
390 ByteBuffer c1= ByteBuffer.allocate(INFLATE_BUFSIZE);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
391 while(inputIndex < inputs.size() ) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
392 ByteBuffer b1 = inputs.get(inputIndex++);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
393 deflater.setInput(b1.array(),b1.position(),b1.remaining());
b32668b8e83c create multicast function
one
parents: 3
diff changeset
394 /**
b32668b8e83c create multicast function
one
parents: 3
diff changeset
395 * If we finish() stream and reset() it, Deflater start new gzip stream, this makes continuous zlib reader unhappy.
b32668b8e83c create multicast function
one
parents: 3
diff changeset
396 * if we remove finish(), Deflater.deflate() never flushes its output. The original zlib deflate has flush flag. I'm pretty
b32668b8e83c create multicast function
one
parents: 3
diff changeset
397 * sure this a kind of bug of Java library.
b32668b8e83c create multicast function
one
parents: 3
diff changeset
398 */
b32668b8e83c create multicast function
one
parents: 3
diff changeset
399 if (inputIndex==inputs.size())
b32668b8e83c create multicast function
one
parents: 3
diff changeset
400 deflater.finish();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
401 int len1 = 0;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
402 do {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
403 len1 = deflater.deflate(c1.array(),c1.position(),c1.remaining());
b32668b8e83c create multicast function
one
parents: 3
diff changeset
404 if (len1>0) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
405 len += len1;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
406 c1.position(c1.position()+len1);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
407 if (c1.remaining()==0) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
408 c1.flip(); outputs.addLast(c1);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
409 c1 = ByteBuffer.allocate(INFLATE_BUFSIZE);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
410 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
411 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
412 } while (len1 >0 || !deflater.needsInput()); // &&!deflater.finished());
b32668b8e83c create multicast function
one
parents: 3
diff changeset
413 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
414 if (c1.position()!=0) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
415 c1.flip(); outputs.addLast(c1);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
416 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
417 deflater.reset();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
418 return len;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
419 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
420
b32668b8e83c create multicast function
one
parents: 3
diff changeset
421
b32668b8e83c create multicast function
one
parents: 3
diff changeset
422 /**
b32668b8e83c create multicast function
one
parents: 3
diff changeset
423 * gunzip byte arrays
b32668b8e83c create multicast function
one
parents: 3
diff changeset
424 * @param inflater
b32668b8e83c create multicast function
one
parents: 3
diff changeset
425 * @param inputs
b32668b8e83c create multicast function
one
parents: 3
diff changeset
426 * byte data[]
b32668b8e83c create multicast function
one
parents: 3
diff changeset
427 * @param outputs
b32668b8e83c create multicast function
one
parents: 3
diff changeset
428 * byte data[]
b32668b8e83c create multicast function
one
parents: 3
diff changeset
429 *@return number of total bytes
b32668b8e83c create multicast function
one
parents: 3
diff changeset
430 * @throws IOException
b32668b8e83c create multicast function
one
parents: 3
diff changeset
431 */
b32668b8e83c create multicast function
one
parents: 3
diff changeset
432 public int unzip(Inflater inflater, LinkedList<ByteBuffer> inputs,
b32668b8e83c create multicast function
one
parents: 3
diff changeset
433 int inputIndex, LinkedList<ByteBuffer> outputs, int bufSize)
b32668b8e83c create multicast function
one
parents: 3
diff changeset
434 throws DataFormatException {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
435 int len = 0;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
436 ByteBuffer buf = ByteBuffer.allocate(bufSize);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
437 while (inputIndex < inputs.size()) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
438 ByteBuffer input = inputs.get(inputIndex++);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
439 inflater.setInput(input.array(), input.position(), input.limit());
b32668b8e83c create multicast function
one
parents: 3
diff changeset
440 // if (inputIndex==inputs.size()) if inflater/deflater has symmetry,
b32668b8e83c create multicast function
one
parents: 3
diff changeset
441 // we need this
b32668b8e83c create multicast function
one
parents: 3
diff changeset
442 // inflater.end(); but this won't work
b32668b8e83c create multicast function
one
parents: 3
diff changeset
443 do {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
444 int len0 = inflater.inflate(buf.array(), buf.position(),
b32668b8e83c create multicast function
one
parents: 3
diff changeset
445 buf.remaining());
b32668b8e83c create multicast function
one
parents: 3
diff changeset
446 if (len0 > 0) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
447 buf.position(buf.position() + len0);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
448 len += len0;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
449 if (buf.remaining() == 0) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
450 buf.flip();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
451 outputs.addLast(buf);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
452 buf = ByteBuffer.allocate(bufSize);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
453 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
454 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
455 } while (!inflater.needsInput());
b32668b8e83c create multicast function
one
parents: 3
diff changeset
456 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
457 if (buf.position() != 0) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
458 buf.flip();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
459 outputs.addLast(buf);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
460 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
461 return len;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
462 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
463
b32668b8e83c create multicast function
one
parents: 3
diff changeset
464 void readSendData(int dataLen,Reader reader) throws IOException, DataFormatException, TransportException {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
465 LinkedList<ByteBuffer>bufs = new LinkedList<ByteBuffer>();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
466 ByteBuffer header = ByteBuffer.allocate(16);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
467 reader.readBytes(header.array(),0,16);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
468 header.limit(16);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
469 if (header.get(0)==FramebufferUpdate) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
470 int encoding = header.getInt(12);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
471 if (encoding==EncodingType.ZRLE.getId()||encoding==EncodingType.ZLIB.getId()) { // ZRLEE is already recompressed
b32668b8e83c create multicast function
one
parents: 3
diff changeset
472 ByteBuffer len = ByteBuffer.allocate(4);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
473 reader.readBytes(len.array(),0,4); len.limit(4);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
474 ByteBuffer inputData = ByteBuffer.allocate(dataLen-20);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
475 reader.readBytes(inputData.array(),0,inputData.capacity()); inputData.limit(dataLen-20);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
476 LinkedList<ByteBuffer>inputs = new LinkedList<ByteBuffer>();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
477 inputs.add(inputData);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
478
b32668b8e83c create multicast function
one
parents: 3
diff changeset
479 header.putInt(12, EncodingType.ZRLEE.getId()); // means recompress every time
b32668b8e83c create multicast function
one
parents: 3
diff changeset
480 // using new Deflecter every time is incompatible with the protocol, clients have to be modified.
b32668b8e83c create multicast function
one
parents: 3
diff changeset
481 Deflater nDeflater = deflater; // new Deflater();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
482 LinkedList<ByteBuffer> out = new LinkedList<ByteBuffer>();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
483 unzip(inflater, inputs, 0 , out, INFLATE_BUFSIZE);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
484 // dump32(inputs);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
485 int len2 = zip(nDeflater, out, 0, bufs);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
486 ByteBuffer blen = ByteBuffer.allocate(4); blen.putInt(len2); blen.flip();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
487 bufs.addFirst(blen);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
488
b32668b8e83c create multicast function
one
parents: 3
diff changeset
489 bufs.addFirst(header);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
490 multicastqueue.put(bufs);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
491 is.reset();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
492 return ;
b32668b8e83c create multicast function
one
parents: 3
diff changeset
493 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
494 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
495 bufs.add(header);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
496 if (dataLen>16) {
b32668b8e83c create multicast function
one
parents: 3
diff changeset
497 ByteBuffer b = ByteBuffer.allocate(dataLen-16);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
498 reader.readBytes(b.array(),0,dataLen-16); b.limit(dataLen-16);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
499 bufs.add(b);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
500 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
501 multicastqueue.put(bufs);
b32668b8e83c create multicast function
one
parents: 3
diff changeset
502 is.reset();
b32668b8e83c create multicast function
one
parents: 3
diff changeset
503
b32668b8e83c create multicast function
one
parents: 3
diff changeset
504 // It may be compressed. We can inflate here to avoid repeating clients decompressing here,
b32668b8e83c create multicast function
one
parents: 3
diff changeset
505 // but it may generate too many large data. It is better to do it in each client.
b32668b8e83c create multicast function
one
parents: 3
diff changeset
506 // But we have do inflation for all input data, so we have to do it here.
b32668b8e83c create multicast function
one
parents: 3
diff changeset
507 }
b32668b8e83c create multicast function
one
parents: 3
diff changeset
508
b32668b8e83c create multicast function
one
parents: 3
diff changeset
509
b32668b8e83c create multicast function
one
parents: 3
diff changeset
510
3
e7ce2b2ffed8 add and modify files
Yu Taninari <you@cr.ie.u-ryukyu.ac.jp>
parents:
diff changeset
511 }