ACE/protocols/examples/INet/FTP_Simple_exec.cpp

   1 #include "ace/Get_Opt.h"
   2 #include <memory>
   3 #include "ace/OS_NS_errno.h"
   4 #include "ace/INet/FTP_URL.h"
   5 #include "ace/INet/FTP_ClientRequestHandler.h"
   6 #include <iostream>
   7 #include <fstream>
   8
   9 ACE_CString url;
  10 ACE_CString outfile;
  11 bool do_login = false;
  12 ACE_CString password;
  13 bool do_active = false;
  14
  15 void
  16 usage ()
  17 {
  18   std::cout << "usage: ftp_simple_wget [options] <url>\n";
  19   std::cout << "Executes an FTP download request and sends the result to STDOUT or file\n";
  20   std::cout << "\t-a\t\tuse active mode for data connections\n";
  21   std::cout << "\t-o <filename>\t\tfile to write output to\n";
  22   std::cout << "\t-l\t\task for password at login\n";
  23   std::cout << "\t-p <password>\t\tuse <password> for login\n";
  24 }
  25
  26 bool
  27 parse_args (int argc, ACE_TCHAR *argv [])
  28 {
  29   ACE_Get_Opt get_opt (argc, argv, ACE_TEXT ("ao:lp:h"), 0, 0, ACE_Get_Opt::RETURN_IN_ORDER);
  30
  31   int c;
  32   ACE_CString s;
  33   while ((c = get_opt ()) != EOF)
  34     {
  35       switch (c)
  36         {
  37         case 1:
  38           url = ACE_TEXT_ALWAYS_CHAR (get_opt.opt_arg ());
  39           break;
  40
  41         case 'a':
  42           do_active = true;
  43           break;
  44
  45         case 'o':
  46           outfile = ACE_TEXT_ALWAYS_CHAR (get_opt.opt_arg ());
  47           break;
  48
  49         case 'l':
  50           do_login = true;
  51           break;
  52
  53         case 'p':
  54           password = ACE_TEXT_ALWAYS_CHAR (get_opt.opt_arg ());
  55           break;
  56
  57         case 'h':
  58         default:
  59           usage ();
  60           return false;
  61         }
  62     }
  63
  64   return true;
  65 }
  66
  67 class My_Authenticator
  68   : public ACE::INet::AuthenticatorBase
  69 {
  70   public:
  71     My_Authenticator () {}
  72     ~My_Authenticator () {}
  73
  74     virtual bool authenticate(ACE::INet::AuthenticationBase& auth) const
  75       {
  76         if (!password.empty ()) auth.password (password);
  77         if (do_login)
  78           {
  79             std::cout << "Authentication (" << auth.scheme() << ") required." << std::endl;
  80             std::cout << "Realm : " << auth.realm () << std::endl
  81                       << "User: " << auth.user() << std::endl
  82                       << "Password: \t\t\t (Enter password or press enter for default)\rPassword: ";
  83             char buf[80] = {0};
  84             u_int n = 0;
  85             int ch = 0;
  86             while (n < (sizeof(buf)-1) && (ch = std::cin.get ()) != '\n' && ch != std::char_traits<char>::eof ())
  87               {
  88                 buf[n++] = char(ch);
  89               }
  90             if (n>0)
  91               auth.password (buf);
  92           }
  93         return true;
  94       }
  95 };
  96
  97 class My_FTP_RequestHandler
  98   : public ACE::FTP::ClientRequestHandler
  99 {
 100   public:
 101     My_FTP_RequestHandler () : read_length_ (0) {}
 102     virtual ~My_FTP_RequestHandler () {}
 103
 104   protected:
 105     virtual void handle_request_error (const ACE::FTP::URL& url)
 106       {
 107         std::cout << "ERROR" << std::endl;
 108         std::cerr << "Failed to handle download for " << url.to_string ().c_str () << std::endl;
 109       }
 110
 111     virtual void handle_connection_error (const ACE::FTP::URL& url)
 112       {
 113         std::cout << "ERROR" << std::endl;
 114         std::cerr << "Failed to set up connection for " << url.to_string ().c_str () << std::endl;
 115       }
 116
 117     virtual void after_read (const char_type* /*buffer*/, int length_read)
 118       {
 119         this->read_length_ += length_read;
 120         std::cout << "\r [" << this->read_length_ << "/???]";
 121       }
 122
 123     virtual void on_eof ()
 124       {
 125         ACE::FTP::ClientRequestHandler::on_eof ();
 126         std::cout << std::endl;
 127       }
 128
 129   private:
 130     int read_length_;
 131 };
 132
 133 int
 134 ACE_TMAIN (int argc, ACE_TCHAR *argv [])
 135 {
 136   std::unique_ptr<std::ofstream> fout;
 137   std::ostream* sout = &std::cout;
 138
 139   if (!parse_args (argc, argv))
 140     {
 141       return 1;
 142     }
 143
 144   ACE::INet::URL_INetAuthBase::add_authenticator ("my_auth",
 145                                                   new My_Authenticator);
 146
 147   std::cout << "Starting..." << std::endl;
 148
 149   if (!url.empty ())
 150     {
 151       if (!outfile.empty ())
 152         {
 153           fout.reset (new std::ofstream (outfile.c_str (), std::ios_base::binary|std::ios_base::out));
 154
 155           if (!*fout)
 156             {
 157               std::cerr << "Failed to open output file : " << outfile.c_str () << std::endl;
 158               return 1;
 159             }
 160
 161           sout = fout.get ();
 162         }
 163
 164       ACE::FTP::URL ftp_url;
 165
 166       std::cout << "Parsing url [" << url.c_str () << "]" << std::endl;
 167
 168       if (!ftp_url.parse (url))
 169         {
 170           std::cerr << "Failed parsing url [" << url << "]" << std::endl;
 171           std::cerr << "\tresult = " << ftp_url.to_string ().c_str ();
 172           return 1;
 173         }
 174
 175       std::cout << "Opening url..." << std::endl;
 176       My_FTP_RequestHandler my_rh;
 177       if (do_active)  my_rh.use_active_mode ();
 178       ACE::INet::URLStream urlin = ftp_url.open (my_rh);
 179       if (urlin)
 180         {
 181           std::cout << "Saving to: ";
 182           if (!outfile.empty ())
 183             std::cout << '\'' << outfile.c_str () << '\'' << std::endl;
 184           else
 185             std::cout << "(stdout)" << std::endl;
 186
 187           (*sout) << urlin->rdbuf ();
 188           sout->flush ();
 189         }
 190     }
 191   else
 192     {
 193       std::cerr << "ERROR: No URL specified!" << std::endl;
 194       usage ();
 195       return 1;
 196     }
 197
 198   std::cout << "Done" << std::endl;
 199
 200   return 0;
 201 }