5 |
|
import uk.ac.ukc.iscream.core.*; |
6 |
|
import uk.ac.ukc.iscream.filter.*; |
7 |
|
import uk.ac.ukc.iscream.util.*; |
8 |
< |
import java.net.Socket; |
8 |
> |
import uk.ac.ukc.iscream.componentmanager.*; |
9 |
> |
import java.net.*; |
10 |
|
import java.io.*; |
11 |
+ |
import java.util.*; |
12 |
|
|
13 |
|
/** |
14 |
|
* Handles setting up a host. |
31 |
|
|
32 |
|
//---CONSTRUCTORS--- |
33 |
|
|
34 |
+ |
/** |
35 |
+ |
* Construct a new HostInit. |
36 |
+ |
* |
37 |
+ |
* @param socket The socket to which the host is connected |
38 |
+ |
*/ |
39 |
|
public HostInit(Socket socket) throws IOException { |
40 |
|
_socket = socket; |
41 |
+ |
// setup reader & writer |
42 |
|
_socketIn = new BufferedReader(new InputStreamReader(_socket.getInputStream())); |
43 |
< |
_socketOut = new PrintWriter(_socket.getOutputStream()); |
43 |
> |
_socketOut = new PrintWriter(_socket.getOutputStream(), true); |
44 |
|
_logger.write(toString(), Logger.SYSINIT, "created"); |
45 |
|
} |
46 |
|
|
47 |
|
//---PUBLIC METHODS--- |
48 |
|
|
49 |
+ |
/** |
50 |
+ |
* Main method in this class, which handles communicating with |
51 |
+ |
* the host to determine it's setup. |
52 |
+ |
*/ |
53 |
|
public void run() { |
54 |
|
try { |
55 |
|
String inBound = _socketIn.readLine(); |
56 |
|
if (!inBound.equals("STARTCONFIG")) { |
57 |
|
_socketOut.println("ERROR"); |
46 |
– |
_socketOut.flush(); |
58 |
|
throw new IOException("protocol error - expected:STARTCONFIG got:" + inBound); |
59 |
|
} |
60 |
|
|
64 |
|
throw new IOException("error in host configuration"); |
65 |
|
} else { |
66 |
|
_socketOut.println("OK"); |
56 |
– |
_socketOut.flush(); |
67 |
|
|
68 |
|
// get lastmodified |
69 |
|
inBound = _socketIn.readLine(); |
70 |
|
if(!inBound.equals("LASTMODIFIED")) { |
71 |
|
// protocol error |
72 |
|
_socketOut.println("ERROR"); |
63 |
– |
_socketOut.flush(); |
73 |
|
throw new IOException("protocol error - expected:LASTMODIFIED got:" + inBound); |
74 |
|
} |
75 |
|
else { |
76 |
|
// send info |
77 |
|
_socketOut.println(myConfig.getLastModified()); |
69 |
– |
_socketOut.flush(); |
78 |
|
} |
79 |
|
|
80 |
|
// get config fileList |
82 |
|
if(!inBound.equals("FILELIST")) { |
83 |
|
// protocol error |
84 |
|
_socketOut.println("ERROR"); |
77 |
– |
_socketOut.flush(); |
85 |
|
throw new IOException("protocol error - expected:FILELIST got:" + inBound); |
86 |
|
} |
87 |
|
else { |
88 |
|
// send info |
89 |
|
_socketOut.println(myConfig.getFileList()); |
83 |
– |
_socketOut.flush(); |
90 |
|
} |
91 |
< |
|
91 |
> |
|
92 |
> |
// send the FQDN to the host |
93 |
> |
inBound = _socketIn.readLine(); |
94 |
> |
if(!inBound.equals("FQDN")) { |
95 |
> |
// protocol error |
96 |
> |
_socketOut.println("ERROR"); |
97 |
> |
throw new IOException("protocol error - expected:FQDN got:" + inBound); |
98 |
> |
} |
99 |
> |
else { |
100 |
> |
// send the fqdn (of the host) to the host |
101 |
> |
_socketOut.println(_socket.getInetAddress().getHostName().toLowerCase()); |
102 |
> |
} |
103 |
> |
|
104 |
|
// get properties |
105 |
|
inBound = _socketIn.readLine(); |
106 |
|
while(!inBound.equals("ENDCONFIG")) { |
110 |
|
String returnedProperty = myConfig.getProperty("Host."+inBound); |
111 |
|
|
112 |
|
_socketOut.println(returnedProperty); |
95 |
– |
_socketOut.flush(); |
113 |
|
|
114 |
|
} catch (org.omg.CORBA.MARSHAL e) { |
115 |
|
_socketOut.println("ERROR"); |
99 |
– |
_socketOut.flush(); |
116 |
|
} |
117 |
|
inBound = _socketIn.readLine(); |
118 |
|
} |
119 |
|
_logger.write(toString(), Logger.SYSMSG, "configured host"); |
120 |
|
_socketOut.println("OK"); |
105 |
– |
_socketOut.flush(); |
121 |
|
|
122 |
|
// get filter reference |
123 |
|
inBound = _socketIn.readLine(); |
124 |
|
if(!inBound.equals("FILTER")) { |
125 |
|
// protocol error |
126 |
|
_socketOut.println("ERROR"); |
112 |
– |
_socketOut.flush(); |
127 |
|
throw new IOException("protocol error - expected:FILTER got:" + inBound); |
128 |
|
} |
129 |
|
else { |
130 |
|
// send info |
131 |
< |
String parentFilter = myConfig.getProperty("Host.filter"); |
132 |
< |
_logger.write(toString(), Logger.DEBUG, " looking for parent - " + parentFilter); |
133 |
< |
Filter filter = FilterHelper.narrow(ReferenceManager.getInstance().getCORBARef("iscream.Filter." + parentFilter)); |
134 |
< |
_socketOut.println(filter.getHostName() + ";" |
135 |
< |
+ filter.getUDPPort() + ";" |
136 |
< |
+ filter.getTCPPort()); |
137 |
< |
_socketOut.flush(); |
131 |
> |
String filterList = myConfig.getProperty("Host.filter"); |
132 |
> |
Filter filterRef = null; |
133 |
> |
String filter = null; |
134 |
> |
StringTokenizer st = new StringTokenizer(filterList, ";"); |
135 |
> |
while (filterRef==null && st.hasMoreTokens()) { |
136 |
> |
filter = st.nextToken(); |
137 |
> |
_logger.write(toString(), Logger.DEBUG, " looking for filter- " + filter); |
138 |
> |
try { |
139 |
> |
filterRef = FilterHelper.narrow(ReferenceManager.getInstance().getCORBARef("iscream.Filter." + filter)); |
140 |
> |
} catch (ComponentCORBAException e) { |
141 |
> |
_logger.write(toString(), Logger.DEBUG, " unable to find filter- " + filter); |
142 |
> |
} |
143 |
> |
} |
144 |
> |
|
145 |
> |
// hopefully we found a filter |
146 |
> |
if(filterRef != null) { |
147 |
> |
_logger.write(toString(), Logger.DEBUG, " found filter- " + filter); |
148 |
> |
// tell the host about it... |
149 |
> |
_socketOut.println(filterRef.getHostName() + ";" |
150 |
> |
+ filterRef.getUDPPort() + ";" |
151 |
> |
+ filterRef.getTCPPort()); |
152 |
> |
} |
153 |
> |
else { |
154 |
> |
// ...or throw a wobbly |
155 |
> |
throw new IOException("unable to find filter for host"); |
156 |
> |
} |
157 |
|
} |
158 |
|
|
159 |
|
// confirm that all is ok |
161 |
|
if(!inBound.equals("END")) { |
162 |
|
// protocol error |
163 |
|
_socketOut.println("ERROR"); |
131 |
– |
_socketOut.flush(); |
164 |
|
throw new IOException("protocol error - expected:END got:" + inBound); |
165 |
|
} |
166 |
|
else { |
167 |
|
// send ok |
168 |
|
_socketOut.println("OK"); |
137 |
– |
_socketOut.flush(); |
169 |
|
} |
170 |
|
|
171 |
|
} |
172 |
|
|
173 |
|
} catch (Exception e) { |
174 |
< |
_logger.write(toString(), Logger.ERROR, "ERROR - " + e.getMessage()); |
174 |
> |
_logger.write(toString(), Logger.ERROR, "ERROR - " + e); |
175 |
|
} |
176 |
|
|
177 |
|
_socketOut.flush(); |