public static void main(String[] args) throws Exception {
boolean success = false;
try {
ServerSocket sock;
sock = new ServerSocket(2333);
sock.setSoTimeout(2);
sock.accept();
} catch (InterruptedIOException e) {
success = true;
}
if (!success) throw new RuntimeException("Socket timeout failure.");
}
/**
* Begins the thread that waits for new clients. If the server is already in listening mode, this
* call has no effect.
*
* @exception IOException if an I/O error occurs when creating the server socket.
*/
public final void listen() throws IOException {
if (!isListening()) {
if (serverSocket == null) {
serverSocket = new ServerSocket(getPort(), backlog);
}
serverSocket.setSoTimeout(timeout);
connectionListener = new Thread(this);
connectionListener.start();
}
}
public void run() {
try {
s1 = s.accept();
is = s1.getInputStream();
os = s1.getOutputStream();
is.read();
String reply = reply1 + port + reply2;
os.write(reply.getBytes());
/* wait for redirected connection */
s.setSoTimeout(5000);
s1 = s.accept();
os = s1.getOutputStream();
os.write(reply3.getBytes());
} catch (Exception e) {
/* Just need thread to terminate */
}
}
/** Accept a connection from a debuggee and handshake with it. */
public Connection accept(ListenKey listener, long acceptTimeout, long handshakeTimeout)
throws IOException {
if (acceptTimeout < 0 || handshakeTimeout < 0) {
throw new IllegalArgumentException("timeout is negative");
}
if (!(listener instanceof SocketListenKey)) {
throw new IllegalArgumentException("Invalid listener");
}
ServerSocket ss;
// obtain the ServerSocket from the listener - if the
// socket is closed it means the listener is invalid
synchronized (listener) {
ss = ((SocketListenKey) listener).socket();
if (ss.isClosed()) {
throw new IllegalArgumentException("Invalid listener");
}
}
// from here onwards it's possible that the ServerSocket
// may be closed by a call to stopListening - that's okay
// because the ServerSocket methods will throw an
// IOException indicating the socket is closed.
//
// Additionally, it's possible that another thread calls accept
// with a different accept timeout - that creates a same race
// condition between setting the timeout and calling accept.
// As it is such an unlikely scenario (requires both threads
// to be using the same listener we've chosen to ignore the issue).
ss.setSoTimeout((int) acceptTimeout);
Socket s;
try {
s = ss.accept();
} catch (SocketTimeoutException x) {
throw new TransportTimeoutException("timeout waiting for connection");
}
// handshake here
handshake(s, handshakeTimeout);
return new SocketConnection(s);
}
public boolean makeconnection() {
if (serversoc == null || serversoc.isClosed()) {
try {
serversoc = new ServerSocket(portnum);
serversoc.setSoTimeout(0);
} catch (IOException ex) {
logger.warning(ex.getMessage());
}
}
try {
System.out.println("server waiting");
mysocket = serversoc.accept();
in = mysocket.getInputStream();
out = mysocket.getOutputStream();
oos = new ObjectOutputStream(out);
ois = new ObjectInputStream(in);
System.out.println("connection established");
waitforreceivingdata();
return true;
} catch (SocketException ex) {
logger.warning(ex.getMessage());
return false;
} catch (IOException ex) {
logger.warning(ex.getMessage());
return false;
} catch (Exception ex) {
logger.warning(ex.getMessage());
return false;
}
}
// initiate either a server or a user session
public void run() {
if (isDaemon) {
daemon();
return;
}
;
boolean loggedIn = false;
int i, h1;
String di, str1, user = "******", user_id = "0";
InetAddress localNode;
byte dataBuffer[] = new byte[1024];
String command = null;
StringBuffer statusMessage = new StringBuffer(40);
File targetFile = null;
try {
// start mysql
Class.forName("com.mysql.jdbc.Driver").newInstance();
this.db_conn = DriverManager.getConnection(db_url);
this.db_stmt = this.db_conn.createStatement();
this.db_stmt.executeUpdate("INSERT INTO test_table (name) VALUES ('hello world')");
incoming.setSoTimeout(inactivityTimer); // enforce I/O timeout
remoteNode = incoming.getInetAddress();
localNode = InetAddress.getLocalHost();
BufferedReader in =
new BufferedReader(new InputStreamReader(incoming.getInputStream(), TELNET));
PrintWriter out =
new PrintWriter(new OutputStreamWriter(incoming.getOutputStream(), TELNET), true);
str1 = "220 Flickr FTP Server Ready";
out.println(str1);
if (log) System.out.println(remoteNode.getHostName() + " " + str1);
boolean done = false;
char dataType = 0;
while (!done) {
statusMessage.setLength(0);
// obtain and tokenize command
String str = in.readLine();
if (str == null) break; // EOS reached
i = str.indexOf(' ');
if (i == -1) i = str.length();
command = str.substring(0, i).toUpperCase().intern();
if (log)
System.out.print(
user
+ "@"
+ remoteNode.getHostName()
+ " "
+ (String) ((command != "PASS") ? str : "PASS ***"));
str = str.substring(i).trim();
try {
if (command == "USER") {
user = str;
statusMessage.append("331 Password");
} else if (command == "PASS") {
String pass = str;
String pass_md5 = md5(pass);
this.db_rs =
this.db_stmt.executeQuery(
"SELECT * FROM users WHERE email='"
+ user
+ "' AND password='******'");
if (this.db_rs.first()) {
loggedIn = true;
user_id = this.db_rs.getString("id");
System.out.println("Account id is " + user_id);
}
statusMessage.append(loggedIn ? "230 logged in User" : "530 Login Incorrect");
} else if (!loggedIn) {
statusMessage.append("530 Not logged in");
} else if (command == "RETR") {
statusMessage.append("999 Not likely");
} else if (command == "STOR") {
out.println(BINARY_XFER);
// trim a leading slash off the filename if there is one
if (str.substring(0, 1).equals("/")) str = str.substring(1);
String filename = user_id + "_" + str;
// TODO: sanitise filename
targetFile = new File(upload_root + "/" + filename);
RandomAccessFile dataFile = null;
InputStream inStream = null;
OutputStream outStream = null;
BufferedReader br = null;
PrintWriter pw = null;
try {
int amount;
dataSocket = setupDataLink();
// ensure timeout on reads.
dataSocket.setSoTimeout(inactivityTimer);
dataFile = new RandomAccessFile(targetFile, "rw");
inStream = dataSocket.getInputStream();
while ((amount = inStream.read(dataBuffer)) != -1)
dataFile.write(dataBuffer, 0, amount);
statusMessage.append(XFER_COMPLETE);
shell_exec(ingest_path + " " + user_id + " " + filename);
} finally {
try {
if (inStream != null) inStream.close();
} catch (Exception e1) {
}
;
try {
if (outStream != null) outStream.close();
} catch (Exception e1) {
}
;
try {
if (dataFile != null) dataFile.close();
} catch (Exception e1) {
}
;
try {
if (dataSocket != null) dataSocket.close();
} catch (Exception e1) {
}
;
dataSocket = null;
}
} else if (command == "REST") {
statusMessage.append("502 Sorry, no resuming");
} else if (command == "TYPE") {
if (Character.toUpperCase(str.charAt(0)) == 'I') {
statusMessage.append(COMMAND_OK);
} else {
statusMessage.append("504 Only binary baybee");
}
} else if (command == "DELE"
|| command == "RMD"
|| command == "XRMD"
|| command == "MKD"
|| command == "XMKD"
|| command == "RNFR"
|| command == "RNTO"
|| command == "CDUP"
|| command == "XCDUP"
|| command == "CWD"
|| command == "SIZE"
|| command == "MDTM") {
statusMessage.append("502 None of that malarky!");
} else if (command == "QUIT") {
statusMessage.append(COMMAND_OK).append("GOOD BYE");
done = true;
} else if (command == "PWD" | command == "XPWD") {
statusMessage.append("257 \"/\" is current directory");
} else if (command == "PORT") {
int lng, lng1, lng2, ip2;
String a1 = "", a2 = "";
lng = str.length() - 1;
lng2 = str.lastIndexOf(",");
lng1 = str.lastIndexOf(",", lng2 - 1);
for (i = lng1 + 1; i < lng2; i++) {
a1 = a1 + str.charAt(i);
}
for (i = lng2 + 1; i <= lng; i++) {
a2 = a2 + str.charAt(i);
}
remotePort = Integer.parseInt(a1);
ip2 = Integer.parseInt(a2);
remotePort = (remotePort << 8) + ip2;
statusMessage.append(COMMAND_OK).append(remotePort);
} else if (command == "LIST" | command == "NLST") {
try {
out.println("150 ASCII data");
dataSocket = setupDataLink();
PrintWriter out2 = new PrintWriter(dataSocket.getOutputStream(), true);
if ((command == "NLST")) {
out2.println(".");
out2.println("..");
} else {
out2.println("total 8.0k");
out2.println("dr--r--r-- 1 owner group 213 Aug 26 16:31 .");
out2.println("dr--r--r-- 1 owner group 213 Aug 26 16:31 ..");
}
// socket MUST be closed before signalling EOD
dataSocket.close();
dataSocket = null;
statusMessage.setLength(0);
statusMessage.append(XFER_COMPLETE);
} finally {
try {
if (dataSocket != null) dataSocket.close();
} catch (Exception e) {
}
;
dataSocket = null;
}
} else if (command == "NOOP") {
statusMessage.append(COMMAND_OK);
} else if (command == "SYST") {
statusMessage.append("215 UNIX"); // allows NS to do long dir
} else if (command == "MODE") {
if (Character.toUpperCase(str.charAt(0)) == 'S') {
statusMessage.append(COMMAND_OK);
} else {
statusMessage.append("504");
}
} else if (command == "STRU") {
if (str.equals("F")) {
statusMessage.append(COMMAND_OK);
} else {
statusMessage.append("504");
}
} else if (command == "PASV") {
try {
int num = 0, j = 0;
if (passiveSocket != null)
try {
passiveSocket.close();
} catch (Exception e) {
}
;
passiveSocket = new ServerSocket(0); // any port
// ensure timeout on reads.
passiveSocket.setSoTimeout(inactivityTimer);
statusMessage.append("227 Entering Passive Mode (");
String s = localNode.getHostAddress().replace('.', ','); // get host #
statusMessage.append(s).append(',');
num = passiveSocket.getLocalPort(); // get port #
j = (num >> 8) & 0xff;
statusMessage.append(j);
statusMessage.append(',');
j = num & 0xff;
statusMessage.append(j);
statusMessage.append(')');
} catch (Exception e) {
try {
if (passiveSocket != null) passiveSocket.close();
} catch (Exception e1) {
}
;
passiveSocket = null;
throw e;
}
} else {
statusMessage.append("502 unimplemented ").append(command);
}
}
// shutdown causes an interruption to be thrown
catch (InterruptedException e) {
throw (e);
} catch (Exception e) // catch all for any errors (including files)
{
statusMessage.append(FAULT).append(e.getMessage());
if (debug) {
System.out.println("\nFAULT - lastfile " + targetFile);
e.printStackTrace();
}
;
}
// send result status to remote
out.println(statusMessage);
if (log) System.out.println("\t" + statusMessage);
}
} catch (Exception e) // usually network errors (including timeout)
{
if (log) System.out.println("forced instance exit " + e);
if (debug) e.printStackTrace();
} finally // exiting server instance
{
// tear down mysql
if (this.db_rs != null) {
try {
this.db_rs.close();
} catch (SQLException SQLE) {;
}
}
if (this.db_stmt != null) {
try {
this.db_stmt.close();
} catch (SQLException SQLE) {;
}
}
if (this.db_pstmt != null) {
try {
this.db_pstmt.close();
} catch (SQLException SQLE) {;
}
}
if (this.db_conn != null) {
try {
this.db_conn.close();
} catch (SQLException SQLE) {;
}
}
forceClose();
}
}