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

FileManager Class Reference Manages read/write/verification of files defined in metainfo file. More... #include <FileManager.h> Collaboration diagram for FileManager: [legend]List of all members. Public Types typedef std::vector< FIL * >  fil_v Public Member Functions   FileManager (PeerHandler &, const std::string &hash)   ~FileManager () size_t  ReadPiece (size_t index, unsigned char *buf, size_t length) size_t  ReadNextPiece (unsigned char *buf, size_t length) void  Write (size_t index, unsigned char *buf, size_t length, size_t begin) PeerHandler &  Handler () void  OpenFiles () size_t  GetFilesize (const std::string &) uint64_t  GetTotalLength () bool  Verify (size_t piece, size_t length) bool  CompareHash (unsigned char *h1, unsigned char *h2) Private Member Functions void  GetFiles (fil_v &files) Private Attributes std::string  m_hash PeerHandler &  m_handler std::string  m_basename fil_v  m_files int64_t  m_length FILE *  m_single_file Classes struct   FIL   File handle + information for one file in metainfo block. More... Detailed Description Manages read/write/verification of files defined in metainfo file. Definition at line 42 of file FileManager.h. Member Typedef Documentation typedef std::vector<FIL *> FileManager::fil_v Definition at line 52 of file FileManager.h. Constructor & Destructor Documentation FileManager::FileManager ( PeerHandler &  , const std::string &  hash   ) Definition at line 45 of file FileManager.cpp. References DEB, GetFiles(), PeerHandler::GetSession(), PeerHandler::GetTorrentDirectory(), m_basename, m_files, m_length, and OpenFiles(). :m_hash(hash) ,m_handler(h) ,m_single_file(NULL) { DEB( printf("FileManager()\n");) Session *sess = h.GetSession(hash); if (sess) { m_basename = h.GetTorrentDirectory() + "/" + sess -> GetInfoHash() + "/" + sess -> GetName(); GetFiles(m_files); m_length = sess -> GetLength(); DEB( printf("OpenFiles()\n");) OpenFiles(); } } FileManager::~FileManager (  )  Definition at line 64 of file FileManager.cpp. References DEB, m_files, and m_single_file. { DEB( printf("~FileManager()\n");) if (m_single_file) { fclose(m_single_file); } for (fil_v::iterator it = m_files.begin(); it != m_files.end(); it++) { FIL *p = *it; if (p -> fil) { fclose(p -> fil); } delete p; } } Member Function Documentation size_t FileManager::ReadPiece ( size_t  index, unsigned char *  buf, size_t  length   ) Definition at line 123 of file FileManager.cpp. References DEB, PeerHandler::GetSession(), Handler(), m_files, m_hash, m_length, and m_single_file. Referenced by Verify(). { Session *sess = Handler().GetSession(m_hash); if (!sess) return 0; size_t piece_length = sess -> GetPieceLength(); if (m_length) { long offset = index * piece_length; if (fseek(m_single_file,offset,SEEK_SET)) { DEB( perror("fseek() failed");) } size_t n = fread(buf,1,length,m_single_file); buf[0]++; return n; } int64_t start = index * piece_length; size_t bufset = 0; // offset in buf size_t left = piece_length; int64_t ptr = 0; size_t read_length = 0; //DEB(printf("ReadPiece(%d, ptr = %lld)\n", index, ptr);) for (fil_v::iterator it = m_files.begin(); it != m_files.end() && left; it++) { FIL *p = *it; // start within this file? if (start < ptr + p -> length) { long offset = start - ptr; size_t len = p -> length - offset; if (len > left) { len = left; } if (fseek(p -> fil,offset,SEEK_SET)) { DEB( perror("fseek() failed");) } read_length += fread(buf + bufset,1,len,p -> fil); // update start += len; bufset += len; left -= len; if (!left) return read_length; } ptr += p -> length; } buf[0]++; return read_length; } size_t FileManager::ReadNextPiece ( unsigned char *  buf, size_t  length   ) void FileManager::Write ( size_t  index, unsigned char *  buf, size_t  length, size_t  begin   ) Definition at line 177 of file FileManager.cpp. References DEB, PeerHandler::GetSession(), Handler(), m_files, m_hash, m_length, and m_single_file. { Session *sess = Handler().GetSession(m_hash); if (!sess) return; size_t piece_length = sess -> GetPieceLength(); if (m_length) { long offset = index * piece_length + begin; if (fseek(m_single_file,offset,SEEK_SET)) { DEB( perror("fseek() failed");) } long pos = ftell(m_single_file); if (pos < offset) { long diff = offset - pos; char *tmp = new char[diff]; memset(tmp,0,diff); fwrite(tmp,1,diff,m_single_file); delete tmp; } fwrite(buf,1,length,m_single_file); fflush(m_single_file); return; } int64_t start = index * piece_length + begin; size_t bufset = 0; // offset in buf size_t left = length; int64_t ptr = 0; for (fil_v::iterator it = m_files.begin(); it != m_files.end() && left; it++) { FIL *p = *it; // start within this file? if (start < ptr + p -> length) { long offset = start - ptr; size_t len = p -> length - offset; // length of file remaining from start of this write if (len > left) { len = left; } //DEB(printf("writing@%08x length %04x\n",offset,len);) if (fseek(p -> fil,offset,SEEK_SET)) { DEB( perror("fseek() failed");) } long pos = ftell(p -> fil); if (pos < offset) { long diff = offset - pos; char *tmp = new char[diff]; memset(tmp,0,diff); fwrite(tmp,1,diff,p -> fil); delete tmp; DEB(printf(" adding %ld bytes\n",diff);) } if (fwrite(buf + bufset,1,len,p -> fil) != len) { DEB( printf(" * short write!!!!!!\n");) } fflush(p -> fil); // update start += len; bufset += len; left -= len; } ptr += p -> length; } } PeerHandler& FileManager::Handler (  )  [inline] Definition at line 60 of file FileManager.h. References m_handler. Referenced by GetFiles(), OpenFiles(), ReadPiece(), Verify(), and Write(). { return m_handler; } void FileManager::OpenFiles (  )  Definition at line 83 of file FileManager.cpp. References DEB, GetFilesize(), Handler(), m_basename, m_files, m_length, m_single_file, and PeerHandler::mkpath(). Referenced by FileManager(). { if (m_length) // single file { std::string full = m_basename; //DEB(printf("single file: %s\n", full.c_str());) Handler().mkpath( full ); m_single_file = fopen(full.c_str(),"r+b"); if (!m_single_file) { m_single_file = fopen(full.c_str(),"w+b"); } size_t size = GetFilesize(full); if (size > m_length) { DEB( printf("Single file already too large - aborting\n");) } return; } // multi file for (fil_v::iterator it = m_files.begin(); it != m_files.end(); it++) { FIL *p = *it; std::string full = p -> path; //DEB(printf("multifile: %s\n", full.c_str());) Handler().mkpath( full ); p -> fil = fopen(full.c_str(),"r+b"); if (!p -> fil) { p -> fil = fopen(full.c_str(),"w+b"); } size_t size = GetFilesize(full); if (size > p -> length) { DEB( printf("File '%s' too large - aborting\n",full.c_str());) } } } size_t FileManager::GetFilesize ( const std::string &   )  Definition at line 251 of file FileManager.cpp. Referenced by OpenFiles(). { struct stat st; stat(path.c_str(),&st); return st.st_size; } uint64_t FileManager::GetTotalLength (  )  Definition at line 259 of file FileManager.cpp. References m_files. { uint64_t l = 0; for (fil_v::iterator it = m_files.begin(); it != m_files.end(); it++) { FIL *p = *it; l += p -> length; } return l; } bool FileManager::Verify ( size_t  piece, size_t  length   ) Definition at line 287 of file FileManager.cpp. References CompareHash(), DEB, PeerHandler::GetSession(), Handler(), m_hash, and ReadPiece(). { Session *sess = Handler().GetSession(m_hash); if (!sess) { DEB(printf("Verify: no session\n");) return false; } Piece *ptr = sess -> GetIncomplete(index); if (!ptr) { DEB(printf("Verify: no piece ptr\n");) return false; } unsigned char *p = new unsigned char[length]; unsigned char hash[20]; size_t read_length; read_length = ReadPiece(index, p, length ); if (read_length != length) { //DEB( printf(" *** Verify(): only read %d bytes out of %d\n",read_length,length);) delete[] p; return false; } SHA1( p, read_length, hash ); delete[] p; if (CompareHash(hash,(unsigned char *)sess -> GetPieces().substr(index * 20,20).c_str())) { DEB( printf("Verify OK!\n");) return true; } //DEB( printf("Verify failed\n");) return false; } bool FileManager::CompareHash ( unsigned char *  h1, unsigned char *  h2   ) Definition at line 323 of file FileManager.cpp. Referenced by Verify(). { for (size_t i = 0; i < 20; i++) if (h1[i] != h2[i]) return false; return true; } void FileManager::GetFiles ( fil_v &  files  )  [private] Definition at line 271 of file FileManager.cpp. References PeerHandler::GetSession(), PeerHandler::GetTorrentDirectory(), Handler(), and m_hash. Referenced by FileManager(). { Session *sess = Handler().GetSession(m_hash); if (!sess) return; file_v& ref = sess -> GetFiles(); for (file_v::iterator it = ref.begin(); it != ref.end(); it++) { file_t *f = *it; // name, offset, length std::string full = Handler().GetTorrentDirectory() + "/" + sess -> GetInfoHash() + "/" + f -> name; FIL *p2 = new FIL(f -> length, full); files.push_back(p2); } } Member Data Documentation std::string FileManager::m_hash [private] Definition at line 71 of file FileManager.h. Referenced by GetFiles(), ReadPiece(), Verify(), and Write(). PeerHandler& FileManager::m_handler [private] Definition at line 72 of file FileManager.h. Referenced by Handler(). std::string FileManager::m_basename [private] Definition at line 73 of file FileManager.h. Referenced by FileManager(), and OpenFiles(). fil_v FileManager::m_files [private] Definition at line 74 of file FileManager.h. Referenced by FileManager(), GetTotalLength(), OpenFiles(), ReadPiece(), Write(), and ~FileManager(). int64_t FileManager::m_length [private] Definition at line 75 of file FileManager.h. Referenced by FileManager(), OpenFiles(), ReadPiece(), and Write(). FILE* FileManager::m_single_file [private] Definition at line 76 of file FileManager.h. Referenced by OpenFiles(), ReadPiece(), Write(), and ~FileManager(). The documentation for this class was generated from the following files: FileManager.h FileManager.cpp

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

---