2009-01-12 Gonzalo Paniagua Javier <gonzalo@novell.com>
[mono.git] / 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                 [Test]
232                 public void DeleteFile1 ()
233                 {
234                         ServerDeleteFile sp = new ServerDeleteFile ();
235                         sp.Start ();
236                         string uri = String.Format ("ftp://{0}:{1}/file.txt", sp.IPAddress, sp.Port);
237                         try {
238                                 FtpWebRequest ftp = (FtpWebRequest) WebRequest.Create (uri);
239                                 Console.WriteLine (ftp.RequestUri);
240                                 ftp.KeepAlive = false;
241                                 ftp.Timeout = 5000;
242                                 ftp.Method = WebRequestMethods.Ftp.DeleteFile;
243                                 ftp.UseBinary = true;
244                                 FtpWebResponse response = (FtpWebResponse) ftp.GetResponse ();
245                                 Assert.IsTrue ((int) response.StatusCode >= 200 && (int) response.StatusCode < 300, "DF#01");
246                                 response.Close ();
247                         } catch (Exception e) {
248                                 Console.WriteLine (e);
249                                 if (!String.IsNullOrEmpty (sp.Where))
250                                         throw new Exception (sp.Where);
251                                 throw;
252                         } finally {
253                                 sp.Stop ();
254                         }
255                 }
256
257                 class ServerDeleteFile : FtpServer {
258                         protected override void Run ()
259                         {
260                                 Socket client = control.Accept ();
261                                 NetworkStream ns = new NetworkStream (client, false);
262                                 StreamWriter writer = new StreamWriter (ns, Encoding.ASCII);
263                                 StreamReader reader = new StreamReader (ns, Encoding.UTF8);
264                                 if (!DoAnonymousLogin (writer, reader)) {
265                                         client.Close ();
266                                         return;
267                                 }
268
269                                 if (!DoInitialDialog (writer, reader, "/home/someuser", "/home/someuser/")) {
270                                         client.Close ();
271                                         return;
272                                 }
273
274                                 string str = reader.ReadLine ();
275                                 if (str.Trim () != "DELE file.txt") {
276                                         Where = "DELE - " + str;
277                                         client.Close ();
278                                         return;
279                                 }
280                                 writer.WriteLine ("250 Delete operation successful");
281                                 writer.Flush ();
282                                 if (!EndConversation (writer, reader)) {
283                                         client.Close ();
284                                         return;
285                                 }
286                                 client.Close ();
287                         }
288                 }
289
290                 class ServerDownload : FtpServer {
291                         protected override void Run ()
292                         {
293                                 Socket client = control.Accept ();
294                                 NetworkStream ns = new NetworkStream (client, false);
295                                 StreamWriter writer = new StreamWriter (ns, Encoding.ASCII);
296                                 StreamReader reader = new StreamReader (ns, Encoding.UTF8);
297                                 if (!DoAnonymousLogin (writer, reader)) {
298                                         client.Close ();
299                                         return;
300                                 }
301
302                                 if (!DoInitialDialog (writer, reader, "/home/someuser", "/home/someuser/")) {
303                                         client.Close ();
304                                         return;
305                                 }
306
307                                 string str = reader.ReadLine ();
308                                 if (str != "PASV") {
309                                         Where = "PASV";
310                                         client.Close ();
311                                         return;
312                                 }
313
314                                 IPEndPoint end_data = (IPEndPoint) data.LocalEndPoint;
315                                 byte [] addr_bytes = end_data.Address.GetAddressBytes ();
316                                 byte [] port = new byte [2];
317                                 port[0] = (byte) ((end_data.Port >> 8) & 255);
318                                 port[1] = (byte) (end_data.Port & 255);
319                                 StringBuilder sb = new StringBuilder ("227 Passive (");
320                                 foreach (byte b in addr_bytes) {
321                                         sb.AppendFormat ("{0},", b);    
322                                 }
323                                 sb.AppendFormat ("{0},", port [0]);     
324                                 sb.AppendFormat ("{0})", port [1]);     
325                                 writer.WriteLine (sb.ToString ());
326                                 writer.Flush ();
327
328                                 str = reader.ReadLine ();
329                                 if (str != "RETR file.txt") {
330                                         Where = "RETR - " + str;
331                                         client.Close ();
332                                         return;
333                                 }
334                                 writer.WriteLine ("150 Opening BINARY mode data connection for blah (n bytes)");
335                                 writer.Flush ();
336
337                                 Socket data_cnc = data.Accept ();
338                                 byte [] dontcare = new byte [1];
339                                 data_cnc.Receive (dontcare, 1, SocketFlags.None);
340                                 data_cnc.Close ();
341                                 writer.WriteLine ("226 File send Ok");
342                                 writer.Flush ();
343                                 if (!EndConversation (writer, reader)) {
344                                         client.Close ();
345                                         return;
346                                 }
347                                 client.Close ();
348                         }
349                 }
350
351                 class ServerPut : FtpServer {
352                         protected override void Run ()
353                         {
354                                 Socket client = control.Accept ();
355                                 NetworkStream ns = new NetworkStream (client, false);
356                                 StreamWriter writer = new StreamWriter (ns, Encoding.ASCII);
357                                 StreamReader reader = new StreamReader (ns, Encoding.UTF8);
358                                 if (!DoAnonymousLogin (writer, reader)) {
359                                         client.Close ();
360                                         return;
361                                 }
362
363                                 if (!DoInitialDialog (writer, reader, "/home/someuser", "/home/someuser/uploads/")) {
364                                         client.Close ();
365                                         return;
366                                 }
367
368                                 string str = reader.ReadLine ();
369                                 if (str != "PASV") {
370                                         Where = "PASV";
371                                         client.Close ();
372                                         return;
373                                 }
374
375                                 IPEndPoint end_data = (IPEndPoint) data.LocalEndPoint;
376                                 byte [] addr_bytes = end_data.Address.GetAddressBytes ();
377                                 byte [] port = new byte [2];
378                                 port[0] = (byte) ((end_data.Port >> 8) & 255);
379                                 port[1] = (byte) (end_data.Port & 255);
380                                 StringBuilder sb = new StringBuilder ("227 Passive (");
381                                 foreach (byte b in addr_bytes) {
382                                         sb.AppendFormat ("{0},", b);    
383                                 }
384                                 sb.AppendFormat ("{0},", port [0]);     
385                                 sb.AppendFormat ("{0})", port [1]);     
386                                 writer.WriteLine (sb.ToString ());
387                                 writer.Flush ();
388
389                                 str = reader.ReadLine ();
390                                 if (str != "STOR file.txt") {
391                                         Where = "STOR - " + str;
392                                         client.Close ();
393                                         return;
394                                 }
395                                 writer.WriteLine ("150 Ok to send data");
396                                 writer.Flush ();
397
398                                 Socket data_cnc = data.Accept ();
399                                 byte [] dontcare = new byte [1];
400                                 data_cnc.Receive (dontcare, 1, SocketFlags.None);
401                                 data_cnc.Close ();
402                                 writer.WriteLine ("226 File received Ok");
403                                 writer.Flush ();
404                                 if (!EndConversation (writer, reader)) {
405                                         client.Close ();
406                                         return;
407                                 }
408                                 client.Close ();
409                         }
410                 }
411
412                 abstract class FtpServer {
413                         protected Socket control;
414                         protected Socket data;
415                         protected ManualResetEvent evt;
416                         public string Where = "";
417
418                         public FtpServer ()
419                         {
420                                 control = new Socket (AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
421                                 control.Bind (new IPEndPoint (IPAddress.Loopback, 0));
422                                 control.Listen (1);
423                                 data = new Socket (AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
424                                 data.Bind (new IPEndPoint (IPAddress.Loopback, 0));
425                                 data.Listen (1);
426                         }
427
428                         public void Start ()
429                         {
430                                 evt = new ManualResetEvent (false);
431                                 Thread th = new Thread (new ThreadStart (Run));
432                                 th.Start ();
433                         }
434
435                         public void Stop ()
436                         {
437                                 evt.Set ();
438                                 data.Close ();
439                                 control.Close ();
440                         }
441
442                         // PWD, CWD and TYPE I (type could be moved out of here)
443                         protected bool DoInitialDialog (StreamWriter writer, StreamReader reader, string pwd, string cwd)
444                         {
445                                 string str = reader.ReadLine ();
446                                 if (!str.StartsWith ("OPTS utf8 on")) {
447                                         Where = "OPTS utf8 - " + str;
448                                         return false;
449                                 }
450                                 writer.WriteLine ("200 Always in UTF8 mode"); // vsftpd
451                                 writer.Flush ();
452                                 str = reader.ReadLine ();
453                                 if (!str.StartsWith ("PWD")) {
454                                         Where = "PWD - " + str;
455                                         return false;
456                                 }
457                                 writer.WriteLine ("257 \"{0}\"", pwd);
458                                 writer.Flush ();
459                                 str = reader.ReadLine ();
460                                 if (str != ("CWD " + cwd)) {
461                                         Where = "CWD - " + str;
462                                         return false;
463                                 }
464                                 writer.WriteLine ("250 Directory changed");
465                                 writer.Flush ();
466                                 str = reader.ReadLine ();
467                                 if (str != ("TYPE I")) {
468                                         Where = "TYPE - " + str;
469                                         return false;
470                                 }
471                                 writer.WriteLine ("200 Switching to binary mode");
472                                 writer.Flush ();
473                                 return true;
474                         }
475
476                         protected bool EndConversation (StreamWriter writer, StreamReader reader)
477                         {
478                                 string str = reader.ReadLine ();
479                                 if (str != "QUIT") {
480                                         Where = "QUIT";
481                                         return false;
482                                 }
483                                 writer.WriteLine ("220 Bye");
484                                 writer.Flush ();
485                                 Thread.Sleep (250);
486                                 return true;
487                         }
488
489                         protected bool DoAnonymousLogin (StreamWriter writer, StreamReader reader)
490                         {
491                                 writer.WriteLine ("220 Welcome to the jungle");
492                                 writer.Flush ();
493                                 string str = reader.ReadLine ();
494                                 if (!str.StartsWith ("USER ")) {
495                                         Where = "USER";
496                                         return false;
497                                 }
498                                 writer.WriteLine ("331 Say 'Mellon'");
499                                 writer.Flush ();
500                                 str = reader.ReadLine ();
501                                 if (!str.StartsWith ("PASS ")) {
502                                         Where = "PASS";
503                                         return false;
504                                 }
505                                 writer.WriteLine ("230 Logged in");
506                                 writer.Flush ();
507                                 return true;
508                         }
509                         
510                         public IPAddress IPAddress {
511                                 get { return ((IPEndPoint) control.LocalEndPoint).Address; }
512                         }
513                         
514                         public int Port {
515                                 get { return ((IPEndPoint) control.LocalEndPoint).Port; }
516                         }
517
518                         protected abstract void Run ();
519                 }
520         }
521 }
522
523 #endif
524