ftps.cc
上传用户:rrhhcc
上传日期:2015-12-11
资源大小:54129k
文件大小:4k
源码类别:

通讯编程

开发平台:

Visual C++

  1. /*
  2.  * Copyright (c) 1997 The Regents of the University of California.
  3.  * All rights reserved.
  4.  * 
  5.  * Redistribution and use in source and binary forms, with or without
  6.  * modification, are permitted provided that the following conditions
  7.  * are met:
  8.  * 1. Redistributions of source code must retain the above copyright
  9.  *    notice, this list of conditions and the following disclaimer.
  10.  * 2. Redistributions in binary form must reproduce the above copyright
  11.  *    notice, this list of conditions and the following disclaimer in the
  12.  *    documentation and/or other materials provided with the distribution.
  13.  * 3. All advertising materials mentioning features or use of this software
  14.  *    must display the following acknowledgement:
  15.  *  This product includes software developed by the Network Research
  16.  *  Group at Lawrence Berkeley National Laboratory.
  17.  * 4. Neither the name of the University nor of the Laboratory may be used
  18.  *    to endorse or promote products derived from this software without
  19.  *    specific prior written permission.
  20.  * 
  21.  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24.  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31.  * SUCH DAMAGE.
  32.  */
  33. /*
  34.  * This agent is paired with one or more ftp clients.
  35.  * It expects to be the target of a FullTcpAgent.
  36.  */
  37. static const char rcsid[] =
  38.     "@(#) $Header: /cvsroot/nsnam/ns-2/baytcp/ftps.cc,v 1.6 2001/09/06 22:09:41 haldar Exp $ ()";
  39. #include "tcp-full-bay.h"
  40. #include "tclcl.h"
  41. #include "random.h"
  42. #include "trace.h"
  43. #include "tcp.h"
  44. class FtpSrvrAgent : public BayTcpAppAgent {
  45.  public:
  46. FtpSrvrAgent();
  47. int command(int argc, const char*const* argv);
  48.   void recv(Packet*, BayFullTcpAgent*, int code);
  49.   //void recv(Packet*, BayFullTcpAgent*);
  50. protected:
  51. double now()  { return Scheduler::instance().clock(); }
  52. int min_response_; //number of bytes in min response file
  53. int max_response_; //number of bytes in max response file
  54. int filesize_;  //file size in Bytes
  55. };
  56. static class FtpSrvrClass : public TclClass {
  57. public:
  58. FtpSrvrClass() : TclClass("Agent/BayTcpApp/FtpServer") {}
  59. TclObject* create(int, const char*const*) {
  60. return (new FtpSrvrAgent());
  61. }
  62. } class_ftps;
  63. FtpSrvrAgent::FtpSrvrAgent() : BayTcpAppAgent(PT_NTYPE)
  64. {
  65. // bind("min_byte_response", &min_response_);
  66. // bind("max_byte_response", &max_response_);
  67. // bind_time("service_time", &service_time_);
  68. filesize_ = 0x10000000;
  69. }
  70. //should only be called when a get is received from a client
  71. //need to make sure it goes to the right tcp connection
  72. void FtpSrvrAgent::recv(Packet*, BayFullTcpAgent* tcp, int code)
  73. {
  74.   if(code == DATA_PUSH) {
  75.     int length = filesize_;
  76.     //tells tcp-full with my mods to send FIN when empty
  77.     tcp->advance(length, 1);
  78.   }
  79. }
  80. /*
  81.  * set length
  82.  */
  83. int FtpSrvrAgent::command(int argc, const char*const* argv)
  84. {
  85. if (argc == 3) {
  86. if (strcmp(argv[1], "file_size") == 0) {
  87. filesize_ = atoi(argv[2]);
  88. return (TCL_OK); 
  89. }
  90. return (Agent::command(argc, argv));
  91. }