mcs/class/System/Test/System.Net/FtpWebRequestTest.cs

   1 //
   2 // FtpWebRequestTest.cs - NUnit Test Cases for System.Net.FtpWebRequest
   3 //
   4 // Authors:
   5 //      Carlos Alberto Cortez <calberto.cortez@gmail.com>
   6 //      Gonzalo Paniagua Javier <gonzalo@novell.com>
   7 //
   8 // Copyright (c) 2006,2007,2008 Novell, Inc. (http://www.novell.com)
   9 //
  10 #if NET_2_0
  11 using NUnit.Framework;
  12 using System;
  13 using System.IO;
  14 using System.Net;
  15 using System.Net.Sockets;
  16 using System.Text;
  17 using System.Threading;
  18
  19 namespace MonoTests.System.Net
  20 {
  21         [TestFixture]
  22         public class FtpWebRequestTest
  23         {
  24                 FtpWebRequest defaultRequest;
  25
  26                 [TestFixtureSetUp]
  27                 public void Init ()
  28                 {
  29                         defaultRequest = (FtpWebRequest) WebRequest.Create ("ftp://www.contoso.com");
  30                 }
  31
  32                 [Test]
  33                 public void ContentLength ()
  34                 {
  35                         try {
  36                                 long l = defaultRequest.ContentLength;
  37                         } catch (NotSupportedException) {
  38                                 Assert.Fail ("#1"); // Not overriden
  39                         }
  40
  41                         try {
  42                                 defaultRequest.ContentLength = 2;
  43                         } catch (NotSupportedException) {
  44                                 Assert.Fail ("#2"); // Not overriden
  45                         }
  46                 }
  47
  48                 [Test]
  49                 public void ContentType ()
  50                 {
  51                         try {
  52                                 string t = defaultRequest.ContentType;
  53                                 Assert.Fail ("#1");
  54                         } catch (NotSupportedException) {
  55                         }
  56
  57                         try {
  58                                 defaultRequest.ContentType = String.Empty;
  59                                 Assert.Fail ("#2");
  60                         } catch (NotSupportedException) {
  61                         }
  62                 }
  63
  64                 [Test]
  65                 public void ContentOffset ()
  66                 {
  67                         try {
  68                                 defaultRequest.ContentOffset = -2;
  69                                 Assert.Fail ("#1");
  70                         } catch (ArgumentOutOfRangeException) {
  71                         }
  72                 }
  73
  74                 [Test]
  75                 public void Credentials ()
  76                 {
  77                         try {
  78                                 defaultRequest.Credentials = null;
  79                                 Assert.Fail ("#1");
  80                         } catch (ArgumentNullException) {
  81                         }
  82
  83                 }
  84
  85                 [Test]
  86                 public void Method ()
  87                 {
  88                         try {
  89                                 defaultRequest.Method = null;
  90                                 Assert.Fail ("#1");
  91                         } catch (ArgumentNullException) {
  92                         }
  93
  94                         try {
  95                                 defaultRequest.Method = String.Empty;
  96                                 Assert.Fail ("#2");
  97                         } catch (ArgumentException) {
  98                         }
  99
 100                         try {
 101                                 defaultRequest.Method = "WrongValue";
 102                                 Assert.Fail ("#3");
 103                         } catch (ArgumentException) {
 104                         }
 105                 }
 106
 107                 [Test]
 108                 public void PreAuthenticate ()
 109                 {
 110                         try {
 111                                 bool p = defaultRequest.PreAuthenticate;
 112                                 Assert.Fail ("#1");
 113                         } catch (NotSupportedException) {
 114                         }
 115
 116                         try {
 117                                 defaultRequest.PreAuthenticate = true;
 118                         } catch (NotSupportedException) {
 119                         }
 120                 }
 121
 122                 [Test]
 123                 public void ReadWriteTimeout ()
 124                 {
 125                         try {
 126                                 defaultRequest.ReadWriteTimeout = -2;
 127                                 Assert.Fail ("#2");
 128                         } catch (ArgumentOutOfRangeException) {
 129                         }
 130                 }
 131
 132                 [Test]
 133                 public void Timeout ()
 134                 {
 135                         try {
 136                                 defaultRequest.Timeout = -2;
 137                                 Assert.Fail ("#2");
 138                         } catch (ArgumentOutOfRangeException) {
 139                         }
 140                 }
 141
 142                 [Test]
 143                 public void DefaultValues ()
 144                 {
 145                         FtpWebRequest request = (FtpWebRequest) WebRequest.Create ("ftp://www.contoso.com");
 146
 147                         Assert.AreEqual (0, request.ContentOffset, "ContentOffset");
 148                         Assert.AreEqual (false, request.EnableSsl, "EnableSsl");
 149                         // FIXME: Disabled this one by now. KeepAlive is not well supported.
 150                         // Assert.AreEqual (true, request.KeepAlive, "KeepAlive");
 151                         Assert.AreEqual (WebRequestMethods.Ftp.DownloadFile, request.Method, "#1");
 152                         Assert.AreEqual (300000, request.ReadWriteTimeout, "ReadWriteTimeout");
 153                         Assert.IsNull (request.RenameTo, "RenameTo");
 154                         Assert.AreEqual (true, request.UseBinary, "UseBinary");
 155                         Assert.AreEqual (100000, request.Timeout, "Timeout");
 156                         Assert.AreEqual (true, request.UsePassive, "UsePassive");
 157                 }
 158
 159                 [Test]
 160                 public void RenameTo ()
 161                 {
 162                         try {
 163                                 defaultRequest.RenameTo = null;
 164                                 Assert.Fail ("#1");
 165                         } catch (ArgumentException) {
 166                         }
 167
 168                         try {
 169                                 defaultRequest.RenameTo = String.Empty;
 170                                 Assert.Fail ("#2");
 171                         } catch (ArgumentException) {
 172                         }
 173                 }
 174
 175                 [Test]
 176                 public void UploadFile1 ()
 177                 {
 178                         ServerPut sp = new ServerPut ();
 179                         sp.Start ();
 180                         string uri = String.Format ("ftp://{0}:{1}/uploads/file.txt", sp.IPAddress, sp.Port);
 181                         try {
 182                                 FtpWebRequest ftp = (FtpWebRequest) WebRequest.Create (uri);
 183                                 ftp.KeepAlive = false;
 184                                 ftp.Timeout = 5000;
 185                                 ftp.Method = WebRequestMethods.Ftp.UploadFile;
 186                                 ftp.ContentLength = 1;
 187                                 ftp.UseBinary = true;
 188                                 Stream stream = ftp.GetRequestStream ();
 189                                 stream.WriteByte (0);
 190                                 stream.Close ();
 191                                 FtpWebResponse response = (FtpWebResponse) ftp.GetResponse ();
 192                                 Assert.IsTrue ((int) response.StatusCode >= 200 && (int) response.StatusCode < 300, "UP#01");
 193                                 response.Close ();
 194                         } catch (Exception) {
 195                                 if (!String.IsNullOrEmpty (sp.Where))
 196                                         throw new Exception (sp.Where);
 197                                 throw;
 198                         } finally {
 199                                 sp.Stop ();
 200                         }
 201                 }
 202
 203                 [Test]
 204                 public void DownloadFile1 ()
 205                 {
 206                         ServerDownload sp = new ServerDownload ();
 207                         sp.Start ();
 208                         string uri = String.Format ("ftp://{0}:{1}/file.txt", sp.IPAddress, sp.Port);
 209                         try {
 210                                 FtpWebRequest ftp = (FtpWebRequest) WebRequest.Create (uri);
 211                                 ftp.KeepAlive = false;
 212                                 ftp.Timeout = 5000;
 213                                 ftp.Method = WebRequestMethods.Ftp.DownloadFile;
 214                                 ftp.UseBinary = true;
 215                                 FtpWebResponse response = (FtpWebResponse) ftp.GetResponse ();
 216                                 Assert.IsTrue ((int) response.StatusCode >= 100 && (int) response.StatusCode < 200, "DL#01");
 217                                 using (Stream st = response.GetResponseStream ()) {
 218                                 }
 219                                 // This should be "220 Bye" or similar (no KeepAlive)
 220                                 Assert.IsTrue ((int) response.StatusCode >= 200 && (int) response.StatusCode < 300, "DL#02");
 221                                 response.Close ();
 222                         } catch (Exception) {
 223                                 if (!String.IsNullOrEmpty (sp.Where))
 224                                         throw new Exception (sp.Where);
 225                                 throw;
 226                         } finally {
 227                                 sp.Stop ();
 228                         }
 229                 }
 230
 231                 class ServerDownload : FtpServer {
 232                         protected override void Run ()
 233                         {
 234                                 Socket client = control.Accept ();
 235                                 NetworkStream ns = new NetworkStream (client, false);
 236                                 StreamWriter writer = new StreamWriter (ns, Encoding.ASCII);
 237                                 StreamReader reader = new StreamReader (ns, Encoding.UTF8);
 238                                 if (!DoAnonymousLogin (writer, reader)) {
 239                                         client.Close ();
 240                                         return;
 241                                 }
 242
 243                                 if (!DoInitialDialog (writer, reader, "/home/someuser", "/home/someuser/")) {
 244                                         client.Close ();
 245                                         return;
 246                                 }
 247
 248                                 string str = reader.ReadLine ();
 249                                 if (str != "PASV") {
 250                                         Where = "PASV";
 251                                         client.Close ();
 252                                         return;
 253                                 }
 254
 255                                 IPEndPoint end_data = (IPEndPoint) data.LocalEndPoint;
 256                                 byte [] addr_bytes = end_data.Address.GetAddressBytes ();
 257                                 byte [] port = new byte [2];
 258                                 port[0] = (byte) ((end_data.Port >> 8) & 255);
 259                                 port[1] = (byte) (end_data.Port & 255);
 260                                 StringBuilder sb = new StringBuilder ("227 Passive (");
 261                                 foreach (byte b in addr_bytes) {
 262                                         sb.AppendFormat ("{0},", b);
 263                                 }
 264                                 sb.AppendFormat ("{0},", port [0]);
 265                                 sb.AppendFormat ("{0})", port [1]);
 266                                 writer.WriteLine (sb.ToString ());
 267                                 writer.Flush ();
 268
 269                                 str = reader.ReadLine ();
 270                                 if (str != "RETR file.txt") {
 271                                         Where = "RETR - " + str;
 272                                         client.Close ();
 273                                         return;
 274                                 }
 275                                 writer.WriteLine ("150 Opening BINARY mode data connection for blah (n bytes)");
 276                                 writer.Flush ();
 277
 278                                 Socket data_cnc = data.Accept ();
 279                                 byte [] dontcare = new byte [1];
 280                                 data_cnc.Receive (dontcare, 1, SocketFlags.None);
 281                                 data_cnc.Close ();
 282                                 writer.WriteLine ("226 File send Ok");
 283                                 writer.Flush ();
 284                                 if (!EndConversation (writer, reader)) {
 285                                         client.Close ();
 286                                         return;
 287                                 }
 288                                 client.Close ();
 289                         }
 290                 }
 291
 292                 class ServerPut : FtpServer {
 293                         protected override void Run ()
 294                         {
 295                                 Socket client = control.Accept ();
 296                                 NetworkStream ns = new NetworkStream (client, false);
 297                                 StreamWriter writer = new StreamWriter (ns, Encoding.ASCII);
 298                                 StreamReader reader = new StreamReader (ns, Encoding.UTF8);
 299                                 if (!DoAnonymousLogin (writer, reader)) {
 300                                         client.Close ();
 301                                         return;
 302                                 }
 303
 304                                 if (!DoInitialDialog (writer, reader, "/home/someuser", "/home/someuser/uploads/")) {
 305                                         client.Close ();
 306                                         return;
 307                                 }
 308
 309                                 string str = reader.ReadLine ();
 310                                 if (str != "PASV") {
 311                                         Where = "PASV";
 312                                         client.Close ();
 313                                         return;
 314                                 }
 315
 316                                 IPEndPoint end_data = (IPEndPoint) data.LocalEndPoint;
 317                                 byte [] addr_bytes = end_data.Address.GetAddressBytes ();
 318                                 byte [] port = new byte [2];
 319                                 port[0] = (byte) ((end_data.Port >> 8) & 255);
 320                                 port[1] = (byte) (end_data.Port & 255);
 321                                 StringBuilder sb = new StringBuilder ("227 Passive (");
 322                                 foreach (byte b in addr_bytes) {
 323                                         sb.AppendFormat ("{0},", b);
 324                                 }
 325                                 sb.AppendFormat ("{0},", port [0]);
 326                                 sb.AppendFormat ("{0})", port [1]);
 327                                 writer.WriteLine (sb.ToString ());
 328                                 writer.Flush ();
 329
 330                                 str = reader.ReadLine ();
 331                                 if (str != "STOR file.txt") {
 332                                         Where = "STOR - " + str;
 333                                         client.Close ();
 334                                         return;
 335                                 }
 336                                 writer.WriteLine ("150 Ok to send data");
 337                                 writer.Flush ();
 338
 339                                 Socket data_cnc = data.Accept ();
 340                                 byte [] dontcare = new byte [1];
 341                                 data_cnc.Receive (dontcare, 1, SocketFlags.None);
 342                                 data_cnc.Close ();
 343                                 writer.WriteLine ("226 File received Ok");
 344                                 writer.Flush ();
 345                                 if (!EndConversation (writer, reader)) {
 346                                         client.Close ();
 347                                         return;
 348                                 }
 349                                 client.Close ();
 350                         }
 351                 }
 352
 353                 abstract class FtpServer {
 354                         protected Socket control;
 355                         protected Socket data;
 356                         protected ManualResetEvent evt;
 357                         public string Where = "";
 358
 359                         public FtpServer ()
 360                         {
 361                                 control = new Socket (AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
 362                                 control.Bind (new IPEndPoint (IPAddress.Loopback, 0));
 363                                 control.Listen (1);
 364                                 data = new Socket (AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
 365                                 data.Bind (new IPEndPoint (IPAddress.Loopback, 0));
 366                                 data.Listen (1);
 367                         }
 368
 369                         public void Start ()
 370                         {
 371                                 evt = new ManualResetEvent (false);
 372                                 Thread th = new Thread (new ThreadStart (Run));
 373                                 th.Start ();
 374                         }
 375
 376                         public void Stop ()
 377                         {
 378                                 evt.Set ();
 379                                 data.Close ();
 380                                 control.Close ();
 381                         }
 382
 383                         // PWD, CWD and TYPE I (type could be moved out of here)
 384                         protected bool DoInitialDialog (StreamWriter writer, StreamReader reader, string pwd, string cwd)
 385                         {
 386                                 string str = reader.ReadLine ();
 387                                 if (!str.StartsWith ("OPTS utf8 on")) {
 388                                         Where = "OPTS utf8 - " + str;
 389                                         return false;
 390                                 }
 391                                 writer.WriteLine ("200 Always in UTF8 mode"); // vsftpd
 392                                 writer.Flush ();
 393                                 str = reader.ReadLine ();
 394                                 if (!str.StartsWith ("PWD")) {
 395                                         Where = "PWD - " + str;
 396                                         return false;
 397                                 }
 398                                 writer.WriteLine ("257 \"{0}\"", pwd);
 399                                 writer.Flush ();
 400                                 str = reader.ReadLine ();
 401                                 if (str != ("CWD " + cwd)) {
 402                                         Where = "CWD - " + str;
 403                                         return false;
 404                                 }
 405                                 writer.WriteLine ("250 Directory changed");
 406                                 writer.Flush ();
 407                                 str = reader.ReadLine ();
 408                                 if (str != ("TYPE I")) {
 409                                         Where = "TYPE - " + str;
 410                                         return false;
 411                                 }
 412                                 writer.WriteLine ("200 Switching to binary mode");
 413                                 writer.Flush ();
 414                                 return true;
 415                         }
 416
 417                         protected bool EndConversation (StreamWriter writer, StreamReader reader)
 418                         {
 419                                 string str = reader.ReadLine ();
 420                                 if (str != "QUIT") {
 421                                         Where = "QUIT";
 422                                         return false;
 423                                 }
 424                                 writer.WriteLine ("220 Bye");
 425                                 writer.Flush ();
 426                                 Thread.Sleep (250);
 427                                 return true;
 428                         }
 429
 430                         protected bool DoAnonymousLogin (StreamWriter writer, StreamReader reader)
 431                         {
 432                                 writer.WriteLine ("220 Welcome to the jungle");
 433                                 writer.Flush ();
 434                                 string str = reader.ReadLine ();
 435                                 if (!str.StartsWith ("USER ")) {
 436                                         Where = "USER";
 437                                         return false;
 438                                 }
 439                                 writer.WriteLine ("331 Say 'Mellon'");
 440                                 writer.Flush ();
 441                                 str = reader.ReadLine ();
 442                                 if (!str.StartsWith ("PASS ")) {
 443                                         Where = "PASS";
 444                                         return false;
 445                                 }
 446                                 writer.WriteLine ("230 Logged in");
 447                                 writer.Flush ();
 448                                 return true;
 449                         }
 450
 451                         public IPAddress IPAddress {
 452                                 get { return ((IPEndPoint) control.LocalEndPoint).Address; }
 453                         }
 454
 455                         public int Port {
 456                                 get { return ((IPEndPoint) control.LocalEndPoint).Port; }
 457                         }
 458
 459                         protected abstract void Run ();
 460                 }
 461         }
 462 }
 463
 464 #endif
 465