OLD | NEW |
(Empty) | |
| 1 /* 7zStream.c -- 7z Stream functions |
| 2 2010-03-11 : Igor Pavlov : Public domain */ |
| 3 |
| 4 #include <string.h> |
| 5 |
| 6 #include "Types.h" |
| 7 |
| 8 SRes SeqInStream_Read2(ISeqInStream *stream, void *buf, size_t size, SRes errorT
ype) |
| 9 { |
| 10 while (size != 0) |
| 11 { |
| 12 size_t processed = size; |
| 13 RINOK(stream->Read(stream, buf, &processed)); |
| 14 if (processed == 0) |
| 15 return errorType; |
| 16 buf = (void *)((Byte *)buf + processed); |
| 17 size -= processed; |
| 18 } |
| 19 return SZ_OK; |
| 20 } |
| 21 |
| 22 SRes SeqInStream_Read(ISeqInStream *stream, void *buf, size_t size) |
| 23 { |
| 24 return SeqInStream_Read2(stream, buf, size, SZ_ERROR_INPUT_EOF); |
| 25 } |
| 26 |
| 27 SRes SeqInStream_ReadByte(ISeqInStream *stream, Byte *buf) |
| 28 { |
| 29 size_t processed = 1; |
| 30 RINOK(stream->Read(stream, buf, &processed)); |
| 31 return (processed == 1) ? SZ_OK : SZ_ERROR_INPUT_EOF; |
| 32 } |
| 33 |
| 34 SRes LookInStream_SeekTo(ILookInStream *stream, UInt64 offset) |
| 35 { |
| 36 Int64 t = offset; |
| 37 return stream->Seek(stream, &t, SZ_SEEK_SET); |
| 38 } |
| 39 |
| 40 SRes LookInStream_LookRead(ILookInStream *stream, void *buf, size_t *size) |
| 41 { |
| 42 const void *lookBuf; |
| 43 if (*size == 0) |
| 44 return SZ_OK; |
| 45 RINOK(stream->Look(stream, &lookBuf, size)); |
| 46 memcpy(buf, lookBuf, *size); |
| 47 return stream->Skip(stream, *size); |
| 48 } |
| 49 |
| 50 SRes LookInStream_Read2(ILookInStream *stream, void *buf, size_t size, SRes erro
rType) |
| 51 { |
| 52 while (size != 0) |
| 53 { |
| 54 size_t processed = size; |
| 55 RINOK(stream->Read(stream, buf, &processed)); |
| 56 if (processed == 0) |
| 57 return errorType; |
| 58 buf = (void *)((Byte *)buf + processed); |
| 59 size -= processed; |
| 60 } |
| 61 return SZ_OK; |
| 62 } |
| 63 |
| 64 SRes LookInStream_Read(ILookInStream *stream, void *buf, size_t size) |
| 65 { |
| 66 return LookInStream_Read2(stream, buf, size, SZ_ERROR_INPUT_EOF); |
| 67 } |
| 68 |
| 69 static SRes LookToRead_Look_Lookahead(void *pp, const void **buf, size_t *size) |
| 70 { |
| 71 SRes res = SZ_OK; |
| 72 CLookToRead *p = (CLookToRead *)pp; |
| 73 size_t size2 = p->size - p->pos; |
| 74 if (size2 == 0 && *size > 0) |
| 75 { |
| 76 p->pos = 0; |
| 77 size2 = LookToRead_BUF_SIZE; |
| 78 res = p->realStream->Read(p->realStream, p->buf, &size2); |
| 79 p->size = size2; |
| 80 } |
| 81 if (size2 < *size) |
| 82 *size = size2; |
| 83 *buf = p->buf + p->pos; |
| 84 return res; |
| 85 } |
| 86 |
| 87 static SRes LookToRead_Look_Exact(void *pp, const void **buf, size_t *size) |
| 88 { |
| 89 SRes res = SZ_OK; |
| 90 CLookToRead *p = (CLookToRead *)pp; |
| 91 size_t size2 = p->size - p->pos; |
| 92 if (size2 == 0 && *size > 0) |
| 93 { |
| 94 p->pos = 0; |
| 95 if (*size > LookToRead_BUF_SIZE) |
| 96 *size = LookToRead_BUF_SIZE; |
| 97 res = p->realStream->Read(p->realStream, p->buf, size); |
| 98 size2 = p->size = *size; |
| 99 } |
| 100 if (size2 < *size) |
| 101 *size = size2; |
| 102 *buf = p->buf + p->pos; |
| 103 return res; |
| 104 } |
| 105 |
| 106 static SRes LookToRead_Skip(void *pp, size_t offset) |
| 107 { |
| 108 CLookToRead *p = (CLookToRead *)pp; |
| 109 p->pos += offset; |
| 110 return SZ_OK; |
| 111 } |
| 112 |
| 113 static SRes LookToRead_Read(void *pp, void *buf, size_t *size) |
| 114 { |
| 115 CLookToRead *p = (CLookToRead *)pp; |
| 116 size_t rem = p->size - p->pos; |
| 117 if (rem == 0) |
| 118 return p->realStream->Read(p->realStream, buf, size); |
| 119 if (rem > *size) |
| 120 rem = *size; |
| 121 memcpy(buf, p->buf + p->pos, rem); |
| 122 p->pos += rem; |
| 123 *size = rem; |
| 124 return SZ_OK; |
| 125 } |
| 126 |
| 127 static SRes LookToRead_Seek(void *pp, Int64 *pos, ESzSeek origin) |
| 128 { |
| 129 CLookToRead *p = (CLookToRead *)pp; |
| 130 p->pos = p->size = 0; |
| 131 return p->realStream->Seek(p->realStream, pos, origin); |
| 132 } |
| 133 |
| 134 void LookToRead_CreateVTable(CLookToRead *p, int lookahead) |
| 135 { |
| 136 p->s.Look = lookahead ? |
| 137 LookToRead_Look_Lookahead : |
| 138 LookToRead_Look_Exact; |
| 139 p->s.Skip = LookToRead_Skip; |
| 140 p->s.Read = LookToRead_Read; |
| 141 p->s.Seek = LookToRead_Seek; |
| 142 } |
| 143 |
| 144 void LookToRead_Init(CLookToRead *p) |
| 145 { |
| 146 p->pos = p->size = 0; |
| 147 } |
| 148 |
| 149 static SRes SecToLook_Read(void *pp, void *buf, size_t *size) |
| 150 { |
| 151 CSecToLook *p = (CSecToLook *)pp; |
| 152 return LookInStream_LookRead(p->realStream, buf, size); |
| 153 } |
| 154 |
| 155 void SecToLook_CreateVTable(CSecToLook *p) |
| 156 { |
| 157 p->s.Read = SecToLook_Read; |
| 158 } |
| 159 |
| 160 static SRes SecToRead_Read(void *pp, void *buf, size_t *size) |
| 161 { |
| 162 CSecToRead *p = (CSecToRead *)pp; |
| 163 return p->realStream->Read(p->realStream, buf, size); |
| 164 } |
| 165 |
| 166 void SecToRead_CreateVTable(CSecToRead *p) |
| 167 { |
| 168 p->s.Read = SecToRead_Read; |
| 169 } |
OLD | NEW |