client/gui/src/XylFTPConnection.java

   1 //////////////////////////////////////////////////////////////////////////
   2 // Copyright (C) Zhou Xiao-wei, Apr. 2007.
   3 // Copyright (C) Wang Cong, Apr. 2007.
   4 // See AUTHORS and CREDITS to learn more.
   5 // This program is free software; you can redistribute it and/or modify
   6 // it under the terms of the GNU General Public License version 2 as
   7 // published by the Free Software Foundation.
   8 // See COPYING to learn more.
   9 //////////////////////////////////////////////////////////////////////////
  10
  11 import java.awt.Container;
  12 import java.net.*;
  13 import java.io.*;
  14
  15 /**
  16  * This class is for FTP connection.
  17  * Extends XylTelnetConnection.
  18  * @author WANG Cong, Zhou Xiao-Wei
  19  * @version 1.14
  20  */
  21 public class XylFTPConnection extends XylTelnetConnection{
  22
  23     /**
  24      *The size of reader's buffer.
  25      */
  26     private final int RDBUF_SIZE = 1024;
  27
  28     /**
  29      *The size of writer's buffer.
  30      */
  31     private final int WRBUF_SIZE = 1024;
  32
  33     /**
  34      *The status of FTP connection.
  35      */
  36     private int Status;
  37     // 0 for not connected
  38     // 1 for connected but not login
  39     // 2 for login and no data transfer
  40     // 3 for login and getting data down
  41     // 4 for login and putting data up
  42
  43     /**
  44      *Presents the mode (active or passive) of FTP connection.
  45      */
  46     private int Mode;
  47     //1 for passive, 0 for active, default to 1. ;)
  48
  49     /**
  50      *Constructs a LocalFile for data storage.
  51      */
  52     private String LocalFile;
  53     // Empty string for this means stdout. ;)
  54
  55     /**
  56      *The data connection socket.
  57      */
  58     private Socket DataSocket;
  59
  60     /**
  61      *The sever data socket, used by active mode.
  62      */
  63     private ServerSocket DataServerSocket;
  64
  65     /**
  66      *The input stream.
  67      */
  68     private BufferedInputStream FTPInStream;
  69
  70     /**
  71      *The output stream.
  72      */
  73     private BufferedOutputStream FTPOutStream;
  74
  75     /**
  76      *Initializes some values.
  77      */
  78     XylFTPConnection(){
  79         super();
  80         Status = 0;
  81         Mode = 1;
  82         LocalFile = null;
  83         DataSocket = null;
  84         DataServerSocket = null;
  85     }
  86
  87     /**
  88      *Gets status of current FTP connection.
  89      *@return the status of connection
  90      */
  91     public int GetStatus(){
  92         return Status;
  93     }
  94
  95     /**
  96      *Sets status of the connection.
  97      *@param NewStatus
  98      */
  99     public void SetStatus(int NewStatus){
 100         Status = NewStatus;
 101     }
 102
 103     /**
 104      *Sets the LocalFile name.
 105      *@param FileName
 106      */
 107     public void SetLocalFile(String FileName){
 108         LocalFile = FileName;
 109     }
 110
 111     /**
 112      *Tests whether the mode is passive
 113      */
 114     public boolean IsPassive(){
 115         if (Mode==1)
 116             return true;
 117         else
 118             return false;
 119     }
 120     /**
 121      *Sets the passive mode.
 122      */
 123     public void SetPassive(){
 124         Mode = 1;
 125     }
 126
 127     /**
 128      *Sets the active mode.
 129      */
 130     public void SetActive(){
 131         Mode = 0;
 132     }
 133
 134     /**
 135      *Opens the data connection in active mode.
 136      *@see Exception
 137      *@throws Exception on opening dataconnection error
 138      */
 139     public void OpenDataConnection() throws Exception{
 140         try {
 141             DataSocket = DataServerSocket.accept();
 142         } catch(Exception e) {
 143             SetStatus(2);
 144             throw e;
 145         }
 146     }
 147
 148     public void ListenForDataConnection() throws Exception{
 149         try {
 150             if(-1 == ReturnSelfPort())
 151                 throw new XylFTPException("Port is unavailable.");
 152             DataServerSocket = new ServerSocket(ReturnSelfPort());
 153             DataServerSocket.setSoTimeout(Timeout);
 154         } catch(Exception e) {
 155             SetStatus(2);
 156             throw e;
 157         }
 158     }
 159
 160     /**
 161      *Opens the data connection in passive mode.
 162      *@param Port the port of server
 163      *@param Host the IP of server
 164      *@see Exception
 165      *@throws Exception on opening DataConnection error
 166      */
 167     public void OpenDataConnection(String Host, int Port) throws Exception{
 168         try {
 169             DataSocket = new Socket(Host, Port);
 170             DataSocket.setSoTimeout(Timeout);
 171         } catch (Exception e) {
 172             SetStatus(2);
 173             throw e;
 174         }
 175     }
 176
 177     /**
 178      *Closes the data connection.
 179      *@see XylFTPException
 180      *@throws XylFTPException on closing DataConnection error
 181      */
 182     public void CloseDataConnection() throws Exception{
 183         if(DataSocket != null) {
 184             DataSocket.close();
 185             DataSocket = null;
 186         }
 187         if(DataServerSocket != null) {
 188             DataServerSocket.close();
 189             DataServerSocket = null;
 190         }
 191     }
 192
 193     /**
 194      *Uploads a file in active mode.
 195      *@throws XylFTPException on sending file error
 196      *@see XylFTPException
 197      */
 198     public void SendFileActive() throws Exception{
 199         int bytesRead;
 200         byte [] buf = new byte[WRBUF_SIZE];
 201         if (Mode!=0) {
 202             SetStatus(2);
 203             throw new XylFTPException("You are not in active mode!");
 204         }
 205         if (LocalFile.equals("")) {
 206             SetStatus(2);
 207             throw new XylFTPException("Can't send unnamed file!");
 208         } else {
 209             FTPInStream = new BufferedInputStream(new FileInputStream(LocalFile));
 210         }
 211         FTPOutStream = new BufferedOutputStream(DataSocket.getOutputStream());
 212         while ((bytesRead = FTPInStream.read(buf, 0, WRBUF_SIZE)) > 0) {
 213             FTPOutStream.write(buf, 0, bytesRead);
 214         }
 215         FTPOutStream.flush();
 216         FTPOutStream.close();
 217         FTPInStream.close();
 218     }
 219     /**
 220      *Uploads a file in passive mode.
 221      *@throws XylFTPException on sending file error
 222      *@see XylFTPException
 223      */
 224     public void SendFilePassive() throws Exception{
 225         int bytesRead;
 226         byte [] buf = new byte[WRBUF_SIZE];
 227         if (Mode!=1) {
 228             SetStatus(2);
 229             throw new XylFTPException("You are not in passive mode!");
 230         }
 231         if (LocalFile.equals("")) {
 232             SetStatus(2);
 233             throw new XylFTPException("Can't send unnamed file!");
 234         } else {
 235             FTPInStream = new BufferedInputStream(new FileInputStream(LocalFile));
 236         }
 237         FTPOutStream = new BufferedOutputStream(DataSocket.getOutputStream());
 238         while ((bytesRead = FTPInStream.read(buf, 0, WRBUF_SIZE)) > 0) {
 239             FTPOutStream.write(buf, 0, bytesRead);
 240         }
 241         FTPOutStream.flush();
 242         FTPOutStream.close();
 243         FTPInStream.close();
 244     }
 245
 246     /**
 247      *Downloads a file in active mode.
 248      *@throws XylFTPException on getting file error
 249      *@see XylFTPException
 250      */
 251     public void GetFileActive() throws Exception{
 252         int bytesRead;
 253         byte [] buf = new byte[RDBUF_SIZE];
 254         if (Mode!=0) {
 255             SetStatus(2);
 256             throw new XylFTPException("You are not in active mode!");
 257         }
 258         if (LocalFile.equals("")) {
 259             FTPOutStream = new BufferedOutputStream((OutputStream)System.out);
 260         } else {
 261             FTPOutStream = new BufferedOutputStream(new FileOutputStream(LocalFile));
 262         }
 263         FTPInStream = new BufferedInputStream(DataSocket.getInputStream());
 264         while ((bytesRead = FTPInStream.read(buf, 0, RDBUF_SIZE)) > 0) {
 265             FTPOutStream.write(buf, 0, bytesRead);
 266         }
 267         FTPOutStream.flush();
 268         if (!LocalFile.equals(""))      //We can NOT close the stdout.
 269             FTPOutStream.close();
 270         FTPInStream.close();
 271     }
 272
 273     /**
 274      *Downloads a file in passive mode.
 275      *@throws XylFTPException on getting file error
 276      *@see XylFTPException
 277      */
 278     public void GetFilePassive() throws Exception{
 279         int bytesRead;
 280         byte [] buf = new byte[RDBUF_SIZE];
 281         if (Mode!=1) {
 282             SetStatus(2);
 283             throw new XylFTPException("You are not in passive mode!");
 284         }
 285         if (LocalFile.equals("")) {
 286             FTPOutStream = new BufferedOutputStream((OutputStream)System.out);
 287         } else {
 288             FTPOutStream = new BufferedOutputStream(new FileOutputStream(LocalFile));
 289         }
 290         FTPInStream = new BufferedInputStream(DataSocket.getInputStream());
 291         while ((bytesRead = FTPInStream.read(buf, 0, RDBUF_SIZE)) > 0) {
 292             FTPOutStream.write(buf, 0, bytesRead);
 293         }
 294         FTPOutStream.flush();
 295         if (!LocalFile.equals(""))      //We can NOT close the stdout.
 296             FTPOutStream.close();
 297         FTPInStream.close();
 298     }
 299
 300     public String GetListActive() throws Exception{
 301         int bytesRead;
 302         byte [] buf = new byte[RDBUF_SIZE];
 303         if (Mode!=0) {
 304             SetStatus(2);
 305             throw new XylFTPException("You are not in active mode!");
 306         }
 307
 308         ByteArrayOutputStream bArrayStream = new ByteArrayOutputStream();
 309         FTPOutStream = new BufferedOutputStream(bArrayStream);
 310
 311         FTPInStream = new BufferedInputStream(DataSocket.getInputStream());
 312         while ((bytesRead = FTPInStream.read(buf, 0, RDBUF_SIZE)) > 0) {
 313             FTPOutStream.write(buf, 0, bytesRead);
 314         }
 315         FTPOutStream.flush();
 316         FTPOutStream.close();
 317         FTPInStream.close();
 318
 319         return bArrayStream.toString();
 320     }
 321
 322     public String GetListPassive() throws Exception{
 323         int bytesRead;
 324         byte [] buf = new byte[RDBUF_SIZE];
 325         if (Mode!=1) {
 326             SetStatus(2);
 327             throw new XylFTPException("You are not in passive mode!");
 328         }
 329
 330         ByteArrayOutputStream bArrayStream = new ByteArrayOutputStream();
 331         FTPOutStream = new BufferedOutputStream(bArrayStream);
 332
 333         FTPInStream = new BufferedInputStream(DataSocket.getInputStream());
 334         while ((bytesRead = FTPInStream.read(buf, 0, RDBUF_SIZE)) > 0) {
 335             FTPOutStream.write(buf, 0, bytesRead);
 336         }
 337         FTPOutStream.flush();
 338         FTPOutStream.close();
 339         FTPInStream.close();
 340
 341         return bArrayStream.toString();
 342     }
 343 }