Search Site ~Apps~  btpeer   trackerbt   Socks4   The Minder   duct  ~Projects~  RobotWorld   Cgi++   IBank   SDL C++ Gui   Applesoft   Wolfshade   Legend  ~Contact~ message board  email author   visit project   homepage at   freshmeat.net

MetainfoSocket Class Reference Sending and receiving end of metainfo file transfer. More... #include <MetainfoSocket.h> Collaboration diagram for MetainfoSocket: [legend]List of all members. Public Member Functions   MetainfoSocket (ISocketHandler &)   MetainfoSocket (ISocketHandler &, const std::string &filename)   ~MetainfoSocket () void  OnAccept () void  OnConnect () void  OnDelete () void  OnRead () Private Types enum   state_t { STATE_GET_LENGTH = 0, STATE_GET_FILE } Private Member Functions   MetainfoSocket (const MetainfoSocket &s) MetainfoSocket &  operator= (const MetainfoSocket &) void  InitSession () Private Attributes int  m_filenr std::string  m_filename FILE *  m_fil state_t  m_state off_t  m_sz off_t  m_sz_read Static Private Attributes static int  m_nr = 0 Detailed Description Sending and receiving end of metainfo file transfer. File ......... MetainfoSocket.h Published .... 2005-06-14 Author ....... grymse@alhem.net Definition at line 32 of file MetainfoSocket.h. Member Enumeration Documentation enum MetainfoSocket::state_t [private] Enumerator: STATE_GET_LENGTH  STATE_GET_FILE  Definition at line 34 of file MetainfoSocket.h. { STATE_GET_LENGTH = 0, STATE_GET_FILE, }; Constructor & Destructor Documentation MetainfoSocket::MetainfoSocket ( ISocketHandler &   )  Definition at line 50 of file MetainfoSocket.cpp. References DEB. :TcpSocket(h) ,m_filenr(++m_nr) ,m_fil(NULL) ,m_state(STATE_GET_LENGTH) ,m_sz(0) ,m_sz_read(0) { DEB(printf("MetaInfoSocket()\n");) } MetainfoSocket::MetainfoSocket ( ISocketHandler &  , const std::string &  filename   ) Definition at line 62 of file MetainfoSocket.cpp. References DEB. :TcpSocket(h) ,m_filenr(0) ,m_filename(filename) ,m_fil(NULL) ,m_state(STATE_GET_LENGTH) ,m_sz(0) ,m_sz_read(0) { DEB(printf("MetaInfoSocket()\n");) } MetainfoSocket::~MetainfoSocket (  )  Definition at line 75 of file MetainfoSocket.cpp. References DEB, and m_fil. { DEB(printf("~MetaInfoSocket()\n");) if (m_fil) fclose(m_fil); } MetainfoSocket::MetainfoSocket ( const MetainfoSocket &  s  )  [inline, private] Definition at line 50 of file MetainfoSocket.h. : TcpSocket(s) {} // copy constructor Member Function Documentation void MetainfoSocket::OnAccept (  )  Definition at line 110 of file MetainfoSocket.cpp. References DEB, m_fil, m_filename, m_filenr, m_state, and STATE_GET_LENGTH. { DEB(printf("MetainfoSocket::OnAccept\n");) m_filename = "metafile." + Utility::l2string(m_filenr); m_fil = fopen(m_filename.c_str(), "wb"); m_state = STATE_GET_LENGTH; } void MetainfoSocket::OnConnect (  )  Definition at line 83 of file MetainfoSocket.cpp. References DEB, and m_filename. { DEB(printf("MetainfoSocket::OnConnect\n");) struct stat st; if (stat(m_filename.c_str(), &st) == -1) { perror("stat failed"); SetCloseAndDelete(); return; } SendBuf( (char *)&st.st_size, sizeof(off_t)); FILE *fil = fopen(m_filename.c_str(), "rb"); if (fil) { char buf[5000]; int n = fread(buf,1,5000,fil); while (n > 0) { SendBuf(buf,n); n = fread(buf,1,5000,fil); } fclose(fil); } SetCloseAndDelete(); } void MetainfoSocket::OnDelete (  )  Definition at line 159 of file MetainfoSocket.cpp. References DEB, InitSession(), m_fil, m_sz, and m_sz_read. { DEB(printf("MetainfoSocket::OnDelete\n");) if (m_fil) { fclose(m_fil); m_fil = NULL; if (m_sz_read == m_sz) { InitSession(); } else { printf("Short metainfo file read - ignoring\n"); } } } void MetainfoSocket::OnRead (  )  Definition at line 129 of file MetainfoSocket.cpp. References DEB, m_fil, m_state, m_sz, m_sz_read, STATE_GET_FILE, and STATE_GET_LENGTH. { TcpSocket::OnRead(); while (ibuf.GetLength()) { size_t l = ibuf.GetLength(); switch (m_state) { case STATE_GET_LENGTH: if (l < sizeof(off_t)) return; ibuf.Read( (char *)&m_sz, sizeof(off_t)); m_state = STATE_GET_FILE; DEB(printf("File length: %d bytes\n", m_sz);) break; case STATE_GET_FILE: { char slask[TCP_BUFSIZE_READ]; ibuf.Read( slask, l); if (m_fil) fwrite(slask,1,l,m_fil); m_sz_read += l; } break; } } } MetainfoSocket& MetainfoSocket::operator= ( const MetainfoSocket &   )  [inline, private] Definition at line 51 of file MetainfoSocket.h. { return *this; } // assignment operator void MetainfoSocket::InitSession (  )  [private] Definition at line 178 of file MetainfoSocket.cpp. References DEB, BString::GetBTObject(), BString::GetHash(), BString::GetHashAsString(), BString::GetInteger(), BString::GetString(), Exception::GetText(), PeerHandler::GetTorrentDirectory(), m_filename, PeerHandler::mkpath(), BString::read_file(), PeerHandler::RegSession(), and PeerHandler::SessionExists(). Referenced by OnDelete(). { DEB(printf("InitSession\n");) try { FILE *fil = fopen(m_filename.c_str(), "rb"); if (fil) { PeerHandler& ref = static_cast<PeerHandler&>(Handler()); BString meta; meta.read_file(fil); fclose(fil); std::string info_hash = meta.GetHashAsString("info"); // copy metainfo file std::string copy_to = ref.GetTorrentDirectory() + "/" + info_hash + "/.metainfo"; ref.mkpath(copy_to); fil = fopen(m_filename.c_str(), "rb"); if (fil) { FILE *fil2 = fopen(copy_to.c_str(), "wb"); char buf[1000]; size_t n = fread(buf, 1, 1000, fil); while (n > 0) { fwrite(buf, 1, n, fil2); n = fread(buf, 1, 1000, fil); } fclose(fil2); fclose(fil); } if (ref.SessionExists(info_hash)) { DEB(printf("Session for hash %s already exists\n", info_hash.c_str());) } else { DEB(printf("Hash: %s\n", info_hash.c_str());) Session *sess = new Session(Handler(), info_hash); ref.RegSession( sess ); sess -> SetHash(meta.GetHash("info")); BTString *p; if ((p = meta.GetString("announce")) != NULL) { sess -> SetAnnounce(p -> GetValue()); } std::string name; if ((p = meta.GetString("info.name")) != NULL) { sess -> SetName(name = p -> GetValue()); } BTInteger *piecelength = meta.GetInteger("info.piece length"); if (piecelength) { sess -> SetPieceLength(piecelength -> GetVal()); } BTInteger *length = meta.GetInteger("info.length"); if (length) { sess -> AddFile(length -> GetVal()); } else // info.files { BTObject *p = meta.GetBTObject("info.files"); BTList *files = dynamic_cast<BTList *>(p); if (files) { btobject_v& ref = files -> GetList(); for (btobject_v::iterator it = ref.begin(); it != ref.end(); it++) { BTDictionary *p = dynamic_cast<BTDictionary *>(*it); if (p) { BTInteger *length = dynamic_cast<BTInteger *>(p -> Find("length")); BTList *path = dynamic_cast<BTList *>(p -> Find("path")); if (path && length) { btobject_v& ref = path -> GetList(); std::string pathname = name; for (btobject_v::iterator it = ref.begin(); it != ref.end(); it++) { BTString *p = dynamic_cast<BTString *>(*it); if (p) pathname += "/" + p -> GetValue(); } sess -> AddFile(pathname, length -> GetVal()); } } } } } // pieces checksum BTString *pieces = meta.GetString("info.pieces"); if (pieces) { sess -> SetPieces( pieces -> GetValue() ); } sess -> CreateFileManager(); // restore session sess -> Load(); // /* DEB(printf("Verifying existing files...\n");) sess -> Verify(); DEB(printf("Verify finished\n");) */ } // unlink(m_filename.c_str()); } } catch (const Exception& e) { Handler().LogError(this, "InitSession", 0, e.GetText(), LOG_LEVEL_ERROR); } } Member Data Documentation int MetainfoSocket::m_nr = 0 [static, private] Definition at line 55 of file MetainfoSocket.h. int MetainfoSocket::m_filenr [private] Definition at line 56 of file MetainfoSocket.h. Referenced by OnAccept(). std::string MetainfoSocket::m_filename [private] Definition at line 57 of file MetainfoSocket.h. Referenced by InitSession(), OnAccept(), and OnConnect(). FILE* MetainfoSocket::m_fil [private] Definition at line 58 of file MetainfoSocket.h. Referenced by OnAccept(), OnDelete(), OnRead(), and ~MetainfoSocket(). state_t MetainfoSocket::m_state [private] Definition at line 59 of file MetainfoSocket.h. Referenced by OnAccept(), and OnRead(). off_t MetainfoSocket::m_sz [private] Definition at line 60 of file MetainfoSocket.h. Referenced by OnDelete(), and OnRead(). off_t MetainfoSocket::m_sz_read [private] Definition at line 61 of file MetainfoSocket.h. Referenced by OnDelete(), and OnRead(). The documentation for this class was generated from the following files: MetainfoSocket.h MetainfoSocket.cpp

Page, code, and content Copyright (C) 2006 by Anders Hedström

---