| /* 7zStream.c -- 7z Stream functions |
| 2017-04-03 : Igor Pavlov : Public domain */ |
| |
| #include "Precomp.h" |
| |
| #include <string.h> |
| |
| #include "7zTypes.h" |
| |
| SRes SeqInStream_Read2(const ISeqInStream *stream, void *buf, size_t size, SRes errorType) |
| { |
| while (size != 0) |
| { |
| size_t processed = size; |
| RINOK(ISeqInStream_Read(stream, buf, &processed)); |
| if (processed == 0) |
| return errorType; |
| buf = (void *)((Byte *)buf + processed); |
| size -= processed; |
| } |
| return SZ_OK; |
| } |
| |
| SRes SeqInStream_Read(const ISeqInStream *stream, void *buf, size_t size) |
| { |
| return SeqInStream_Read2(stream, buf, size, SZ_ERROR_INPUT_EOF); |
| } |
| |
| SRes SeqInStream_ReadByte(const ISeqInStream *stream, Byte *buf) |
| { |
| size_t processed = 1; |
| RINOK(ISeqInStream_Read(stream, buf, &processed)); |
| return (processed == 1) ? SZ_OK : SZ_ERROR_INPUT_EOF; |
| } |
| |
| |
| |
| SRes LookInStream_SeekTo(const ILookInStream *stream, UInt64 offset) |
| { |
| Int64 t = offset; |
| return ILookInStream_Seek(stream, &t, SZ_SEEK_SET); |
| } |
| |
| SRes LookInStream_LookRead(const ILookInStream *stream, void *buf, size_t *size) |
| { |
| const void *lookBuf; |
| if (*size == 0) |
| return SZ_OK; |
| RINOK(ILookInStream_Look(stream, &lookBuf, size)); |
| memcpy(buf, lookBuf, *size); |
| return ILookInStream_Skip(stream, *size); |
| } |
| |
| SRes LookInStream_Read2(const ILookInStream *stream, void *buf, size_t size, SRes errorType) |
| { |
| while (size != 0) |
| { |
| size_t processed = size; |
| RINOK(ILookInStream_Read(stream, buf, &processed)); |
| if (processed == 0) |
| return errorType; |
| buf = (void *)((Byte *)buf + processed); |
| size -= processed; |
| } |
| return SZ_OK; |
| } |
| |
| SRes LookInStream_Read(const ILookInStream *stream, void *buf, size_t size) |
| { |
| return LookInStream_Read2(stream, buf, size, SZ_ERROR_INPUT_EOF); |
| } |
| |
| |
| |
| #define GET_LookToRead2 CLookToRead2 *p = CONTAINER_FROM_VTBL(pp, CLookToRead2, vt); |
| |
| static SRes LookToRead2_Look_Lookahead(const ILookInStream *pp, const void **buf, size_t *size) |
| { |
| SRes res = SZ_OK; |
| GET_LookToRead2 |
| size_t size2 = p->size - p->pos; |
| if (size2 == 0 && *size != 0) |
| { |
| p->pos = 0; |
| p->size = 0; |
| size2 = p->bufSize; |
| res = ISeekInStream_Read(p->realStream, p->buf, &size2); |
| p->size = size2; |
| } |
| if (*size > size2) |
| *size = size2; |
| *buf = p->buf + p->pos; |
| return res; |
| } |
| |
| static SRes LookToRead2_Look_Exact(const ILookInStream *pp, const void **buf, size_t *size) |
| { |
| SRes res = SZ_OK; |
| GET_LookToRead2 |
| size_t size2 = p->size - p->pos; |
| if (size2 == 0 && *size != 0) |
| { |
| p->pos = 0; |
| p->size = 0; |
| if (*size > p->bufSize) |
| *size = p->bufSize; |
| res = ISeekInStream_Read(p->realStream, p->buf, size); |
| size2 = p->size = *size; |
| } |
| if (*size > size2) |
| *size = size2; |
| *buf = p->buf + p->pos; |
| return res; |
| } |
| |
| static SRes LookToRead2_Skip(const ILookInStream *pp, size_t offset) |
| { |
| GET_LookToRead2 |
| p->pos += offset; |
| return SZ_OK; |
| } |
| |
| static SRes LookToRead2_Read(const ILookInStream *pp, void *buf, size_t *size) |
| { |
| GET_LookToRead2 |
| size_t rem = p->size - p->pos; |
| if (rem == 0) |
| return ISeekInStream_Read(p->realStream, buf, size); |
| if (rem > *size) |
| rem = *size; |
| memcpy(buf, p->buf + p->pos, rem); |
| p->pos += rem; |
| *size = rem; |
| return SZ_OK; |
| } |
| |
| static SRes LookToRead2_Seek(const ILookInStream *pp, Int64 *pos, ESzSeek origin) |
| { |
| GET_LookToRead2 |
| p->pos = p->size = 0; |
| return ISeekInStream_Seek(p->realStream, pos, origin); |
| } |
| |
| void LookToRead2_CreateVTable(CLookToRead2 *p, int lookahead) |
| { |
| p->vt.Look = lookahead ? |
| LookToRead2_Look_Lookahead : |
| LookToRead2_Look_Exact; |
| p->vt.Skip = LookToRead2_Skip; |
| p->vt.Read = LookToRead2_Read; |
| p->vt.Seek = LookToRead2_Seek; |
| } |
| |
| |
| |
| static SRes SecToLook_Read(const ISeqInStream *pp, void *buf, size_t *size) |
| { |
| CSecToLook *p = CONTAINER_FROM_VTBL(pp, CSecToLook, vt); |
| return LookInStream_LookRead(p->realStream, buf, size); |
| } |
| |
| void SecToLook_CreateVTable(CSecToLook *p) |
| { |
| p->vt.Read = SecToLook_Read; |
| } |
| |
| static SRes SecToRead_Read(const ISeqInStream *pp, void *buf, size_t *size) |
| { |
| CSecToRead *p = CONTAINER_FROM_VTBL(pp, CSecToRead, vt); |
| return ILookInStream_Read(p->realStream, buf, size); |
| } |
| |
| void SecToRead_CreateVTable(CSecToRead *p) |
| { |
| p->vt.Read = SecToRead_Read; |
| } |