1 /*
2 * Copyright (C) 2021 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include "incremental_source_stream.h"
17
18 #include <algorithm>
19 #include <vector>
20 #include "image_log.h"
21 #ifndef _WIN32
22 #include "securec.h"
23 #else
24 #include "memory.h"
25 #endif
26
27 #undef LOG_DOMAIN
28 #define LOG_DOMAIN LOG_TAG_DOMAIN_ID_IMAGE
29
30 #undef LOG_TAG
31 #define LOG_TAG "IncrementalSourceStream"
32
33 namespace OHOS {
34 namespace Media {
35 using namespace std;
36 using namespace ImagePlugin;
37
IncrementalSourceStream(IncrementalMode mode)38 IncrementalSourceStream::IncrementalSourceStream(IncrementalMode mode)
39 : incrementalMode_(mode), isFinalize_(false), dataSize_(0), dataOffset_(0)
40 {}
41
CreateSourceStream(IncrementalMode mode)42 unique_ptr<IncrementalSourceStream> IncrementalSourceStream::CreateSourceStream(IncrementalMode mode)
43 {
44 IMAGE_LOGD("[IncrementalSourceStream]mode:%{public}d.", mode);
45 return (unique_ptr<IncrementalSourceStream>(new IncrementalSourceStream(mode)));
46 }
47
Read(uint32_t desiredSize,DataStreamBuffer & outData)48 bool IncrementalSourceStream::Read(uint32_t desiredSize, DataStreamBuffer &outData)
49 {
50 if (!Peek(desiredSize, outData)) {
51 IMAGE_LOGE("[IncrementalSourceStream]read fail.");
52 return false;
53 }
54 dataOffset_ += outData.dataSize;
55 return true;
56 }
57
Peek(uint32_t desiredSize,DataStreamBuffer & outData)58 bool IncrementalSourceStream::Peek(uint32_t desiredSize, DataStreamBuffer &outData)
59 {
60 if (desiredSize == 0) {
61 IMAGE_LOGE("[IncrementalSourceStream]input the parameter exception.");
62 return false;
63 }
64 if (sourceData_.empty() || dataSize_ == 0 || dataOffset_ >= dataSize_) {
65 IMAGE_LOGE("[IncrementalSourceStream]source data exception. dataSize_:%{public}zu,"
66 "dataOffset_:%{public}zu.", dataSize_, dataOffset_);
67 return false;
68 }
69 outData.bufferSize = dataSize_ - dataOffset_;
70 if (desiredSize > dataSize_ - dataOffset_) {
71 desiredSize = dataSize_ - dataOffset_;
72 }
73 outData.dataSize = desiredSize;
74 outData.inputStreamBuffer = sourceData_.data() + dataOffset_;
75 IMAGE_LOGD("[IncrementalSourceStream]Peek end. desiredSize:%{public}u, offset:%{public}zu,"
76 "dataSize_:%{public}zu,dataOffset_:%{public}zu.", desiredSize, dataOffset_, dataSize_, dataOffset_);
77 return true;
78 }
79
Read(uint32_t desiredSize,uint8_t * outBuffer,uint32_t bufferSize,uint32_t & readSize)80 bool IncrementalSourceStream::Read(uint32_t desiredSize, uint8_t *outBuffer, uint32_t bufferSize, uint32_t &readSize)
81 {
82 if (!Peek(desiredSize, outBuffer, bufferSize, readSize)) {
83 IMAGE_LOGE("[IncrementalSourceStream]read fail.");
84 return false;
85 }
86 dataOffset_ += readSize;
87 return true;
88 }
89
Peek(uint32_t desiredSize,uint8_t * outBuffer,uint32_t bufferSize,uint32_t & readSize)90 bool IncrementalSourceStream::Peek(uint32_t desiredSize, uint8_t *outBuffer, uint32_t bufferSize, uint32_t &readSize)
91 {
92 if (desiredSize == 0 || outBuffer == nullptr || desiredSize > bufferSize) {
93 IMAGE_LOGE("[IncrementalSourceStream]input parameter exception, desiredSize:%{public}u,"
94 "bufferSize:%{public}u.", desiredSize, bufferSize);
95 return false;
96 }
97 if (sourceData_.empty() || dataSize_ == 0 || dataOffset_ >= dataSize_) {
98 IMAGE_LOGE("[IncrementalSourceStream]source data exception. dataSize_:%{public}zu,"
99 "dataOffset_:%{public}zu.", dataSize_, dataOffset_);
100 return false;
101 }
102 if (desiredSize > (dataSize_ - dataOffset_)) {
103 desiredSize = dataSize_ - dataOffset_;
104 }
105 errno_t ret = memcpy_s(outBuffer, bufferSize, sourceData_.data() + dataOffset_, desiredSize);
106 if (ret != 0) {
107 IMAGE_LOGE("[IncrementalSourceStream]copy data fail, ret:%{public}d, bufferSize:%{public}u,"
108 "offset:%{public}zu, desiredSize:%{public}u, dataSize:%{public}zu.", ret, bufferSize, dataOffset_,
109 desiredSize, dataSize_);
110 return false;
111 }
112 readSize = desiredSize;
113 return true;
114 }
115
Tell()116 uint32_t IncrementalSourceStream::Tell()
117 {
118 return dataOffset_;
119 }
120
Seek(uint32_t position)121 bool IncrementalSourceStream::Seek(uint32_t position)
122 {
123 if (position >= dataSize_) {
124 IMAGE_LOGE("[IncrementalSourceStream]Seek the position greater than the Data Size.");
125 return false;
126 }
127 dataOffset_ = position;
128 return true;
129 }
130
UpdateData(const uint8_t * data,uint32_t size,bool isCompleted)131 uint32_t IncrementalSourceStream::UpdateData(const uint8_t *data, uint32_t size, bool isCompleted)
132 {
133 if (data == nullptr) {
134 IMAGE_LOGE("[IncrementalSourceStream]input the parameter exception.");
135 return ERR_IMAGE_DATA_ABNORMAL;
136 }
137 if (size == 0) {
138 IMAGE_LOGD("[IncrementalSourceStream]no need to update data.");
139 return SUCCESS;
140 }
141 if (incrementalMode_ == IncrementalMode::INCREMENTAL_DATA) {
142 vector<uint8_t> newData;
143 newData.resize(size);
144 copy(data, data + size, newData.begin());
145 sourceData_.resize(dataSize_ + size);
146 sourceData_.insert(sourceData_.begin() + dataSize_, newData.begin(), newData.end());
147 dataSize_ += size;
148 isFinalize_ = isCompleted;
149 } else {
150 sourceData_.clear();
151 sourceData_.resize(size);
152 dataSize_ = size;
153 copy(data, data + size, sourceData_.begin());
154 isFinalize_ = true;
155 }
156 return SUCCESS;
157 }
158
IsStreamCompleted()159 bool IncrementalSourceStream::IsStreamCompleted()
160 {
161 return isFinalize_;
162 }
163
GetStreamSize()164 size_t IncrementalSourceStream::GetStreamSize()
165 {
166 return dataSize_;
167 }
168
GetDataPtr()169 uint8_t *IncrementalSourceStream::GetDataPtr()
170 {
171 return sourceData_.data();
172 }
173
174 } // namespace Media
175 } // namespace OHOS
176