• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // InOutTempBuffer.cpp
2 
3 #include "StdAfx.h"
4 
5 #include "../../../C/7zCrc.h"
6 
7 #include "../../Common/Defs.h"
8 
9 #include "InOutTempBuffer.h"
10 #include "StreamUtils.h"
11 
12 using namespace NWindows;
13 using namespace NFile;
14 using namespace NDir;
15 
16 static const size_t kTempBufSize = (1 << 20);
17 
18 static CFSTR kTempFilePrefixString = FTEXT("7zt");
19 
CInOutTempBuffer()20 CInOutTempBuffer::CInOutTempBuffer(): _buf(NULL) { }
21 
Create()22 void CInOutTempBuffer::Create()
23 {
24   if (!_buf)
25     _buf = new Byte[kTempBufSize];
26 }
27 
~CInOutTempBuffer()28 CInOutTempBuffer::~CInOutTempBuffer()
29 {
30   delete []_buf;
31 }
32 
InitWriting()33 void CInOutTempBuffer::InitWriting()
34 {
35   _bufPos = 0;
36   _tempFileCreated = false;
37   _size = 0;
38   _crc = CRC_INIT_VAL;
39 }
40 
WriteToFile(const void * data,UInt32 size)41 bool CInOutTempBuffer::WriteToFile(const void *data, UInt32 size)
42 {
43   if (size == 0)
44     return true;
45   if (!_tempFileCreated)
46   {
47     if (!_tempFile.CreateRandomInTempFolder(kTempFilePrefixString, &_outFile))
48       return false;
49     _tempFileCreated = true;
50   }
51   UInt32 processed;
52   if (!_outFile.Write(data, size, processed))
53     return false;
54   _crc = CrcUpdate(_crc, data, processed);
55   _size += processed;
56   return (processed == size);
57 }
58 
Write(const void * data,UInt32 size)59 bool CInOutTempBuffer::Write(const void *data, UInt32 size)
60 {
61   if (size == 0)
62     return true;
63   size_t cur = kTempBufSize - _bufPos;
64   if (cur != 0)
65   {
66     if (cur > size)
67       cur = size;
68     memcpy(_buf + _bufPos, data, cur);
69     _crc = CrcUpdate(_crc, data, cur);
70     _bufPos += cur;
71     _size += cur;
72     size -= (UInt32)cur;
73     data = ((const Byte *)data) + cur;
74   }
75   return WriteToFile(data, size);
76 }
77 
WriteToStream(ISequentialOutStream * stream)78 HRESULT CInOutTempBuffer::WriteToStream(ISequentialOutStream *stream)
79 {
80   if (!_outFile.Close())
81     return E_FAIL;
82 
83   UInt64 size = 0;
84   UInt32 crc = CRC_INIT_VAL;
85 
86   if (_bufPos != 0)
87   {
88     RINOK(WriteStream(stream, _buf, _bufPos));
89     crc = CrcUpdate(crc, _buf, _bufPos);
90     size += _bufPos;
91   }
92 
93   if (_tempFileCreated)
94   {
95     NIO::CInFile inFile;
96     if (!inFile.Open(_tempFile.GetPath()))
97       return E_FAIL;
98     while (size < _size)
99     {
100       UInt32 processed;
101       if (!inFile.ReadPart(_buf, kTempBufSize, processed))
102         return E_FAIL;
103       if (processed == 0)
104         break;
105       RINOK(WriteStream(stream, _buf, processed));
106       crc = CrcUpdate(crc, _buf, processed);
107       size += processed;
108     }
109   }
110 
111   return (_crc == crc && size == _size) ? S_OK : E_FAIL;
112 }
113 
114 /*
115 STDMETHODIMP CSequentialOutTempBufferImp::Write(const void *data, UInt32 size, UInt32 *processed)
116 {
117   if (!_buf->Write(data, size))
118   {
119     if (processed)
120       *processed = 0;
121     return E_FAIL;
122   }
123   if (processed)
124     *processed = size;
125   return S_OK;
126 }
127 */
128