MediaBlockList.h

Go to the documentation of this file.

/*---------------------------------------------------------------------\
|                          ____ _   __ __ ___                          |
|                         |__  / \ / / . \ . \                         |
|                           / / \ V /|  _/  _/                         |
|                          / /__ | | | | | |                           |
|                         /_____||_| |_| |_|                           |
|                                                                      |
\---------------------------------------------------------------------*/
#ifndef ZYPP_MEDIA_MEDIABLOCKLIST_H
#define ZYPP_MEDIA_MEDIABLOCKLIST_H
 
#include <sys/types.h>
#include <vector>
#include <optional>
 
#include <zypp/Digest.h>
 
namespace zypp {
  namespace media {
 
struct MediaBlock {
  MediaBlock( off_t off_r, size_t size_r )
  : off( off_r )
  , size( size_r )
  {}
  off_t off;
  size_t size;
};
 
class MediaBlockList {
public:
  MediaBlockList(off_t filesize=off_t(-1));
 
  inline bool haveBlocks() const {
    return haveblocks;
  }
  size_t addBlock(off_t off, size_t size);
 
  inline MediaBlock getBlock(size_t blkno) const {
    return blocks[blkno];
  }
  inline size_t numBlocks() const {
    return blocks.size();
  }
 
  inline void setFilesize(off_t newfilesize=off_t(-1)) {
    filesize = newfilesize;
  }
  inline off_t getFilesize() const {
    return filesize;
  }
  inline bool haveFilesize() const {
    return filesize != off_t(-1);
  }
 
  void setFileChecksum(std::string ctype, int cl, unsigned char *c);
  std::string fileChecksumType () const;
 
  const UByteArray &getFileChecksum( );
  bool createFileDigest(Digest &digest) const;
  bool verifyFileDigest(Digest &digest) const;
  inline bool haveFileChecksum() const {
    return !fsumtype.empty() && fsum.size();
  }
 
  void setChecksum(size_t blkno, std::string cstype, int csl, unsigned char *cs, size_t cspad=0);
  bool checkChecksum(size_t blkno, const unsigned char *buf, size_t bufl) const;
  UByteArray getChecksum( size_t blkno ) const;
  std::string getChecksumType( ) const;
  bool createDigest(Digest &digest) const;
  bool verifyDigest(size_t blkno, Digest &digest) const;
  inline bool haveChecksum(size_t blkno) const {
    return chksumlen && chksums.size() >= chksumlen * (blkno + 1);
  }
 
  void setRsum(size_t blkno, int rsl, unsigned int rs, size_t rspad=0);
  bool checkRsum(size_t blkno, const unsigned char *buf, size_t bufl) const;
  unsigned int updateRsum(unsigned int rs, const char *bytes, size_t len) const;
  bool verifyRsum(size_t blkno, unsigned int rs) const;
  inline bool haveRsum(size_t blkno) const {
    return rsumlen && rsums.size() >= blkno + 1;
  }
 
  void reuseBlocks(FILE *wfp, std::string filename);
 
  std::string asString() const;
 
private:
  void writeBlock(size_t blkno, FILE *fp, const unsigned char *buf, size_t bufl, size_t start, std::vector<bool> &found) const;
  bool checkChecksumRotated(size_t blkno, const unsigned char *buf, size_t bufl, size_t start) const;
 
  off_t filesize;
  std::string fsumtype;
  UByteArray fsum;
 
  bool haveblocks;
  std::vector<MediaBlock> blocks;
 
  std::string chksumtype;
  int chksumlen;
  size_t chksumpad;
  std::vector<unsigned char> chksums;
 
  std::string rsumtype;
  int rsumlen;
  size_t rsumpad;
  std::vector<unsigned int> rsums;
};
 
inline std::ostream & operator<<(std::ostream &str, const MediaBlockList &bl)
{ return str << bl.asString(); }
 
  } // namespace media
} // namespace zypp
 
#endif // ZYPP_MEDIA_MEDIABLOCKLIST_H