Crypto++
files.h
00001 #ifndef CRYPTOPP_FILES_H
00002 #define CRYPTOPP_FILES_H
00003 
00004 #include "cryptlib.h"
00005 #include "filters.h"
00006 #include "argnames.h"
00007 
00008 #include <iostream>
00009 #include <fstream>
00010 
00011 NAMESPACE_BEGIN(CryptoPP)
00012 
00013 //! file-based implementation of Store interface
00014 class CRYPTOPP_DLL FileStore : public Store, private FilterPutSpaceHelper, public NotCopyable
00015 {
00016 public:
00017     class Err : public Exception
00018     {
00019     public:
00020         Err(const std::string &s) : Exception(IO_ERROR, s) {}
00021     };
00022     class OpenErr : public Err {public: OpenErr(const std::string &filename) : Err("FileStore: error opening file for reading: " + filename) {}};
00023     class ReadErr : public Err {public: ReadErr() : Err("FileStore: error reading file") {}};
00024 
00025     FileStore() : m_stream(NULL) {}
00026     FileStore(std::istream &in)
00027         {StoreInitialize(MakeParameters(Name::InputStreamPointer(), &in));}
00028     FileStore(const char *filename)
00029         {StoreInitialize(MakeParameters(Name::InputFileName(), filename));}
00030 #if defined(CRYPTOPP_UNIX_AVAILABLE) || _MSC_VER >= 1400
00031     //! specify file with Unicode name. On non-Windows OS, this function assumes that setlocale() has been called.
00032     FileStore(const wchar_t *filename)
00033         {StoreInitialize(MakeParameters(Name::InputFileNameWide(), filename));}
00034 #endif
00035 
00036     std::istream* GetStream() {return m_stream;}
00037 
00038     lword MaxRetrievable() const;
00039     size_t TransferTo2(BufferedTransformation &target, lword &transferBytes, const std::string &channel=DEFAULT_CHANNEL, bool blocking=true);
00040     size_t CopyRangeTo2(BufferedTransformation &target, lword &begin, lword end=LWORD_MAX, const std::string &channel=DEFAULT_CHANNEL, bool blocking=true) const;
00041     lword Skip(lword skipMax=ULONG_MAX);
00042 
00043 private:
00044     void StoreInitialize(const NameValuePairs &parameters);
00045     
00046     member_ptr<std::ifstream> m_file;
00047     std::istream *m_stream;
00048     byte *m_space;
00049     size_t m_len;
00050     bool m_waiting;
00051 };
00052 
00053 //! file-based implementation of Source interface
00054 class CRYPTOPP_DLL FileSource : public SourceTemplate<FileStore>
00055 {
00056 public:
00057     typedef FileStore::Err Err;
00058     typedef FileStore::OpenErr OpenErr;
00059     typedef FileStore::ReadErr ReadErr;
00060 
00061     FileSource(BufferedTransformation *attachment = NULL)
00062         : SourceTemplate<FileStore>(attachment) {}
00063     FileSource(std::istream &in, bool pumpAll, BufferedTransformation *attachment = NULL)
00064         : SourceTemplate<FileStore>(attachment) {SourceInitialize(pumpAll, MakeParameters(Name::InputStreamPointer(), &in));}
00065     FileSource(const char *filename, bool pumpAll, BufferedTransformation *attachment = NULL, bool binary=true)
00066         : SourceTemplate<FileStore>(attachment) {SourceInitialize(pumpAll, MakeParameters(Name::InputFileName(), filename)(Name::InputBinaryMode(), binary));}
00067 #if defined(CRYPTOPP_UNIX_AVAILABLE) || _MSC_VER >= 1400
00068     //! specify file with Unicode name. On non-Windows OS, this function assumes that setlocale() has been called.
00069     FileSource(const wchar_t *filename, bool pumpAll, BufferedTransformation *attachment = NULL, bool binary=true)
00070         : SourceTemplate<FileStore>(attachment) {SourceInitialize(pumpAll, MakeParameters(Name::InputFileNameWide(), filename)(Name::InputBinaryMode(), binary));}
00071 #endif
00072 
00073     std::istream* GetStream() {return m_store.GetStream();}
00074 };
00075 
00076 //! file-based implementation of Sink interface
00077 class CRYPTOPP_DLL FileSink : public Sink, public NotCopyable
00078 {
00079 public:
00080     class Err : public Exception
00081     {
00082     public:
00083         Err(const std::string &s) : Exception(IO_ERROR, s) {}
00084     };
00085     class OpenErr : public Err {public: OpenErr(const std::string &filename) : Err("FileSink: error opening file for writing: " + filename) {}};
00086     class WriteErr : public Err {public: WriteErr() : Err("FileSink: error writing file") {}};
00087 
00088     FileSink() : m_stream(NULL) {}
00089     FileSink(std::ostream &out)
00090         {IsolatedInitialize(MakeParameters(Name::OutputStreamPointer(), &out));}
00091     FileSink(const char *filename, bool binary=true)
00092         {IsolatedInitialize(MakeParameters(Name::OutputFileName(), filename)(Name::OutputBinaryMode(), binary));}
00093 #if defined(CRYPTOPP_UNIX_AVAILABLE) || _MSC_VER >= 1400
00094     //! specify file with Unicode name. On non-Windows OS, this function assumes that setlocale() has been called.
00095     FileSink(const wchar_t *filename, bool binary=true)
00096         {IsolatedInitialize(MakeParameters(Name::OutputFileNameWide(), filename)(Name::OutputBinaryMode(), binary));}
00097 #endif
00098 
00099     std::ostream* GetStream() {return m_stream;}
00100 
00101     void IsolatedInitialize(const NameValuePairs &parameters);
00102     size_t Put2(const byte *inString, size_t length, int messageEnd, bool blocking);
00103     bool IsolatedFlush(bool hardFlush, bool blocking);
00104 
00105 private:
00106     member_ptr<std::ofstream> m_file;
00107     std::ostream *m_stream;
00108 };
00109 
00110 NAMESPACE_END
00111 
00112 #endif