stdex
Additional custom or not Standard C++ covered algorithms
Loading...
Searching...
No Matches
stream.hpp
1/*
2 SPDX-License-Identifier: MIT
3 Copyright © 2023 Amebis
4*/
5
6#pragma once
7
8#include "endian.hpp"
9#include "interval.hpp"
10#include "math.hpp"
11#include "ring.hpp"
12#include "sal.hpp"
13#include "string.hpp"
14#include "system.hpp"
15#include "unicode.hpp"
16#include <assert.h>
17#include <stdint.h>
18#include <stdlib.h>
19#if defined(_WIN32) && !defined(WIN32_LEAN_AND_MEAN)
20#include <asptlb.h>
21#endif
22#include <chrono>
23#include <condition_variable>
24#include <list>
25#include <memory>
26#include <string>
27#include <thread>
28#include <vector>
29
30#if !defined(SET_FILE_OP_TIMES) && defined(RDAT_BELEZI_CAS_DOSTOPA_VER)
31#define SET_FILE_OP_TIMES 1
32#pragma message("RDAT_BELEZI_CAS_DOSTOPA_VER is deprecated. Use SET_FILE_OP_TIMES instead.")
33#elif !defined(SET_FILE_OP_TIMES)
34#define SET_FILE_OP_TIMES 0
35#endif
36#if !defined(CHECK_STREAM_STATE) && defined(RDAT_NE_PREVERJAJ_STANJA_VER)
37#define CHECK_STREAM_STATE 0
38#pragma message("RDAT_NE_PREVERJAJ_EOF_VER is deprecated. Use CHECK_STREAM_STATE=0 instead.")
39#else
40#define CHECK_STREAM_STATE 1
41#endif
42
43namespace stdex
44{
45 namespace stream
46 {
50 enum class state_t {
51 ok = 0,
52 eof,
53 fail,
54 };
55
59 using fsize_t = uint64_t;
60 constexpr fsize_t fsize_max = UINT64_MAX;
61
62 constexpr size_t iterate_count = 0x10;
63 constexpr size_t default_block_size = 0x10000;
64 constexpr wchar_t utf16_bom = L'\ufeff';
65 constexpr const char utf8_bom[3] = { '\xef', '\xbb', '\xbf' };
66
70 class basic
71 {
72 public:
73 basic(_In_ state_t state = state_t::ok) : m_state(state) {}
74
86 virtual _Success_(return != 0 || length == 0) size_t read(
87 _Out_writes_bytes_to_opt_(length, return) void* data, _In_ size_t length)
88 {
89 UNREFERENCED_PARAMETER(data);
90 UNREFERENCED_PARAMETER(length);
91 m_state = state_t::fail;
92 return 0;
93 }
94
104 virtual _Success_(return != 0) size_t write(
105 _In_reads_bytes_opt_(length) const void* data, _In_ size_t length)
106 {
107 UNREFERENCED_PARAMETER(data);
108 UNREFERENCED_PARAMETER(length);
109 m_state = state_t::fail;
110 return 0;
111 }
112
116 virtual void flush()
117 {
118 m_state = state_t::ok;
119 }
120
124 virtual void close()
125 {
126 m_state = state_t::ok;
127 }
128
132 virtual void skip(_In_ fsize_t amount)
133 {
134 if (amount == 1)
135 read_byte();
136 else if (amount < iterate_count) {
137 for (size_t i = 0; i < static_cast<size_t>(amount); i++) {
138 read_byte();
139 if (!ok()) _Unlikely_
140 break;
141 }
142 }
143 else {
144 size_t block = static_cast<size_t>(std::min<fsize_t>(amount, default_block_size));
145 try {
146 std::unique_ptr<uint8_t[]> dummy(new uint8_t[block]);
147 while (amount) {
148 amount -= read_array(dummy.get(), sizeof(uint8_t), static_cast<size_t>(std::min<fsize_t>(amount, block)));
149 if (!ok()) _Unlikely_
150 break;
151 }
152 }
153 catch (std::bad_alloc) { m_state = state_t::fail; }
154 }
155 }
156
160 inline state_t state() const { return m_state; };
161
165 inline bool ok() const { return m_state == state_t::ok; };
166
174 virtual std::vector<uint8_t> read_remainder(_In_ size_t max_length = SIZE_MAX)
175 {
176 std::vector<uint8_t> result;
177 size_t offset, length;
178 offset = 0;
179 length = default_block_size;
180 while (offset < max_length) {
181 length = std::min(length, max_length);
182 try { result.resize(length); }
183 catch (std::bad_alloc) {
184 m_state = state_t::fail;
185 return result;
186 }
187 auto num_read = read_array(result.data() + offset, sizeof(uint8_t), length - offset);
188 offset += num_read;
189 if (!ok()) _Unlikely_
190 break;
191 length += default_block_size;
192 }
193 result.resize(offset);
194 return result;
195 }
196
200 inline uint8_t read_byte()
201 {
202 uint8_t byte;
203 if (read_array(&byte, sizeof(byte), 1) == 1)
204 return byte;
205 throw std::runtime_error("failed to read");
206 }
207
211 void write_byte(_In_ uint8_t byte, _In_ fsize_t amount = 1)
212 {
213 if (amount == 1)
214 write(&byte, sizeof(uint8_t));
215 else if (amount < iterate_count) {
216 for (size_t i = 0; i < static_cast<size_t>(amount); i++) {
217 write(&byte, sizeof(uint8_t));
218 if (!ok()) _Unlikely_
219 break;
220 }
221 }
222 else {
223 size_t block = static_cast<size_t>(std::min<fsize_t>(amount, default_block_size));
224 try {
225 std::unique_ptr<uint8_t[]> dummy(new uint8_t[block]);
226 memset(dummy.get(), byte, block);
227 while (amount) {
228 amount -= write_array(dummy.get(), sizeof(uint8_t), static_cast<size_t>(std::min<fsize_t>(amount, block)));
229 if (!ok()) _Unlikely_
230 break;
231 }
232 }
233 catch (std::bad_alloc) { m_state = state_t::fail; }
234 }
235 }
236
248 template <class T>
249 inline basic& read_data(_Out_ T& data)
250 {
251 if (!ok()) _Unlikely_ {
252 data = 0;
253 return *this;
254 }
255 if (read_array(&data, sizeof(T), 1) == 1)
256 LE2HE(&data);
257 else {
258 data = 0;
259 if (ok())
260 m_state = state_t::eof;
261 }
262 return *this;
263 }
264
276 template <class T>
277 inline basic& write_data(_In_ const T data)
278 {
279 if (!ok()) _Unlikely_
280 return *this;
281#ifdef BIG_ENDIAN
282 T data_le = HE2LE(data);
283 write(&data_le, sizeof(T));
284#else
285 write(&data, sizeof(T));
286#endif
287 return *this;
288 }
289
295 template<class _Traits = std::char_traits<char>, class _Ax = std::allocator<char>>
296 inline size_t readln(_Inout_ std::basic_string<char, _Traits, _Ax>& str)
297 {
298 str.clear();
299 return readln_and_attach(str);
300 }
301
307 template<class _Traits = std::char_traits<wchar_t>, class _Ax = std::allocator<wchar_t>>
308 inline size_t readln(_Inout_ std::basic_string<wchar_t, _Traits, _Ax>& wstr)
309 {
310 wstr.clear();
311 return readln_and_attach(wstr);
312 }
313
319 template<class _Traits = std::char_traits<wchar_t>, class _Ax = std::allocator<wchar_t>>
320 size_t readln(_Inout_ std::basic_string<wchar_t, _Traits, _Ax>& wstr, _In_ charset_id charset)
321 {
322 if (charset == charset_id::utf16)
323 return readln(wstr);
324 std::string str;
326 wstr.clear();
327 str2wstr(wstr, str, charset);
328 return wstr.size();
329 }
330
336 template<class _Elem, class _Traits = std::char_traits<_Elem>, class _Ax = std::allocator<_Elem>>
337 size_t readln_and_attach(_Inout_ std::basic_string<_Elem, _Traits, _Ax>& str)
338 {
339 bool initial = true;
340 _Elem chr, previous = (_Elem)0;
341 do {
342 read_array(&chr, sizeof(_Elem), 1);
343 if (!initial && !(previous == static_cast<_Elem>('\r') && chr == static_cast<_Elem>('\n')))
344 str += previous;
345 else
346 initial = false;
347 previous = chr;
348 } while (ok() && chr != static_cast<_Elem>('\n'));
349 return str.size();
350 }
351
357 template<class _Traits = std::char_traits<wchar_t>, class _Ax = std::allocator<wchar_t>>
358 size_t readln_and_attach(_Inout_ std::basic_string<wchar_t, _Traits, _Ax>& wstr, _In_ charset_id charset)
359 {
360 if (charset == charset_id::utf16)
361 return readln_and_attach(wstr);
362 std::string str;
364 str2wstr(wstr, str, charset);
365 return wstr.size();
366 }
367
373 size_t read_array(_Out_writes_bytes_(size* count) void* array, _In_ size_t size, _In_ size_t count)
374 {
375 for (size_t to_read = mul(size, count);;) {
376 size_t num_read = read(array, to_read);
377 to_read -= num_read;
378 if (!to_read)
379 return count;
380 if (!ok()) _Unlikely_
381 return count - to_read / size;
382 reinterpret_cast<uint8_t*&>(array) += num_read;
383 }
384 }
385
391 inline size_t write_array(_In_reads_bytes_opt_(size* count) const void* array, _In_ size_t size, _In_ size_t count)
392 {
393 return write(array, mul(size, count)) / size;
394 }
395
405 size_t write_array(_In_reads_or_z_opt_(num_chars) const wchar_t* wstr, _In_ size_t num_chars, _In_ charset_id charset)
406 {
407 if (!ok()) _Unlikely_
408 return 0;
409 num_chars = stdex::strnlen(wstr, num_chars);
410 if (charset != charset_id::utf16) {
411 std::string str(wstr2str(wstr, num_chars, charset));
412 return write_array(str.data(), sizeof(char), str.size());
413 }
414 return write_array(wstr, sizeof(wchar_t), num_chars);
415 }
416
428 template<class _Elem, class _Traits = std::char_traits<_Elem>, class _Ax = std::allocator<_Elem>>
429 inline basic& read_str(_Inout_ std::basic_string<_Elem, _Traits, _Ax>& data)
430 {
431 uint32_t num_chars;
432 read_data(num_chars);
433 if (!ok()) _Unlikely_ {
434 data.clear();
435 return *this;
436 }
437 data.resize(num_chars);
438 data.resize(read_array(data.data(), sizeof(_Elem), num_chars));
439 return *this;
440 }
441
453 template <class T>
454 inline basic& write_str(_In_z_ const T* data)
455 {
456 // Stream state will be checked in write_data.
457 size_t num_chars = stdex::strlen(data);
458 if (num_chars > UINT32_MAX)
459 throw std::invalid_argument("string too long");
460 write_data((uint32_t)num_chars);
461 if (!ok()) _Unlikely_
462 return *this;
463 write_array(data, sizeof(T), num_chars);
464 return *this;
465 }
466
467#ifdef _WIN32
473 size_t write_sa(_In_ LPSAFEARRAY sa)
474 {
475 safearray_accessor<void> a(sa);
476 long ubound, lbound;
477 if (FAILED(SafeArrayGetUBound(sa, 1, &ubound)) ||
478 FAILED(SafeArrayGetLBound(sa, 1, &lbound)))
479 throw std::invalid_argument("SafeArrayGet[UL]Bound failed");
480 return write(a.data(), static_cast<size_t>(ubound) - lbound + 1);
481 }
482#endif
483
489 fsize_t write_stream(_Inout_ basic& stream, _In_ fsize_t amount = fsize_max)
490 {
491 std::unique_ptr<uint8_t[]> data(new uint8_t[static_cast<size_t>(std::min<fsize_t>(amount, default_block_size))]);
492 fsize_t num_copied = 0, to_write = amount;
493 m_state = state_t::ok;
494 while (to_write) {
495 size_t num_read = stream.read(data.get(), static_cast<size_t>(std::min<fsize_t>(default_block_size, to_write)));
496 size_t num_written = write(data.get(), num_read);
497 num_copied += num_written;
498 to_write -= num_written;
499 if (stream.m_state == state_t::eof) {
500 // EOF is not an error.
501 m_state = state_t::ok;
502 break;
503 }
504 m_state = stream.m_state;
505 if (!ok())
506 break;
507 }
508 return num_copied;
509 }
510
514 void write_charset(_In_ charset_id charset)
515 {
516 if (charset == charset_id::utf16)
517 write_data(utf16_bom);
518 else if (charset == charset_id::utf8)
519 write_array(utf8_bom, sizeof(utf8_bom), 1);
520 }
521
527 size_t write_sprintf(_In_z_ _Printf_format_string_params_(2) const char* format, _In_opt_ locale_t locale, ...)
528 {
529 va_list params;
530 va_start(params, locale);
531 size_t num_chars = write_vsprintf(format, locale, params);
532 va_end(params);
533 return num_chars;
534 }
535
541 size_t write_sprintf(_In_z_ _Printf_format_string_params_(2) const wchar_t* format, _In_opt_ locale_t locale, ...)
542 {
543 va_list params;
544 va_start(params, locale);
545 size_t num_chars = write_vsprintf(format, locale, params);
546 va_end(params);
547 return num_chars;
548 }
549
555 size_t write_vsprintf(_In_z_ _Printf_format_string_params_(2) const char* format, _In_opt_ locale_t locale, _In_ va_list params)
556 {
557 std::string str;
558 str.reserve(default_block_size);
559 vappendf(str, format, locale, params);
560 return write_array(str.data(), sizeof(char), str.size());
561 }
562
568 size_t write_vsprintf(_In_z_ _Printf_format_string_params_(2) const wchar_t* format, _In_opt_ locale_t locale, _In_ va_list params)
569 {
570 std::wstring str;
571 str.reserve(default_block_size);
572 vappendf(str, format, locale, params);
573 return write_array(str.data(), sizeof(wchar_t), str.size());
574 }
575
576 inline basic& operator >>(_Out_ int8_t& data) { return read_data(data); }
577 inline basic& operator <<(_In_ const int8_t data) { return write_data(data); }
578 inline basic& operator >>(_Out_ int16_t& data) { return read_data(data); }
579 inline basic& operator <<(_In_ const int16_t data) { return write_data(data); }
580 inline basic& operator >>(_Out_ int32_t& data) { return read_data(data); }
581 inline basic& operator <<(_In_ const int32_t data) { return write_data(data); }
582 inline basic& operator >>(_Out_ int64_t& data) { return read_data(data); }
583 inline basic& operator <<(_In_ const int64_t data) { return write_data(data); }
584 inline basic& operator >>(_Out_ uint8_t& data) { return read_data(data); }
585 inline basic& operator <<(_In_ const uint8_t data) { return write_data(data); }
586 inline basic& operator >>(_Out_ uint16_t& data) { return read_data(data); }
587 inline basic& operator <<(_In_ const uint16_t data) { return write_data(data); }
588 inline basic& operator >>(_Out_ uint32_t& data) { return read_data(data); }
589 inline basic& operator <<(_In_ const uint32_t data) { return write_data(data); }
590 inline basic& operator >>(_Out_ uint64_t& data) { return read_data(data); }
591 inline basic& operator <<(_In_ const uint64_t data) { return write_data(data); }
592#if defined(_WIN64) && defined(_NATIVE_SIZE_T_DEFINED)
593 inline basic& operator >>(_Out_ size_t& data) { return read_data(data); }
594 inline basic& operator <<(_In_ const size_t data) { return write_data(data); }
595#endif
596 inline basic& operator >>(_Out_ float& data) { return read_data(data); }
597 inline basic& operator <<(_In_ const float data) { return write_data(data); }
598 inline basic& operator >>(_Out_ double& data) { return read_data(data); }
599 inline basic& operator <<(_In_ const double data) { return write_data(data); }
600 inline basic& operator >>(_Out_ char& data) { return read_data(data); }
601 inline basic& operator <<(_In_ const char data) { return write_data(data); }
602#ifdef _NATIVE_WCHAR_T_DEFINED
603 inline basic& operator >>(_Out_ wchar_t& data) { return read_data(data); }
604 inline basic& operator <<(_In_ const wchar_t data) { return write_data(data); }
605#endif
606 template<class _Elem, class _Traits = std::char_traits<_Elem>, class _Ax = std::allocator<_Elem>>
607 inline basic& operator >>(_Inout_ std::basic_string<_Elem, _Traits, _Ax>& data) { return read_str(data); }
608 template <class T>
609 inline basic& operator <<(_In_ const T* data) { return write_str(data); }
610
611 protected:
612 state_t m_state;
613 };
614
618 using fpos_t = uint64_t;
619 constexpr fpos_t fpos_max = UINT64_MAX;
620 constexpr fpos_t fpos_min = 0;
621
625 using foff_t = int64_t;
626 constexpr foff_t foff_max = INT64_MAX;
627 constexpr foff_t foff_min = INT64_MIN;
628
632 enum class seek_t {
633#ifdef _WIN32
634 beg = FILE_BEGIN,
635 cur = FILE_CURRENT,
636 end = FILE_END
637#else
638 beg = SEEK_SET,
639 cur = SEEK_CUR,
640 end = SEEK_END
641#endif
642 };
643
644#if _HAS_CXX20
645 using time_point = std::chrono::time_point<std::chrono::file_clock>;
646#else
647 using time_point = std::chrono::time_point<std::chrono::system_clock>;
648#endif
649
653 class basic_file : virtual public basic
654 {
655 public:
656 virtual std::vector<uint8_t> read_remainder(_In_ size_t max_length = SIZE_MAX)
657 {
658 size_t length = std::min<size_t>(max_length, static_cast<size_t>(size() - tell()));
659 std::vector<uint8_t> result;
660 try { result.resize(length); }
661 catch (std::bad_alloc) {
662 m_state = state_t::fail;
663 return result;
664 }
665 result.resize(read_array(result.data(), sizeof(uint8_t), length));
666 return result;
667 }
668
674 virtual fpos_t seek(_In_ foff_t offset, _In_ seek_t how = seek_t::beg) = 0;
675
681 inline fpos_t seekbeg(_In_ fpos_t offset) { return seek(offset, seek_t::beg); }
682
688 inline fpos_t seekcur(_In_ foff_t offset) { return seek(offset, seek_t::cur); }
689
695 inline fpos_t seekend(_In_ foff_t offset) { return seek(offset, seek_t::end); }
696
697 virtual void skip(_In_ fsize_t amount)
698 {
699 seek(amount, seek_t::cur);
700 }
701
708 virtual fpos_t tell() const = 0;
709
713 virtual void lock(_In_ fpos_t offset, _In_ fsize_t length)
714 {
715 UNREFERENCED_PARAMETER(offset);
716 UNREFERENCED_PARAMETER(length);
717 throw std::exception("not implemented");
718 }
719
723 virtual void unlock(_In_ fpos_t offset, _In_ fsize_t length)
724 {
725 UNREFERENCED_PARAMETER(offset);
726 UNREFERENCED_PARAMETER(length);
727 throw std::exception("not implemented");
728 }
729
734 virtual fsize_t size() = 0;
735
739 virtual void truncate() = 0;
740
744 virtual time_point ctime() const
745 {
746 return time_point::min();
747 }
748
752 virtual time_point atime() const
753 {
754 return time_point::min();
755 }
756
760 virtual time_point mtime() const
761 {
762 return time_point::min();
763 }
764
768 virtual void set_ctime(time_point date)
769 {
770 UNREFERENCED_PARAMETER(date);
771 throw std::exception("not implemented");
772 }
773
777 virtual void set_atime(time_point date)
778 {
779 UNREFERENCED_PARAMETER(date);
780 throw std::exception("not implemented");
781 }
782
786 virtual void set_mtime(time_point date)
787 {
788 UNREFERENCED_PARAMETER(date);
789 throw std::exception("not implemented");
790 }
791
792#ifdef _WIN32
796 LPSAFEARRAY read_sa()
797 {
798 assert(size() <= SIZE_MAX);
799 size_t length = static_cast<size_t>(size());
800 std::unique_ptr<SAFEARRAY, SafeArrayDestroy_delete> sa(SafeArrayCreateVector(VT_UI1, 0, (ULONG)length));
801 if (!sa)
802 throw std::runtime_error("SafeArrayCreateVector failed");
803 safearray_accessor<void> a(sa.get());
804 if (seek(0) != 0)
805 throw std::runtime_error("failed to seek");
806 if (read_array(a.data(), 1, length) != length)
807 throw std::runtime_error("failed to read");
808 return sa.release();
809 }
810#endif
811
817 charset_id read_charset(_In_ charset_id default_charset = charset_id::default)
818 {
819 if (seek(0) != 0)
820 throw std::runtime_error("failed to seek");
821 wchar_t id_utf16;
822 read_array(&id_utf16, sizeof(wchar_t), 1);
823 if (!ok()) _Unlikely_
824 return default_charset;
825 if (id_utf16 == utf16_bom)
826 return charset_id::utf16;
827
828 if (seek(0) != 0)
829 throw std::runtime_error("failed to seek");
830 char id_utf8[3] = { 0 };
831 read_array(id_utf8, sizeof(id_utf8), 1);
832 if (!ok()) _Unlikely_
833 return default_charset;
834 if (strncmp(id_utf8, _countof(id_utf8), utf8_bom, _countof(utf8_bom)) == 0)
835 return charset_id::utf8;
836
837 if (seek(0) != 0)
838 throw std::runtime_error("failed to seek");
839 return default_charset;
840 }
841 };
842
846 class converter : public basic
847 {
848 protected:
849 explicit converter() :
850 basic(state_t::fail),
851 m_source(nullptr)
852 {}
853
854 void init(_Inout_ basic& source)
855 {
856 m_state = source.state();
857 m_source = &source;
858 }
859
860 public:
861 converter(_Inout_ basic& source) :
862 basic(source.state()),
863 m_source(&source)
864 {}
865
866 virtual _Success_(return != 0 || length == 0) size_t read(
867 _Out_writes_bytes_to_opt_(length, return) void* data, _In_ size_t length)
868 {
869 size_t num_read = m_source->read(data, length);
870 m_state = m_source->state();
871 return num_read;
872 }
873
874 virtual _Success_(return != 0) size_t write(
875 _In_reads_bytes_opt_(length) const void* data, _In_ size_t length)
876 {
877 size_t num_written = m_source->write(data, length);
878 m_state = m_source->state();
879 return num_written;
880 }
881
882 virtual void close()
883 {
884 m_source->close();
885 m_state = m_source->state();
886 }
887
888 virtual void flush()
889 {
890 m_source->flush();
891 m_state = m_source->state();
892 }
893
894 protected:
895 basic* m_source;
896 };
897
901 class replicator : public basic
902 {
903 public:
904 virtual ~replicator()
905 {
906 for (auto w = m_workers.begin(), w_end = m_workers.end(); w != w_end; ++w) {
907 auto _w = w->get();
908 {
909 const std::lock_guard<std::mutex> lk(_w->mutex);
910 _w->op = worker::op_t::quit;
911 }
912 _w->cv.notify_one();
913 }
914 for (auto w = m_workers.begin(), w_end = m_workers.end(); w != w_end; ++w)
915 w->get()->thread.join();
916 }
917
921 void push_back(_In_ basic* source)
922 {
923 m_workers.push_back(std::unique_ptr<worker>(new worker(source)));
924 }
925
929 void remove(basic* source)
930 {
931 for (auto w = m_workers.begin(), w_end = m_workers.end(); w != w_end; ++w) {
932 auto _w = w->get();
933 if (_w->source == source) {
934 {
935 const std::lock_guard<std::mutex> lk(_w->mutex);
936 _w->op = worker::op_t::quit;
937 }
938 _w->cv.notify_one();
939 _w->thread.join();
940 m_workers.erase(w);
941 return;
942 }
943 }
944 }
945
946 virtual _Success_(return != 0) size_t write(
947 _In_reads_bytes_opt_(length) const void* data, _In_ size_t length)
948 {
949 for (auto w = m_workers.begin(), w_end = m_workers.end(); w != w_end; ++w) {
950 auto _w = w->get();
951 {
952 const std::lock_guard<std::mutex> lk(_w->mutex);
953 _w->op = worker::op_t::write;
954 _w->data = data;
955 _w->length = length;
956 }
957 _w->cv.notify_one();
958 }
959 size_t num_written = length;
960 m_state = state_t::ok;
961 for (auto w = m_workers.begin(), w_end = m_workers.end(); w != w_end; ++w) {
962 auto _w = w->get();
963 std::unique_lock<std::mutex> lk(_w->mutex);
964 _w->cv.wait(lk, [&] {return _w->op == worker::op_t::noop; });
965 if (_w->num_written < num_written)
966 num_written = _w->num_written;
967 if (ok() && !_w->source->ok())
968 m_state = _w->source->state();
969 }
970 return num_written;
971 }
972
973 virtual void close()
974 {
975 foreach_worker(worker::op_t::close);
976 }
977
978 virtual void flush()
979 {
980 foreach_worker(worker::op_t::flush);
981 }
982
983 protected:
984 class worker
985 {
986 public:
987 worker(_In_ basic* _source) :
988 source(_source),
989 op(op_t::noop),
990 data(nullptr),
991 length(0),
992 num_written(0),
993 thread(process_op, std::ref(*this))
994 {}
995
996 protected:
997 static void process_op(_Inout_ worker& w)
998 {
999 for (;;) {
1000 std::unique_lock<std::mutex> lk(w.mutex);
1001 w.cv.wait(lk, [&] {return w.op != op_t::noop; });
1002 switch (w.op) {
1003 case op_t::quit:
1004 return;
1005 case op_t::write:
1006 w.num_written = w.source->write(w.data, w.length);
1007 break;
1008 case op_t::close:
1009 w.source->close();
1010 break;
1011 case op_t::flush:
1012 w.source->flush();
1013 break;
1014 }
1015 w.op = op_t::noop;
1016 lk.unlock();
1017 w.cv.notify_one();
1018 }
1019 }
1020
1021 public:
1022 basic* source;
1023 enum class op_t {
1024 noop = 0,
1025 quit,
1026 write,
1027 close,
1028 flush,
1029 } op;
1030 const void* data;
1031 size_t length;
1033 std::mutex mutex;
1034 std::condition_variable cv;
1035 std::thread thread;
1036 };
1037
1038 void foreach_worker(_In_ worker::op_t op)
1039 {
1040 for (auto w = m_workers.begin(), w_end = m_workers.end(); w != w_end; ++w) {
1041 auto _w = w->get();
1042 {
1043 const std::lock_guard<std::mutex> lk(_w->mutex);
1044 _w->op = op;
1045 }
1046 _w->cv.notify_one();
1047 }
1048 m_state = state_t::ok;
1049 for (auto w = m_workers.begin(), w_end = m_workers.end(); w != w_end; ++w) {
1050 auto _w = w->get();
1051 std::unique_lock<std::mutex> lk(_w->mutex);
1052 _w->cv.wait(lk, [&] {return _w->op == worker::op_t::noop; });
1053 if (ok())
1054 m_state = _w->source->state();
1055 }
1056 }
1057
1058 std::list<std::unique_ptr<worker>> m_workers;
1059 };
1060
1061 constexpr size_t default_async_limit = 0x100000;
1062
1068 template <size_t CAPACITY = default_async_limit>
1070 {
1071 public:
1072 async_reader(_Inout_ basic& source) :
1073 converter(source),
1074 m_worker(process, std::ref(*this))
1075 {}
1076
1077 virtual ~async_reader()
1078 {
1079 m_ring.quit();
1080 m_worker.join();
1081 }
1082
1083#pragma warning(suppress: 6101) // See [1] below
1084 virtual _Success_(return != 0 || length == 0) size_t read(
1085 _Out_writes_bytes_to_opt_(length, return) void* data, _In_ size_t length)
1086 {
1087 assert(data || !length);
1088 for (size_t to_read = length;;) {
1089 uint8_t* ptr; size_t num_read;
1090 std::tie(ptr, num_read) = m_ring.front();
1091 if (!ptr) _Unlikely_ {
1092 // [1] Code analysis misses length - to_read bytes were written to data in previous loop iterations.
1093 m_state = to_read < length || !length ? state_t::ok : m_source->state();
1094 return length - to_read;
1095 }
1096 if (to_read < num_read)
1097 num_read = to_read;
1098 memcpy(data, ptr, num_read);
1099 m_ring.pop(num_read);
1100 to_read -= num_read;
1101 if (!to_read) {
1102 m_state = state_t::ok;
1103 return length;
1104 }
1105 reinterpret_cast<uint8_t*&>(data) += num_read;
1106 }
1107 }
1108
1109 protected:
1110 static void process(_Inout_ async_reader& w)
1111 {
1112 for (;;) {
1113 uint8_t* ptr; size_t num_write;
1114 std::tie(ptr, num_write) = w.m_ring.back();
1115 if (!ptr) _Unlikely_
1116 break;
1117 num_write = w.m_source->read(ptr, num_write);
1118 w.m_ring.push(num_write);
1119 if (!w.m_source->ok()) {
1120 w.m_ring.quit();
1121 break;
1122 }
1123 }
1124 }
1125
1126 protected:
1127 ring<uint8_t, CAPACITY> m_ring;
1128 std::thread m_worker;
1129 };
1130
1136 template <size_t CAPACITY = default_async_limit>
1138 {
1139 public:
1140 async_writer(_Inout_ basic& source) :
1141 converter(source),
1142 m_worker(process, std::ref(*this))
1143 {}
1144
1145 virtual ~async_writer()
1146 {
1147 m_ring.quit();
1148 m_worker.join();
1149 }
1150
1151 virtual _Success_(return != 0) size_t write(
1152 _In_reads_bytes_opt_(length) const void* data, _In_ size_t length)
1153 {
1154 assert(data || !length);
1155 for (size_t to_write = length;;) {
1156 uint8_t* ptr; size_t num_write;
1157 std::tie(ptr, num_write) = m_ring.back();
1158 if (!ptr) _Unlikely_ {
1159 m_state = state_t::fail;
1160 return length - to_write;
1161 }
1162 if (to_write < num_write)
1163 num_write = to_write;
1164 memcpy(ptr, data, num_write);
1165 m_ring.push(num_write);
1166 to_write -= num_write;
1167 if (!to_write) {
1168 m_state = state_t::ok;
1169 return length;
1170 }
1171 reinterpret_cast<const uint8_t*&>(data) += num_write;
1172 }
1173 }
1174
1175 virtual void flush()
1176 {
1177 m_ring.sync();
1179 }
1180
1181 protected:
1182 static void process(_Inout_ async_writer& w)
1183 {
1184 for (;;) {
1185 uint8_t* ptr; size_t num_read;
1186 std::tie(ptr, num_read) = w.m_ring.front();
1187 if (!ptr)
1188 break;
1189 num_read = w.m_source->write(ptr, num_read);
1190 w.m_ring.pop(num_read);
1191 if (!w.m_source->ok()) {
1192 w.m_ring.quit();
1193 break;
1194 }
1195 }
1196 }
1197
1198 protected:
1199 ring<uint8_t, CAPACITY> m_ring;
1200 std::thread m_worker;
1201 };
1202
1203 constexpr size_t default_buffer_size = 0x400;
1204
1208 class buffer : public converter
1209 {
1210 protected:
1211 explicit buffer(_In_ size_t read_buffer_size = default_buffer_size, _In_ size_t write_buffer_size = default_buffer_size) :
1212 converter(),
1213 m_read_buffer(read_buffer_size),
1214 m_write_buffer(write_buffer_size)
1215 {}
1216
1217 public:
1218 buffer(_Inout_ basic& source, _In_ size_t read_buffer_size = default_buffer_size, _In_ size_t write_buffer_size = default_buffer_size) :
1219 converter(source),
1220 m_read_buffer(read_buffer_size),
1221 m_write_buffer(write_buffer_size)
1222 {}
1223
1224 virtual ~buffer()
1225 {
1226 if (m_source)
1227 flush_write();
1228 }
1229
1230 virtual _Success_(return != 0 || length == 0) size_t read(
1231 _Out_writes_bytes_to_opt_(length, return) void* data, _In_ size_t length)
1232 {
1233 assert(data || !length);
1234 for (size_t to_read = length;;) {
1235 size_t buffer_size = m_read_buffer.tail - m_read_buffer.head;
1236 if (to_read <= buffer_size) {
1237 memcpy(data, m_read_buffer.data + m_read_buffer.head, to_read);
1238 m_read_buffer.head += to_read;
1239 m_state = state_t::ok;
1240 return length;
1241 }
1242 if (buffer_size) {
1243 memcpy(data, m_read_buffer.data + m_read_buffer.head, buffer_size);
1244 reinterpret_cast<uint8_t*&>(data) += buffer_size;
1245 to_read -= buffer_size;
1246 }
1247 m_read_buffer.head = 0;
1248 if (to_read > m_read_buffer.capacity) {
1249 // When needing to read more data than buffer capacity, bypass the buffer.
1250 m_read_buffer.tail = 0;
1251 to_read -= m_source->read(data, to_read);
1252 m_state = to_read < length ? state_t::ok : m_source->state();
1253 return length - to_read;
1254 }
1255 m_read_buffer.tail = m_source->read(m_read_buffer.data, m_read_buffer.capacity);
1256 if (m_read_buffer.tail < m_read_buffer.capacity && m_read_buffer.tail < to_read) _Unlikely_ {
1257 memcpy(data, m_read_buffer.data, m_read_buffer.tail);
1258 m_read_buffer.head = m_read_buffer.tail;
1259 to_read -= m_read_buffer.tail;
1260 m_state = to_read < length ? state_t::ok : m_source->state();
1261 return length - to_read;
1262 }
1263 }
1264 }
1265
1266 virtual _Success_(return != 0) size_t write(
1267 _In_reads_bytes_opt_(length) const void* data, _In_ size_t length)
1268 {
1269 assert(data || !length);
1270 if (!length) _Unlikely_ {
1271 // Pass null writes (zero-byte length). Null write operations have special meaning with with Windows pipes.
1272 flush_write();
1273 if (!ok()) _Unlikely_
1274 return 0;
1275 m_source->write(nullptr, 0);
1276 m_state = m_source->state();
1277 return 0;
1278 }
1279
1280 for (size_t to_write = length;;) {
1281 size_t available_buffer = m_write_buffer.capacity - m_write_buffer.tail;
1282 if (to_write <= available_buffer) {
1283 memcpy(m_write_buffer.data + m_write_buffer.tail, data, to_write);
1284 m_write_buffer.tail += to_write;
1285 m_state = state_t::ok;
1286 return length;
1287 }
1288 if (available_buffer) {
1289 memcpy(m_write_buffer.data + m_write_buffer.tail, data, available_buffer);
1290 reinterpret_cast<const uint8_t*&>(data) += available_buffer;
1291 to_write -= available_buffer;
1292 m_write_buffer.tail += available_buffer;
1293 }
1294 size_t buffer_size = m_write_buffer.tail - m_write_buffer.head;
1295 if (buffer_size) {
1296 m_write_buffer.head += m_source->write(m_write_buffer.data + m_write_buffer.head, buffer_size);
1297 m_state = m_source->state();
1298 if (m_write_buffer.head == m_write_buffer.tail)
1299 m_write_buffer.head = m_write_buffer.tail = 0;
1300 else
1301 return length - to_write;
1302 }
1303 if (to_write > m_write_buffer.capacity) {
1304 // When needing to write more data than buffer capacity, bypass the buffer.
1305 to_write -= m_source->write(data, to_write);
1306 m_state = m_source->state();
1307 return length - to_write;
1308 }
1309 }
1310 }
1311
1312 virtual void flush()
1313 {
1314 flush_write();
1315 if (ok())
1317 }
1318
1319 protected:
1320 void flush_write()
1321 {
1322 size_t buffer_size = m_write_buffer.tail - m_write_buffer.head;
1323 if (buffer_size) {
1324 m_write_buffer.head += m_source->write(m_write_buffer.data + m_write_buffer.head, buffer_size);
1325 if (m_write_buffer.head == m_write_buffer.tail) {
1326 m_write_buffer.head = 0;
1327 m_write_buffer.tail = 0;
1328 }
1329 else {
1330 m_state = m_source->state();
1331 return;
1332 }
1333 }
1334 m_state = state_t::ok;
1335 }
1336
1337 struct buffer_t {
1338 uint8_t* data;
1339 size_t head, tail, capacity;
1340
1341 buffer_t(_In_ size_t buffer_size) :
1342 head(0),
1343 tail(0),
1344 capacity(buffer_size),
1345 data(buffer_size ? new uint8_t[buffer_size] : nullptr)
1346 {}
1347
1348 ~buffer_t()
1349 {
1350 if (data)
1351 delete[] data;
1352 }
1353 } m_read_buffer, m_write_buffer;
1354 };
1355
1359 class limiter : public converter
1360 {
1361 public:
1362 limiter(_Inout_ basic& source, _In_ fsize_t _read_limit = 0, _In_ fsize_t _write_limit = 0) :
1363 converter(source),
1364 read_limit(_read_limit),
1365 write_limit(_write_limit)
1366 {}
1367
1368 virtual _Success_(return != 0 || length == 0) size_t read(
1369 _Out_writes_bytes_to_opt_(length, return) void* data, _In_ size_t length)
1370 {
1371 size_t num_read;
1372 if (read_limit == fsize_max) {
1373 num_read = m_source->read(data, length);
1374 m_state = m_source->state();
1375 }
1376 else if (length <= read_limit) {
1377 num_read = m_source->read(data, length);
1378 m_state = m_source->state();
1379 read_limit -= num_read;
1380 }
1381 else if (length && !read_limit) {
1382 num_read = 0;
1383 m_state = state_t::eof;
1384 }
1385 else {
1386 num_read = m_source->read(data, static_cast<size_t>(read_limit));
1387 m_state = m_source->state();
1388 read_limit -= num_read;
1389 }
1390 return num_read;
1391 }
1392
1393 virtual _Success_(return != 0) size_t write(
1394 _In_reads_bytes_opt_(length) const void* data, _In_ size_t length)
1395 {
1396 size_t num_written;
1397 if (write_limit == fsize_max) {
1398 num_written = m_source->write(data, length);
1399 m_state = m_source->state();
1400 }
1401 else if (length <= write_limit) {
1402 num_written = m_source->write(data, length);
1403 m_state = m_source->state();
1404 write_limit -= num_written;
1405 }
1406 else if (length && !write_limit) {
1407 num_written = 0;
1408 m_state = state_t::fail;
1409 }
1410 else {
1411 num_written = m_source->write(data, static_cast<size_t>(write_limit));
1412 m_state = m_source->state();
1413 write_limit -= num_written;
1414 }
1415 return num_written;
1416 }
1417
1418 public:
1419 fsize_t
1422 };
1423
1427 class window : public limiter
1428 {
1429 public:
1430 window(_Inout_ basic& source, _In_ fpos_t _read_offset = 0, _In_ fsize_t read_limit = fsize_max, _In_ fpos_t _write_offset = 0, _In_ fsize_t write_limit = fsize_max) :
1431 limiter(source, read_limit, write_limit),
1432 read_offset(_read_offset),
1433 write_offset(_write_offset)
1434 {}
1435
1436 virtual _Success_(return != 0 || length == 0) size_t read(
1437 _Out_writes_bytes_to_opt_(length, return) void* data, _In_ size_t length)
1438 {
1439 if (read_offset) {
1440 m_source->skip(read_offset);
1441 m_state = m_source->state();
1442 if (!ok()) _Unlikely_
1443 return 0;
1444 read_offset = 0;
1445 }
1446 size_t num_read;
1447 if (read_limit == fsize_max) {
1448 num_read = m_source->read(data, length);
1449 m_state = m_source->state();
1450 }
1451 else if (length <= read_limit) {
1452 num_read = m_source->read(data, length);
1453 m_state = m_source->state();
1454 read_limit -= num_read;
1455 }
1456 else if (length && !read_limit) {
1457 num_read = 0;
1458 m_source->skip(length);
1459 m_state = state_t::eof;
1460 }
1461 else {
1462 num_read = m_source->read(data, static_cast<size_t>(read_limit));
1463 m_state = m_source->state();
1464 read_limit -= num_read;
1465 }
1466 return num_read;
1467 }
1468
1469 virtual _Success_(return != 0) size_t write(
1470 _In_reads_bytes_opt_(length) const void* data, _In_ size_t length)
1471 {
1472 size_t num_skipped, num_written;
1473 if (length <= write_offset) {
1474 write_offset -= length;
1475 m_state = state_t::ok;
1476 return length;
1477 }
1478 if (write_offset) {
1479 reinterpret_cast<const uint8_t*&>(data) += static_cast<size_t>(write_offset);
1480 length -= static_cast<size_t>(write_offset);
1481 num_skipped = static_cast<size_t>(write_offset);
1482 write_offset = 0;
1483 }
1484 else
1485 num_skipped = 0;
1486 if (write_limit == fsize_max) {
1487 num_written = m_source->write(data, length);
1488 m_state = m_source->state();
1489 }
1490 else if (length <= write_limit) {
1491 num_written = m_source->write(data, length);
1492 m_state = m_source->state();
1493 write_limit -= num_written;
1494 }
1495 else if (length && !write_limit) {
1496 num_skipped += length;
1497 num_written = 0;
1498 m_state = state_t::ok;
1499 }
1500 else {
1501 num_skipped += length - static_cast<size_t>(write_limit);
1502 num_written = m_source->write(data, static_cast<size_t>(write_limit));
1503 m_state = m_source->state();
1504 write_limit -= num_written;
1505 }
1506 return num_skipped + num_written;
1507 }
1508
1509 public:
1510 fpos_t
1513 };
1514
1519 {
1520 public:
1521 file_window(_Inout_ basic_file& source, fpos_t offset = 0, fsize_t length = 0) :
1522 basic(source.state()),
1523 m_source(source),
1524 m_offset(source.tell()),
1525 m_region(offset, offset + length)
1526 {}
1527
1528 virtual _Success_(return != 0 || length == 0) size_t read(
1529 _Out_writes_bytes_to_opt_(length, return) void* data, _In_ size_t length)
1530 {
1531 assert(data || !length);
1532 if (m_region.contains(m_offset)) {
1533 size_t num_read = m_source.read(data, static_cast<size_t>(std::min<fpos_t>(length, m_region.end - m_offset)));
1534 m_state = m_source.state();
1535 m_offset += num_read;
1536 return num_read;
1537 }
1538 m_state = length ? state_t::eof : state_t::ok;
1539 return 0;
1540 }
1541
1542 virtual _Success_(return != 0) size_t write(
1543 _In_reads_bytes_opt_(length) const void* data, _In_ size_t length)
1544 {
1545 assert(data || !length);
1546 if (m_region.contains(m_offset)) {
1547 size_t num_written = m_source.write(data, static_cast<size_t>(std::min<fpos_t>(length, m_region.end - m_offset)));
1548 m_state = m_source.state();
1549 m_offset += num_written;
1550 return num_written;
1551 }
1552 m_state = state_t::fail;
1553 return 0;
1554 }
1555
1556 virtual void close()
1557 {
1558 m_source.close();
1559 m_state = m_source.state();
1560 }
1561
1562 virtual void flush()
1563 {
1564 m_source.flush();
1565 m_state = m_source.state();
1566 }
1567
1568 virtual fpos_t seek(_In_ foff_t offset, _In_ seek_t how = seek_t::beg)
1569 {
1570 m_offset = m_source.seek(offset, how);
1571 m_state = m_source.state();
1572 return ok() ? m_offset - m_region.start : fpos_max;
1573 }
1574
1575 virtual void skip(_In_ fsize_t amount)
1576 {
1577 m_source.skip(amount);
1578 m_state = m_source.state();
1579 }
1580
1581 virtual fpos_t tell() const
1582 {
1583 fpos_t offset = m_source.tell();
1584 return m_region.contains(offset) ? offset - m_region.start : fpos_max;
1585 }
1586
1587 virtual void lock(_In_ fpos_t offset, _In_ fsize_t length)
1588 {
1589 if (m_region.contains(offset)) {
1590 m_source.lock(m_region.start + offset, std::min<fsize_t>(length, m_region.end - offset));
1591 m_state = m_source.state();
1592 }
1593 else
1594 m_state = state_t::fail;
1595 }
1596
1597 virtual void unlock(_In_ fpos_t offset, _In_ fsize_t length)
1598 {
1599 if (m_region.contains(offset)) {
1600 m_source.unlock(m_region.start + offset, std::min<fsize_t>(length, m_region.end - offset));
1601 m_state = m_source.state();
1602 }
1603 else
1604 m_state = state_t::fail;
1605 }
1606
1607 virtual fsize_t size()
1608 {
1609 return m_region.size();
1610 }
1611
1612 virtual void truncate()
1613 {
1614 m_state = state_t::fail;
1615 }
1616
1617 protected:
1618 basic_file& m_source;
1619 fpos_t m_offset;
1620 interval<fpos_t> m_region;
1621 };
1622
1623 constexpr size_t default_cache_size = 0x1000;
1624
1628 class cache : public basic_file
1629 {
1630 protected:
1631 explicit cache(_In_ size_t cache_size = default_cache_size) :
1632 basic(state_t::fail),
1633 m_source(nullptr),
1634 m_cache(cache_size),
1635 m_offset(0)
1636#if SET_FILE_OP_TIMES
1637 , m_atime(time_point::min()),
1638 m_mtime(time_point::min())
1639#endif
1640 {}
1641
1642 void init(_Inout_ basic_file& source)
1643 {
1644 m_state = source.state();
1645 m_source = &source;
1646 m_offset = source.tell();
1647 }
1648
1649 public:
1650 cache(_Inout_ basic_file& source, _In_ size_t cache_size = default_cache_size) :
1651 basic(source.state()),
1652 m_source(&source),
1653 m_cache(cache_size),
1654 m_offset(source.tell())
1655#if SET_FILE_OP_TIMES
1656 , m_atime(time_point::min()),
1657 m_mtime(time_point::min())
1658#endif
1659 {}
1660
1661 virtual ~cache()
1662 {
1663 if (m_source) {
1664 flush_cache();
1665 m_source->seek(m_offset);
1666 }
1667 }
1668
1669 virtual _Success_(return != 0 || length == 0) size_t read(
1670 _Out_writes_bytes_to_opt_(length, return) void* data, _In_ size_t length)
1671 {
1672 assert(data || !length);
1673#if SET_FILE_OP_TIMES
1674 m_atime = time_point::now();
1675#endif
1676 for (size_t to_read = length;;) {
1677 if (m_cache.status != cache_t::cache_t::status_t::empty) {
1678 if (m_cache.region.contains(m_offset)) {
1679 size_t remaining_cache = static_cast<size_t>(m_cache.region.end - m_offset);
1680 if (to_read <= remaining_cache) {
1681 memcpy(data, m_cache.data + static_cast<size_t>(m_offset - m_cache.region.start), to_read);
1682 m_offset += to_read;
1683 m_state = state_t::ok;
1684 return length;
1685 }
1686 memcpy(data, m_cache.data + static_cast<size_t>(m_offset - m_cache.region.start), remaining_cache);
1687 reinterpret_cast<uint8_t*&>(data) += remaining_cache;
1688 to_read -= remaining_cache;
1689 m_offset += remaining_cache;
1690 }
1691 flush_cache();
1692 if (!ok()) _Unlikely_ {
1693 if (to_read < length)
1694 m_state = state_t::ok;
1695 return length - to_read;
1696 }
1697 }
1698 {
1699 fpos_t end_max = m_offset + to_read;
1700 if (m_offset / m_cache.capacity < end_max / m_cache.capacity) {
1701 // Read spans multiple cache blocks. Bypass cache to the last block.
1702 m_source->seek(m_offset);
1703 if (!m_source->ok()) _Unlikely_ {
1704 m_state = to_read < length ? state_t::ok : state_t::fail;
1705 return length - to_read;
1706 }
1707 size_t num_read = m_source->read(data, to_read - static_cast<size_t>(end_max % m_cache.capacity));
1708 m_offset += num_read;
1709 to_read -= num_read;
1710 if (!to_read) {
1711 m_state = state_t::ok;
1712 return length;
1713 }
1714 reinterpret_cast<uint8_t*&>(data) += num_read;
1715 m_state = m_source->state();
1716 if (!ok()) {
1717 if (to_read < length)
1718 m_state = state_t::ok;
1719 return length - to_read;
1720 }
1721 }
1722 }
1723 load_cache(m_offset);
1724 if (!ok() || m_cache.region.end <= m_offset) _Unlikely_ {
1725 m_state = to_read < length ? state_t::ok : state_t::fail;
1726 return length - to_read;
1727 }
1728 }
1729 }
1730
1731 virtual _Success_(return != 0) size_t write(
1732 _In_reads_bytes_opt_(length) const void* data, _In_ size_t length)
1733 {
1734 assert(data || !length);
1735#if SET_FILE_OP_TIMES
1736 m_atime = m_mtime = time_point::now();
1737#endif
1738 for (size_t to_write = length;;) {
1739 if (m_cache.status != cache_t::cache_t::status_t::empty) {
1740 fpos_t end_max = m_cache.region.start + m_cache.capacity;
1741 if (m_cache.region.start <= m_offset && m_offset < end_max) {
1742 size_t remaining_cache = static_cast<size_t>(end_max - m_offset);
1743 if (to_write <= remaining_cache) {
1744 memcpy(m_cache.data + static_cast<size_t>(m_offset - m_cache.region.start), data, to_write);
1745 m_offset += to_write;
1746 m_cache.status = cache_t::cache_t::status_t::dirty;
1747 m_cache.region.end = std::max(m_cache.region.end, m_offset);
1748 m_state = state_t::ok;
1749 return length;
1750 }
1751 memcpy(m_cache.data + static_cast<size_t>(m_offset - m_cache.region.start), data, remaining_cache);
1752 reinterpret_cast<const uint8_t*&>(data) += remaining_cache;
1753 to_write -= remaining_cache;
1754 m_offset += remaining_cache;
1755 m_cache.status = cache_t::cache_t::status_t::dirty;
1756 m_cache.region.end = end_max;
1757 }
1758 flush_cache();
1759 if (!ok()) _Unlikely_
1760 return length - to_write;
1761 }
1762 {
1763 fpos_t end_max = m_offset + to_write;
1764 if (m_offset / m_cache.capacity < end_max / m_cache.capacity) {
1765 // Write spans multiple cache blocks. Bypass cache to the last block.
1766 m_source->seek(m_offset);
1767 if (!ok()) _Unlikely_
1768 return length - to_write;
1769 size_t num_written = m_source->write(data, to_write - static_cast<size_t>(end_max % m_cache.capacity));
1770 reinterpret_cast<const uint8_t*&>(data) += num_written;
1771 to_write -= num_written;
1772 m_offset += num_written;
1773 m_state = m_source->state();
1774 if (!to_write || !ok())
1775 return length - to_write;
1776 }
1777 }
1778 load_cache(m_offset);
1779 if (!ok()) _Unlikely_
1780 return length - to_write;
1781 }
1782 }
1783
1784 virtual void close()
1785 {
1786 flush_cache();
1787 m_source->close();
1788 m_state = m_source->state();
1789 }
1790
1791 virtual void flush()
1792 {
1793#if SET_FILE_OP_TIMES
1794 m_atime = m_mtime = time_point::min();
1795#endif
1796 flush_cache();
1797 if (!ok()) _Unlikely_
1798 return;
1799 m_source->flush();
1800 }
1801
1802 virtual fpos_t seek(_In_ foff_t offset, _In_ seek_t how = seek_t::beg)
1803 {
1804 m_state = state_t::ok;
1805 switch (how) {
1806 case seek_t::beg:
1807 return m_offset = offset;
1808 case seek_t::cur:
1809 return m_offset += offset;
1810 case seek_t::end:
1811 return m_offset = size() + offset;
1812 default:
1813 throw std::invalid_argument("unknown seek origin");
1814 }
1815 }
1816
1817 virtual fpos_t tell() const
1818 {
1819 return m_offset;
1820 }
1821
1822 virtual void lock(_In_ fpos_t offset, _In_ fsize_t length)
1823 {
1824 m_source->lock(offset, length);
1825 m_state = m_source->state();
1826 }
1827
1828 virtual void unlock(_In_ fpos_t offset, _In_ fsize_t length)
1829 {
1830 m_source->unlock(offset, length);
1831 m_state = m_source->state();
1832 }
1833
1834 virtual fsize_t size()
1835 {
1836 return m_cache.data ? std::max(m_source->size(), m_cache.region.end) : m_source->size();
1837 }
1838
1839 virtual void truncate()
1840 {
1841#if SET_FILE_OP_TIMES
1842 m_atime = m_mtime = time_point::now();
1843#endif
1844 m_source->seek(m_offset);
1845 if (m_cache.region.end <= m_offset) {
1846 // Truncation does not affect cache.
1847 }
1848 else if (m_cache.region.start <= m_offset) {
1849 // Truncation truncates cache.
1850 m_cache.region.end = m_offset;
1851 }
1852 else {
1853 // Truncation invalidates cache.
1854 m_cache.region = 0;
1855 m_cache.status = cache_t::cache_t::status_t::empty;
1856 }
1857 m_source->truncate();
1858 m_state = m_source->state();
1859 }
1860
1861 virtual time_point ctime() const
1862 {
1863 return m_source->ctime();
1864 }
1865
1866 virtual time_point atime() const
1867 {
1868#if SET_FILE_OP_TIMES
1869 return std::max(m_atime, m_source->atime());
1870#else
1871 return m_source->atime();
1872#endif
1873 }
1874
1875 virtual time_point mtime() const
1876 {
1877#if SET_FILE_OP_TIMES
1878 return std::max(m_mtime, m_source->mtime());
1879#else
1880 return m_source->mtime();
1881#endif
1882 }
1883
1884 virtual void set_ctime(time_point date)
1885 {
1886 m_source->set_ctime(date);
1887 }
1888
1889 virtual void set_atime(time_point date)
1890 {
1891#if SET_FILE_OP_TIMES
1892 m_atime = date;
1893#endif
1894 m_source->set_atime(date);
1895 }
1896
1897 virtual void set_mtime(time_point date)
1898 {
1899#if SET_FILE_OP_TIMES
1900 m_mtime = date;
1901#endif
1902 m_source->set_mtime(date);
1903 }
1904
1905 protected:
1906 void flush_cache()
1907 {
1908 if (m_cache.status != cache_t::cache_t::status_t::dirty) {
1909 m_state = state_t::ok;
1910 }
1911 else if (!m_cache.region.empty()) {
1912 m_source->seek(m_cache.region.start);
1913 m_source->write(m_cache.data, static_cast<size_t>(m_cache.region.size()));
1914 m_state = m_source->state();
1915 if (ok())
1916 m_cache.status = cache_t::cache_t::status_t::loaded;
1917 }
1918 else {
1919 m_state = state_t::ok;
1920 m_cache.status = cache_t::cache_t::status_t::loaded;
1921 }
1922 }
1923
1924 void invalidate_cache()
1925 {
1926 flush_cache();
1927 if (ok()) {
1928 m_cache.region = 0;
1929 m_cache.status = cache_t::cache_t::status_t::empty;
1930 }
1931 }
1932
1933 void load_cache(_In_ fpos_t start)
1934 {
1935 assert(m_cache.status != cache_t::cache_t::status_t::dirty);
1936 start -= start % m_cache.capacity; // Align to cache block size.
1937 m_source->seek(m_cache.region.start = start);
1938 if (m_source->ok()) {
1939 m_cache.region.end = start + m_source->read(m_cache.data, m_cache.capacity);
1940 m_cache.status = cache_t::cache_t::status_t::loaded;
1941 m_state = state_t::ok; // Regardless the read failure, we still might have cached some data.
1942 }
1943 else
1944 m_state = state_t::fail;
1945 }
1946
1947 basic_file* m_source;
1948 struct cache_t {
1949 uint8_t* data;
1950 size_t capacity;
1951 enum class status_t {
1952 empty = 0,
1953 loaded,
1954 dirty,
1955 } status;
1957
1958 cache_t(_In_ size_t _capacity) :
1959 data(new uint8_t[_capacity]),
1960 capacity(_capacity),
1961 status(status_t::empty),
1962 region(0)
1963 {}
1964
1965 ~cache_t()
1966 {
1967 delete[] data;
1968 }
1969 } m_cache;
1970 fpos_t m_offset;
1971#if SET_FILE_OP_TIMES
1972 time_point
1973 m_atime,
1974 m_mtime;
1975#endif
1976 };
1977
1981 class basic_sys : virtual public basic, public sys_object
1982 {
1983 public:
1984 basic_sys(_In_opt_ sys_handle h = invalid_handle, _In_ state_t state = state_t::ok) :
1985 basic(state),
1986 sys_object(h)
1987 {}
1988
1989 virtual _Success_(return != 0 || length == 0) size_t read(
1990 _Out_writes_bytes_to_opt_(length, return) void* data, _In_ size_t length)
1991 {
1992 assert(data || !length);
1993 // Windows Server 2003 and Windows XP: Pipe write operations across a network are limited in size per write.
1994 // The amount varies per platform. For x86 platforms it's 63.97 MB. For x64 platforms it's 31.97 MB. For Itanium
1995 // it's 63.95 MB. For more information regarding pipes, see the Remarks section.
1996 size_t
1997#if defined(_WIN64)
1998 block_size = 0x1F80000;
1999#elif defined(_WIN32)
2000 block_size = 0x3f00000;
2001#else
2002 block_size = SSIZE_MAX;
2003#endif
2004 for (size_t to_read = length;;) {
2005#ifdef _WIN32
2006 // ReadFile() might raise exception (e.g. STATUS_FILE_BAD_FORMAT/0xE0000002).
2007 BOOL succeeded;
2008 DWORD num_read;
2009 __try { succeeded = ReadFile(m_h, data, static_cast<DWORD>(std::min<size_t>(to_read, block_size)), &num_read, nullptr); }
2010 __except (EXCEPTION_EXECUTE_HANDLER) { succeeded = FALSE; SetLastError(ERROR_UNHANDLED_EXCEPTION); num_read = 0; }
2011 if (!succeeded && GetLastError() == ERROR_NO_SYSTEM_RESOURCES && block_size > default_block_size) _Unlikely_ {
2012 // Error "Insufficient system resources exist to complete the requested service." occurs
2013 // ocasionally, when attempting to read too much data at once (e.g. over \\TSClient).
2014 block_size = default_block_size;
2015 continue;
2016 }
2017 if (!succeeded) _Unlikely_
2018#else
2019 ssize_t num_read = static_cast<ssize_t>(std::min<size_t>(to_read, block_size));
2020 num_read = read(m_h, data, num_read);
2021 if (num_read < 0) _Unlikely_
2022#endif
2023 {
2024 m_state = to_read < length ? state_t::ok : state_t::fail;
2025 return length - to_read;
2026 }
2027 if (!num_read) _Unlikely_ {
2028 m_state = to_read < length || !length ? state_t::ok : state_t::eof;
2029 return length - to_read;
2030 }
2031 to_read -= num_read;
2032 if (!to_read) {
2033 m_state = state_t::ok;
2034 return length;
2035 }
2036 reinterpret_cast<uint8_t*&>(data) += num_read;
2037 }
2038 }
2039
2040 virtual _Success_(return != 0) size_t write(
2041 _In_reads_bytes_opt_(length) const void* data, _In_ size_t length)
2042 {
2043 // Windows Server 2003 and Windows XP: Pipe write operations across a network are limited in size per write.
2044 // The amount varies per platform. For x86 platforms it's 63.97 MB. For x64 platforms it's 31.97 MB. For Itanium
2045 // it's 63.95 MB. For more information regarding pipes, see the Remarks section.
2046 constexpr size_t
2047#if defined(_WIN64)
2048 block_size = 0x1F80000;
2049#elif defined(_WIN32)
2050 block_size = 0x3f00000;
2051#else
2052 block_size = SSIZE_MAX;
2053#endif
2054 for (size_t to_write = length;;) {
2055#ifdef _WIN32
2056 // ReadFile() might raise an exception. Be cautious with WriteFile() too.
2057 BOOL succeeded;
2058 DWORD num_written;
2059 __try { succeeded = WriteFile(m_h, data, static_cast<DWORD>(std::min<size_t>(to_write, block_size)), &num_written, nullptr); }
2060 __except (EXCEPTION_EXECUTE_HANDLER) { succeeded = FALSE; SetLastError(ERROR_UNHANDLED_EXCEPTION); num_written = 0; }
2061 to_write -= num_written;
2062 if (!to_write) {
2063 m_state = state_t::ok;
2064 return length;
2065 }
2066 reinterpret_cast<const uint8_t*&>(data) += num_written;
2067 if (!succeeded) _Unlikely_ {
2068 m_state = state_t::fail;
2069 return length - to_write;
2070 }
2071#else
2072 ssize_t num_written = write(m_h, data, static_cast<ssize_t>(std::min<size_t>(to_write, block_size)));
2073 if (num_written < 0) _Unlikely_ {
2074 m_state = state_t::fail;
2075 return length - to_write;
2076 }
2077 to_write -= num_written;
2078 if (!to_write) {
2079 m_state = state_t::ok;
2080 return length;
2081 }
2082 reinterpret_cast<const uint8_t*&>(data) += num_written;
2083#endif
2084 }
2085 }
2086
2087 virtual void close()
2088 {
2089 try {
2091 m_state = state_t::ok;
2092 }
2093 catch (std::exception) {
2094 m_state = state_t::fail;
2095 }
2096 }
2097
2098 virtual void flush()
2099 {
2100#ifdef _WIN32
2101 m_state = FlushFileBuffers(m_h) ? state_t::ok : state_t::fail;
2102#else
2103 m_state = fsync(m_h) >= 0 ? state_t::ok : state_t::fail;
2104#endif
2105 }
2106 };
2107
2111 class buffered_sys : public buffer
2112 {
2113 public:
2114 buffered_sys(_In_opt_ sys_handle h = invalid_handle, size_t read_buffer_size = default_buffer_size, size_t write_buffer_size = default_buffer_size) :
2115 buffer(read_buffer_size, write_buffer_size),
2116 m_source(h)
2117 {
2118 init(m_source);
2119 }
2120
2121 protected:
2122 basic_sys m_source;
2123 };
2124
2125#ifdef _WIN32
2129 class ISequentialStream : public basic
2130 {
2131 public:
2132 ISequentialStream(_In_::ISequentialStream* source) : m_source(source)
2133 {
2134 m_source->AddRef();
2135 }
2136
2137 virtual ~ISequentialStream()
2138 {
2139 m_source->Release();
2140 }
2141
2142 virtual _Success_(return != 0 || length == 0) size_t read(
2143 _Out_writes_bytes_to_opt_(length, return) void* data, _In_ size_t length)
2144 {
2145 assert(data || !length);
2146 for (size_t to_read = length;;) {
2147 HRESULT hr;
2148 ULONG num_read = 0;
2149 __try { hr = m_source->Read(data, (ULONG)std::min<size_t>(to_read, ULONG_MAX), &num_read); }
2150 __except (EXCEPTION_EXECUTE_HANDLER) { hr = E_FAIL; }
2151 if (FAILED(hr)) _Unlikely_ {
2152 m_state = to_read < length ? state_t::ok : state_t::fail;
2153 return length - to_read;
2154 }
2155 to_read -= num_read;
2156 if (hr == S_FALSE) _Unlikely_ {
2157 m_state = to_read < length || !length ? state_t::ok : state_t::eof;
2158 return length - to_read;
2159 }
2160 if (!to_read) {
2161 m_state = state_t::ok;
2162 return length;
2163 }
2164 reinterpret_cast<uint8_t*&>(data) += num_read;
2165 }
2166 }
2167
2168 virtual _Success_(return != 0) size_t write(
2169 _In_reads_bytes_opt_(length) const void* data, _In_ size_t length)
2170 {
2171 assert(data || !length);
2172 for (size_t to_write = length;;) {
2173 HRESULT hr;
2174 ULONG num_written = 0;
2175 __try { hr = m_source->Write(data, static_cast<ULONG>(std::min<size_t>(to_write, ULONG_MAX)), &num_written); }
2176 __except (EXCEPTION_EXECUTE_HANDLER) { hr = E_FAIL; }
2177 // In abscence of documentation whether num_written gets set when FAILED(hr) (i.e. partially succesful writes),
2178 // assume write failed completely.
2179 if (FAILED(hr)) _Unlikely_ {
2180 m_state = state_t::fail;
2181 return length - to_write;
2182 }
2183 to_write -= num_written;
2184 if (!to_write) {
2185 m_state = state_t::ok;
2186 return length;
2187 }
2188 reinterpret_cast<const uint8_t*&>(data) += num_written;
2189 }
2190 }
2191
2192 protected:
2193 ::ISequentialStream* m_source;
2194 };
2195
2196#ifndef WIN32_LEAN_AND_MEAN
2200 class asp : public basic
2201 {
2202 public:
2203 asp(_In_opt_ IRequest* request, _In_opt_ IResponse* response) :
2204 m_request(request),
2205 m_response(response)
2206 {
2207 if (m_request)
2208 m_request->AddRef();
2209 if (m_response)
2210 m_response->AddRef();
2211 }
2212
2213 virtual ~asp()
2214 {
2215 if (m_request)
2216 m_request->Release();
2217 if (m_response)
2218 m_response->Release();
2219 }
2220
2221 virtual _Success_(return != 0 || length == 0) size_t read(
2222 _Out_writes_bytes_to_opt_(length, return) void* data, _In_ size_t length)
2223 {
2224 assert(data || !length);
2225 if (!m_request) _Unlikely_ {
2226 m_state = state_t::fail;
2227 return 0;
2228 }
2229 for (size_t to_read = length;;) {
2230 VARIANT var_amount, var_data;
2231 V_VT(&var_amount) = VT_I4;
2232 V_I4(&var_amount) = (LONG)std::min<size_t>(to_read, LONG_MAX);
2233 V_VT(&var_data) = VT_EMPTY;
2234 HRESULT hr = [&]() {
2235 __try { return m_request->BinaryRead(&var_amount, &var_data); }
2236 __except (EXCEPTION_EXECUTE_HANDLER) { return E_FAIL; }
2237 }();
2238 if (FAILED(hr)) _Unlikely_ {
2239 m_state = to_read < length ? state_t::ok : state_t::fail;
2240 return length - to_read;
2241 }
2242 assert(V_VT(&var_amount) == VT_I4);
2243 assert(V_VT(&var_data) == (VT_ARRAY | VT_UI1));
2244 std::unique_ptr<SAFEARRAY, SafeArrayDestroy_delete> sa(V_ARRAY(&var_data));
2245 if (!V_I4(&var_amount)) _Unlikely_ {
2246 m_state = to_read < length || !length ? state_t::ok : state_t::eof;
2247 return length - to_read;
2248 }
2249 safearray_accessor<uint8_t> a(sa.get());
2250 memcpy(data, a.data(), V_I4(&var_amount));
2251 to_read -= V_I4(&var_amount);
2252 if (!to_read) {
2253 m_state = state_t::ok;
2254 return length;
2255 }
2256 reinterpret_cast<uint8_t*&>(data) += V_I4(&var_amount);
2257 }
2258 }
2259
2260 virtual _Success_(return != 0) size_t write(
2261 _In_reads_bytes_opt_(length) const void* data, _In_ size_t length)
2262 {
2263 if (!m_response) {
2264 m_state = state_t::fail;
2265 return 0;
2266 }
2267 for (size_t to_write = length;;) {
2268 UINT num_written = static_cast<UINT>(std::min<size_t>(to_write, UINT_MAX));
2269 std::unique_ptr<OLECHAR, SysFreeString_delete> bstr_data(SysAllocStringByteLen(reinterpret_cast<LPCSTR>(data), num_written));
2270 VARIANT var_data;
2271 V_VT(&var_data) = VT_BSTR;
2272 V_BSTR(&var_data) = bstr_data.get();
2273 HRESULT hr = [&]() {
2274 __try { return m_response->BinaryWrite(var_data); }
2275 __except (EXCEPTION_EXECUTE_HANDLER) { return E_FAIL; }
2276 }();
2277 if (FAILED(hr)) _Unlikely_ {
2278 m_state = state_t::fail;
2279 return length - to_write;
2280 }
2281 to_write -= num_written;
2282 if (!to_write) {
2283 m_state = state_t::ok;
2284 return length;
2285 }
2286 reinterpret_cast<const uint8_t*&>(data) += num_written;
2287 }
2288 }
2289
2290 virtual void close()
2291 {
2292 if (m_response) {
2293 __try { m_response->End(); }
2294 __except (EXCEPTION_EXECUTE_HANDLER) {}
2295 }
2296 m_state = state_t::ok;
2297 }
2298
2299 virtual void flush()
2300 {
2301 if (m_response) {
2302 HRESULT hr;
2303 __try { hr = m_response->Flush(); }
2304 __except (EXCEPTION_EXECUTE_HANDLER) { hr = E_FAIL; }
2305 m_state = SUCCEEDED(hr) ? state_t::ok : state_t::fail;
2306 }
2307 }
2308
2309 protected:
2310 IRequest* m_request;
2311 IResponse* m_response;
2312 };
2313#endif
2314#endif
2315
2319 enum mode_t
2320 {
2321 mode_for_reading = 1 << 0,
2322 mode_for_writing = 1 << 1,
2323 mode_for_chmod = 1 << 2,
2324 mode_create = 1 << 3,
2325 mode_preserve_existing = mode_create | (1 << 4),
2326 mode_append = 1 << 5,
2327 mode_text = 0,
2328 mode_binary = 1 << 6,
2329
2330 share_none = 0,
2331 share_reading = 1 << 7,
2332 share_writing = 1 << 8,
2333 share_deleting = 1 << 9,
2334 share_all = share_reading | share_writing | share_deleting, // Allow others all operations on our file
2335
2336 inherit_handle = 1 << 10,
2337
2338 hint_write_thru = 1 << 11,
2339 hint_no_buffering = 1 << 12,
2340 hint_random_access = 1 << 13,
2341 hint_sequential_access = 1 << 14,
2342 };
2343
2344#pragma warning(push)
2345#pragma warning(disable: 4250)
2349 class file : virtual public basic_file, virtual public basic_sys
2350 {
2351 public:
2352 file(_In_opt_ sys_handle h = invalid_handle, _In_ state_t state = state_t::ok) : basic_sys(h, state) {}
2353
2360 file(_In_z_ const sys_char* filename, _In_ int mode)
2361 {
2362 open(filename, mode);
2363 }
2364
2371 void open(_In_z_ const sys_char* filename, _In_ int mode)
2372 {
2373 if (m_h != invalid_handle)
2374 close();
2375
2376#ifdef _WIN32
2377 DWORD dwDesiredAccess = 0;
2378 if (mode & mode_for_reading) dwDesiredAccess |= GENERIC_READ;
2379 if (mode & mode_for_writing) dwDesiredAccess |= GENERIC_WRITE;
2380 if (mode & mode_for_chmod) dwDesiredAccess |= FILE_WRITE_ATTRIBUTES;
2381
2382 DWORD dwShareMode = 0;
2383 if (mode & share_reading) dwShareMode |= FILE_SHARE_READ;
2384 if (mode & share_writing) dwShareMode |= FILE_SHARE_WRITE;
2385 if (mode & share_deleting) dwShareMode |= FILE_SHARE_DELETE;
2386
2387 SECURITY_ATTRIBUTES sa = { sizeof(SECURITY_ATTRIBUTES) };
2388 sa.bInheritHandle = mode & inherit_handle ? true : false;
2389
2390 DWORD dwCreationDisposition;
2391 switch (mode & mode_preserve_existing) {
2392 case mode_create: dwCreationDisposition = CREATE_ALWAYS; break;
2393 case mode_preserve_existing: dwCreationDisposition = OPEN_ALWAYS; break;
2394 case 0: dwCreationDisposition = OPEN_EXISTING; break;
2395 default: throw std::invalid_argument("invalid mode");
2396 }
2397
2398 DWORD dwFlagsAndAttributes = FILE_ATTRIBUTE_NORMAL;
2399 if (mode & hint_write_thru) dwFlagsAndAttributes |= FILE_FLAG_WRITE_THROUGH;
2400 if (mode & hint_no_buffering) dwFlagsAndAttributes |= FILE_FLAG_NO_BUFFERING;
2401 if (mode & hint_random_access) dwFlagsAndAttributes |= FILE_FLAG_RANDOM_ACCESS;
2402 if (mode & hint_sequential_access) dwFlagsAndAttributes |= FILE_FLAG_SEQUENTIAL_SCAN;
2403
2404 m_h = CreateFile(filename, dwDesiredAccess, dwShareMode, &sa, dwCreationDisposition, dwFlagsAndAttributes, nullptr);
2405#else
2406 int flags = 0;
2407 if (mode & mode_for_reading) flags |= O_RDONLY;
2408 if (mode & mode_for_writing) flags |= O_WRONLY;
2409 if (mode & mode_create) flags |= mode & mode_preserve_existing ? O_CREAT : (O_CREAT | O_EXCL);
2410 if (mode & hint_write_thru) flags |= O_DSYNC;
2411 if (mode & hint_no_buffering) flags |= O_RSYNC;
2412
2413 m_h = open(filename, flags, DEFFILEMODE);
2414#endif
2415 if (m_h != invalid_handle) {
2416 m_state = state_t::ok;
2417 if (mode & mode_append)
2418 seek(0, seek_t::end);
2419 }
2420 else
2421 m_state = state_t::fail;
2422 }
2423
2424 virtual fpos_t seek(_In_ foff_t offset, _In_ seek_t how = seek_t::beg)
2425 {
2426#ifdef _WIN32
2427 LARGE_INTEGER li;
2428 li.QuadPart = offset;
2429 li.LowPart = SetFilePointer(m_h, li.LowPart, &li.HighPart, static_cast<DWORD>(how));
2430 if (li.LowPart != 0xFFFFFFFF || GetLastError() == NO_ERROR) {
2431 m_state = state_t::ok;
2432 return li.QuadPart;
2433 }
2434#else
2435 off64_t result = lseek64(m_h, offset, how);
2436 if (result >= 0) {
2437 m_state = state_t::ok;
2438 return result;
2439 }
2440#endif
2441 m_state = state_t::fail;
2442 return fpos_max;
2443 }
2444
2445 virtual fpos_t tell() const
2446 {
2447 if (m_h != invalid_handle) {
2448#ifdef _WIN32
2449 LARGE_INTEGER li;
2450 li.QuadPart = 0;
2451 li.LowPart = SetFilePointer(m_h, 0, &li.HighPart, FILE_CURRENT);
2452 if (li.LowPart != 0xFFFFFFFF || GetLastError() == NO_ERROR)
2453 return li.QuadPart;
2454#else
2455 off64_t result = lseek64(m_h, 0, SEEK_CUR);
2456 if (result >= 0)
2457 return result;
2458#endif
2459 }
2460 return fpos_max;
2461 }
2462
2463 virtual void lock(_In_ fpos_t offset, _In_ fsize_t length)
2464 {
2465#ifdef _WIN32
2466 LARGE_INTEGER liOffset;
2467 LARGE_INTEGER liSize;
2468 liOffset.QuadPart = offset;
2469 liSize.QuadPart = length;
2470 if (LockFile(m_h, liOffset.LowPart, liOffset.HighPart, liSize.LowPart, liSize.HighPart)) {
2471 m_state = state_t::ok;
2472 return;
2473 }
2474#else
2475 off64_t orig = lseek64(m_h, 0, SEEK_CUR);
2476 if (orig >= 0) {
2477 m_state = lseek64(m_h, offset, SEEK_SET) >= 0 && lockf64(m_h, F_LOCK, length) >= 0 ? state_t::ok : state_t::fail;
2478 lseek64(m_h, orig, SEEK_SET);
2479 m_state = state_t::ok;
2480 return;
2481 }
2482#endif
2483 m_state = state_t::fail;
2484 }
2485
2486 virtual void unlock(_In_ fpos_t offset, _In_ fsize_t length)
2487 {
2488#ifdef _WIN32
2489 LARGE_INTEGER liOffset;
2490 LARGE_INTEGER liSize;
2491 liOffset.QuadPart = offset;
2492 liSize.QuadPart = length;
2493 if (UnlockFile(m_h, liOffset.LowPart, liOffset.HighPart, liSize.LowPart, liSize.HighPart)) {
2494 m_state = state_t::ok;
2495 return;
2496 }
2497#else
2498 off64_t orig = lseek64(m_h, 0, SEEK_CUR);
2499 if (orig >= 0) {
2500 if (lseek64(m_h, offset, SEEK_SET) >= 0 && lockf64(m_h, F_ULOCK, length) >= 0) {
2501 lseek64(m_h, orig, SEEK_SET);
2502 m_state = state_t::ok;
2503 return;
2504 }
2505 lseek64(m_h, orig, SEEK_SET);
2506 }
2507#endif
2508 m_state = state_t::fail;
2509 }
2510
2511 virtual fsize_t size()
2512 {
2513#ifdef _WIN32
2514 LARGE_INTEGER li;
2515 li.LowPart = GetFileSize(m_h, (LPDWORD)&li.HighPart);
2516 if (li.LowPart == 0xFFFFFFFF && GetLastError() != NO_ERROR)
2517 li.QuadPart = -1;
2518 return li.QuadPart;
2519#else
2520 off64_t length = -1, orig = lseek64(m_h, 0, SEEK_CUR);
2521 if (orig >= 0) {
2522 length = lseek64(m_h, 0, SEEK_END);
2523 lseek64(m_h, orig, SEEK_SET);
2524 }
2525 return length;
2526#endif
2527 }
2528
2529 virtual void truncate()
2530 {
2531#ifdef _WIN32
2532 if (SetEndOfFile(m_h)) {
2533 m_state = state_t::ok;
2534 return;
2535 }
2536#else
2537 off64_t length = lseek64(m_h, 0, SEEK_CUR);
2538 if (length >= 0 && ftruncate64(m_h, length) >= 0) {
2539 m_state = state_t::ok;
2540 return;
2541 }
2542#endif
2543 m_state = state_t::fail;
2544 }
2545
2546#ifdef _WIN32
2547 static inline time_point ft2tp(_In_ const FILETIME& ft)
2548 {
2549#if _HAS_CXX20
2550 uint64_t t = (static_cast<int64_t>(ft.dwHighDateTime) << 32) | ft.dwLowDateTime;
2551#else
2552 uint64_t t = ((static_cast<int64_t>(ft.dwHighDateTime) << 32) | ft.dwLowDateTime) - 116444736000000000ll;
2553#endif
2554 return time_point(time_point::duration(t));
2555 }
2556
2557 static inline void tp2ft(_In_ time_point tp, _Out_ FILETIME& ft)
2558 {
2559#if _HAS_CXX20
2560 uint64_t t = tp.time_since_epoch().count();
2561#else
2562 uint64_t t = tp.time_since_epoch().count() + 116444736000000000ll;
2563#endif
2564 ft.dwHighDateTime = static_cast<DWORD>((t >> 32) & 0xffffffff);
2565 ft.dwLowDateTime = static_cast<DWORD>(t & 0xffffffff);
2566 }
2567#endif
2568
2569 virtual time_point ctime() const
2570 {
2571#ifdef _WIN32
2572 FILETIME ft;
2573 if (GetFileTime(m_h, &ft, nullptr, nullptr))
2574 return ft2tp(ft);
2575#endif
2576 return time_point::min();
2577 }
2578
2579 virtual time_point atime() const
2580 {
2581#ifdef _WIN32
2582 FILETIME ft;
2583 if (GetFileTime(m_h, nullptr, &ft, nullptr))
2584 return ft2tp(ft);
2585#else
2586 struct stat buf;
2587 if (fstat(m_h, &buf) >= 0);
2588 return time_point::from_time_t(buf.st_atim);
2589#endif
2590 return time_point::min();
2591 }
2592
2593 virtual time_point mtime() const
2594 {
2595#ifdef _WIN32
2596 FILETIME ft;
2597 if (GetFileTime(m_h, nullptr, nullptr, &ft))
2598 return ft2tp(ft);
2599#else
2600 struct stat buf;
2601 if (fstat(m_h, &buf) >= 0)
2602 return time_point::from_time_t(buf.st_mtim);
2603#endif
2604 return time_point::min();
2605 }
2606
2607 virtual void set_ctime(time_point date)
2608 {
2609 assert(m_h != invalid_handle);
2610#ifdef _WIN32
2611 FILETIME ft;
2612 tp2ft(date, ft);
2613 if (SetFileTime(m_h, &ft, nullptr, nullptr))
2614 return;
2615#endif
2616 throw std::runtime_error("failed to set file ctime");
2617 }
2618
2619 virtual void set_atime(time_point date)
2620 {
2621 assert(m_h != invalid_handle);
2622#ifdef _WIN32
2623 FILETIME ft;
2624 tp2ft(date, ft);
2625 if (SetFileTime(m_h, nullptr, &ft, nullptr))
2626 return;
2627#else
2628 struct timespec ts[2];
2629 ts[0].tv_sec = date;
2630 ts[1].tv_nsec = UTIME_OMIT;
2631 if (futimens(m_h, ts) >= 0)
2632 return;
2633#endif
2634 throw std::runtime_error("failed to set file atime");
2635 }
2636
2637 virtual void set_mtime(time_point date)
2638 {
2639#ifdef _WIN32
2640 FILETIME ft;
2641 tp2ft(date, ft);
2642 if (SetFileTime(m_h, nullptr, nullptr, &ft))
2643 return;
2644#else
2645 struct timespec ts[2];
2646 ts[0].tv_nsec = UTIME_OMIT;
2647 ts[1].tv_sec = date;
2648 if (futimens(m_h, ts) >= 0)
2649 return;
2650#endif
2651 throw std::runtime_error("failed to set file mtime");
2652 }
2653 };
2654#pragma warning(pop)
2655
2659 class cached_file : public cache
2660 {
2661 public:
2662 cached_file(_In_opt_ sys_handle h = invalid_handle, _In_ state_t state = state_t::ok, _In_ size_t cache_size = default_cache_size) :
2663 cache(cache_size),
2664 m_source(h, state)
2665 {
2666 init(m_source);
2667 }
2668
2676 cached_file(_In_z_ const sys_char* filename, _In_ int mode, _In_ size_t cache_size = default_cache_size) :
2677 cache(cache_size),
2678 m_source(filename, mode& mode_for_writing ? mode | mode_for_reading : mode)
2679 {
2680 init(m_source);
2681 }
2682
2690 void open(_In_z_ const sys_char* filename, _In_ int mode)
2691 {
2692 if (mode & mode_for_writing) mode |= mode_for_reading;
2693 m_source.open(filename, mode);
2694 if (m_source.ok()) {
2695#if SET_FILE_OP_TIMES
2696 m_atime = m_mtime = time_point::min();
2697#endif
2698 m_offset = m_source.tell();
2699 m_state = state_t::ok;
2700 return;
2701 }
2702 m_state = state_t::fail;
2703 }
2704
2705 protected:
2706 file m_source;
2707 };
2708
2713 {
2714 public:
2715 memory_file(_In_ state_t state = state_t::ok) :
2716 basic(state),
2717 m_data(nullptr),
2718 m_offset(0),
2719 m_size(0),
2720 m_reserved(0),
2721 m_manage(true)
2722 {
2723#if SET_FILE_OP_TIMES
2724 m_ctime = m_atime = m_mtime = time_point::now();
2725#endif
2726 }
2727
2734 memory_file(_In_ size_t size, _In_ state_t state = state_t::ok) :
2735 basic(state),
2736 m_data(reinterpret_cast<uint8_t*>(malloc(size))),
2737 m_offset(0),
2738 m_size(0),
2740 m_manage(true)
2741 {
2742 if (!m_data)
2743 throw std::bad_alloc();
2744#if SET_FILE_OP_TIMES
2745 m_ctime = m_atime = m_mtime = time_point::now();
2746#endif
2747 }
2748
2758 memory_file(_Inout_ void* data, _In_ size_t size, _In_ size_t reserved, _In_ bool manage = false, _In_ state_t state = state_t::ok) :
2759 basic(state),
2760 m_data(reinterpret_cast<uint8_t*>(data)),
2761 m_offset(0),
2762 m_size(size),
2763 m_reserved(reserved),
2764 m_manage(manage)
2765 {
2766 assert(data || !size);
2767 assert(reserved >= size);
2768#if SET_FILE_OP_TIMES
2769 m_ctime = m_atime = m_mtime = time_point::now();
2770#endif
2771 }
2772
2781 memory_file(_Inout_ void* data, _In_ size_t size, _In_ bool manage = false, _In_ state_t state = state_t::ok) :
2782 memory_file(data, size, size, manage, state)
2783 {}
2784
2791 memory_file(_In_z_ const sys_char* filename, _In_ int mode) : memory_file()
2792 {
2793 load(filename, mode);
2794 }
2795
2796 virtual ~memory_file()
2797 {
2798 if (m_manage && m_data)
2799 free(m_data);
2800 }
2801
2808 void reserve(_In_ size_t required, _In_ bool tight = false) noexcept
2809 {
2810 if (required <= m_reserved && (!tight || required >= m_reserved)) {
2811 m_state = state_t::ok;
2812 return;
2813 }
2814 if (!m_manage) {
2815 m_state = state_t::fail;
2816 return;
2817 }
2818 size_t reserved = tight ? required : ((required + required / 4 + (default_block_size - 1)) / default_block_size) * default_block_size;
2819 auto data = reinterpret_cast<uint8_t*>(realloc(m_data, reserved));
2820 if (!data && reserved) _Unlikely_ {
2821 m_state = state_t::fail;
2822 return;
2823 }
2824 m_data = data;
2825 if (reserved < m_size)
2826 m_size = reserved;
2827 m_reserved = reserved;
2828 m_state = state_t::ok;
2829 }
2830
2837 void load(_In_z_ const sys_char* filename, _In_ int mode)
2838 {
2839 file f(filename, (mode & ~hint_random_access) | mode_for_reading | hint_sequential_access);
2840 if (!f.ok()) {
2841 m_state = state_t::fail;
2842 return;
2843 }
2844 fsize_t size = f.size();
2845 if (size > SIZE_MAX) {
2846 m_state = state_t::fail;
2847 return;
2848 }
2849 reserve(static_cast<size_t>(size), true);
2850 if (!ok()) _Unlikely_ {
2851 return;
2852 }
2853 m_offset = m_size = 0;
2854 write_stream(f);
2855 if (ok())
2856 m_offset = 0;
2857#if SET_FILE_OP_TIMES
2858 m_ctime = f.ctime();
2859 m_atime = f.atime();
2860 m_mtime = f.mtime();
2861#endif
2862 }
2863
2870 void save(_In_z_ const sys_char* filename, _In_ int mode)
2871 {
2872 file f(filename, (mode & ~hint_random_access) | mode_for_writing | hint_sequential_access);
2873 if (!f.ok()) {
2874 m_state = state_t::fail;
2875 return;
2876 }
2877 f.write(m_data, m_size);
2878 if (!f.ok()) {
2879 m_state = state_t::fail;
2880 return;
2881 }
2882 f.truncate();
2883#if SET_FILE_OP_TIMES
2884 f.set_ctime(m_ctime);
2885 f.set_atime(m_atime);
2886 f.set_mtime(m_mtime);
2887#endif
2888 }
2889
2893 inline const void* data() const { return m_data; }
2894
2895 virtual _Success_(return != 0 || length == 0) size_t read(
2896 _Out_writes_bytes_to_opt_(length, return) void* data, _In_ size_t length)
2897 {
2898 assert(data || !length);
2899#if SET_FILE_OP_TIMES
2900 m_atime = time_point::now();
2901#endif
2902 size_t available = m_size - m_offset;
2903 if (length <= available) {
2904 memcpy(data, m_data + m_offset, length);
2905 m_offset += length;
2906 m_state = state_t::ok;
2907 return length;
2908 }
2909 if (length && !available) {
2910 m_state = state_t::eof;
2911 return 0;
2912 }
2913 memcpy(data, m_data + m_offset, available);
2914 m_offset += available;
2915 m_state = state_t::ok;
2916 return available;
2917 }
2918
2933 template <class T>
2935 {
2936#if SET_FILE_OP_TIMES
2937 m_atime = time_point::now();
2938#endif
2939 if (CHECK_STREAM_STATE && !ok()) _Unlikely_ {
2940 data = 0;
2941 return *this;
2942 }
2943 size_t end_offset = m_offset + sizeof(T);
2944 if (end_offset <= m_size) {
2945 data = LE2HE(*reinterpret_cast<T*>(m_data + m_offset));
2946 m_offset = end_offset;
2947#if !CHECK_STREAM_STATE
2948 m_state = state_t::ok;
2949#endif
2950 }
2951 else {
2952 data = 0;
2953 m_offset = m_size;
2954 m_state = state_t::eof;
2955 }
2956 return *this;
2957 }
2958
2973 template<class _Elem, class _Traits = std::char_traits<_Elem>, class _Ax = std::allocator<_Elem>>
2974 memory_file& read_str(_Inout_ std::basic_string<_Elem, _Traits, _Ax>&data)
2975 {
2976#if SET_FILE_OP_TIMES
2977 m_atime = time_point::now();
2978#endif
2979 if (CHECK_STREAM_STATE && !ok()) _Unlikely_ {
2980 data.clear();
2981 return *this;
2982 }
2983 size_t end_offset = m_offset + sizeof(uint32_t);
2984 if (end_offset <= m_size) {
2985 uint32_t num_chars = LE2HE(*reinterpret_cast<uint32_t*>(m_data + m_offset));
2986 m_offset = end_offset;
2987 end_offset = stdex::add(m_offset + stdex::mul(num_chars, sizeof(_Elem)));
2988 _Elem* start = reinterpret_cast<_Elem*>(m_data + m_offset);
2989 if (end_offset <= m_size) {
2990 data.assign(start, start + num_chars);
2991 m_offset = end_offset;
2992#if !CHECK_STREAM_STATE
2993 m_state = state_t::ok;
2994#endif
2995 return *this;
2996 }
2997 if (end_offset <= m_size)
2998 data.assign(start, reinterpret_cast<_Elem*>(m_data + m_size));
2999 }
3000 m_offset = m_size;
3001 m_state = state_t::eof;
3002 return *this;
3003 }
3004
3005 virtual _Success_(return != 0) size_t write(
3006 _In_reads_bytes_opt_(length) const void* data, _In_ size_t length)
3007 {
3008 assert(data || !length);
3009#if SET_FILE_OP_TIMES
3010 m_atime = m_mtime = time_point::now();
3011#endif
3012 size_t end_offset = m_offset + length;
3013 if (end_offset > m_reserved) {
3014 reserve(end_offset);
3015 if (!ok()) _Unlikely_
3016 return 0;
3017 }
3018 memcpy(m_data + m_offset, data, length);
3019 m_offset = end_offset;
3020 if (m_offset > m_size)
3021 m_size = m_offset;
3022 m_state = state_t::ok;
3023 return length;
3024 }
3025
3029 void write_byte(_In_ uint8_t byte, _In_ size_t amount = 1)
3030 {
3031#if SET_FILE_OP_TIMES
3032 m_atime = m_mtime = time_point::now();
3033#endif
3034 size_t end_offset = m_offset + amount;
3035 if (end_offset > m_reserved) {
3036 reserve(end_offset);
3037 if (!ok()) _Unlikely_
3038 return;
3039 }
3040 memset(m_data + m_offset, byte, amount);
3041 m_offset = end_offset;
3042 if (m_offset > m_size)
3043 m_size = m_offset;
3044 m_state = state_t::ok;
3045 }
3046
3061 template <class T>
3062 inline memory_file& write_data(const T data)
3063 {
3064#if SET_FILE_OP_TIMES
3065 m_atime = m_mtime = time_point::now();
3066#endif
3067 if (CHECK_STREAM_STATE && !ok()) _Unlikely_
3068 return *this;
3069 size_t end_offset = m_offset + sizeof(T);
3070 if (end_offset > m_reserved) {
3071 reserve(end_offset);
3072 if (!ok()) _Unlikely_
3073 return *this;
3074 }
3075 (*reinterpret_cast<T*>(m_data + m_offset)) = HE2LE(data);
3076 m_offset = end_offset;
3077 if (m_offset > m_size)
3078 m_size = m_offset;
3079#if !CHECK_STREAM_STATE
3080 m_state = state_t::ok;
3081#endif
3082 return *this;
3083 }
3084
3099 template <class T>
3100 inline memory_file& write_str(_In_z_ const T * data)
3101 {
3102#if SET_FILE_OP_TIMES
3103 m_atime = m_mtime = time_point::now();
3104#endif
3105 if (CHECK_STREAM_STATE && !ok()) _Unlikely_
3106 return *this;
3107 size_t num_chars = stdex::strlen(data);
3108 if (num_chars > UINT32_MAX)
3109 throw std::invalid_argument("string too long");
3110 size_t size_chars = num_chars * sizeof(T);
3111 size_t size = sizeof(uint32_t) + size_chars;
3112 size_t end_offset = m_offset + size;
3113 if (end_offset > m_reserved) {
3114 reserve(end_offset);
3115 if (!ok()) _Unlikely_
3116 return *this;
3117 }
3118 auto p = tok.m_podatki + m_offset;
3119 *reinterpret_cast<uint32_t*>(p) = HE2LE((uint32_t)num_chars);
3120 memcpy(p + sizeof(uint32_t), data, size_chars);
3121 m_offset = end_offset;
3122 if (m_offset > m_size)
3123 m_size = m_offset;
3124#if !CHECK_STREAM_STATE
3125 m_state = state_t::ok;
3126#endif
3127 return *this;
3128 }
3129
3135 size_t write_stream(_Inout_ basic & stream, _In_ size_t amount = SIZE_MAX)
3136 {
3137#if SET_FILE_OP_TIMES
3138 m_atime = m_mtime = time_point::now();
3139#endif
3140 size_t num_read, dst_offset = m_offset, dst_size = m_offset;
3141 size_t num_copied = 0, to_write = amount;
3142 m_state = state_t::ok;
3143 if (amount != SIZE_MAX) {
3144 dst_size = stdex::add(dst_size, amount);
3145 reserve(dst_size);
3146 if (!ok()) _Unlikely_
3147 return 0;
3148 while (to_write) {
3149 num_read = stream.read(m_data + dst_offset, to_write);
3150 dst_size = dst_offset += num_read;
3151 num_copied += num_read;
3152 to_write -= num_read;
3153 if (!stream.ok()) {
3154 if (stream.state() != state_t::eof)
3155 m_state = state_t::fail;
3156 break;
3157 }
3158 };
3159 }
3160 else {
3161 size_t block_size;
3162 while (to_write) {
3163 block_size = std::min(to_write, default_block_size);
3164 dst_size = stdex::add(dst_size, block_size);
3165 reserve(dst_size);
3166 if (!ok()) _Unlikely_
3167 break;
3168 num_read = stream.read(m_data + dst_offset, block_size);
3169 dst_size = dst_offset += num_read;
3170 num_copied += num_read;
3171 to_write -= num_read;
3172 if (!stream.ok()) {
3173 if (stream.state() != state_t::eof)
3174 m_state = state_t::fail;
3175 break;
3176 }
3177 };
3178 }
3179 m_offset = dst_offset;
3180 if (m_offset > m_size)
3181 m_size = m_offset;
3182 return num_copied;
3183 }
3184
3185 virtual void close()
3186 {
3187 if (m_manage && m_data)
3188 free(m_data);
3189 m_data = nullptr;
3190 m_manage = true;
3191 m_offset = 0;
3192 m_size = m_reserved = 0;
3193#if SET_FILE_OP_TIMES
3194 m_ctime = m_atime = m_mtime = time_point::min();
3195#endif
3196 m_state = state_t::ok;
3197 }
3198
3199 virtual fpos_t seek(_In_ foff_t offset, _In_ seek_t how = seek_t::beg)
3200 {
3201 fpos_t target;
3202 switch (how) {
3203 case seek_t::beg: target = offset; break;
3204 case seek_t::cur: target = static_cast<fpos_t>(m_offset) + offset; break;
3205 case seek_t::end: target = static_cast<fpos_t>(m_size) + offset; break;
3206 default: throw std::invalid_argument("unknown seek origin");
3207 }
3208 if (target <= SIZE_MAX) {
3209 m_state = state_t::ok;
3210 return m_offset = static_cast<size_t>(target);
3211 }
3212 m_state = state_t::fail;
3213 return fpos_max;
3214 }
3215
3216 virtual fpos_t tell() const
3217 {
3218 return m_offset;
3219 }
3220
3221 virtual fsize_t size()
3222 {
3223 return m_size;
3224 }
3225
3226 virtual void truncate()
3227 {
3228#if SET_FILE_OP_TIMES
3229 m_atime = m_mtime = time_point::now();
3230#endif
3231 m_size = m_offset;
3233 }
3234
3235#if SET_FILE_OP_TIMES
3236 virtual time_point ctime() const
3237 {
3238 return m_ctime;
3239 }
3240
3241 virtual time_point atime() const
3242 {
3243 return m_atime;
3244 }
3245
3246 virtual time_point mtime() const
3247 {
3248 return m_mtime;
3249 }
3250
3251 virtual void set_ctime(time_point date)
3252 {
3253 m_ctime = date;
3254 }
3255
3256 virtual void set_atime(time_point date)
3257 {
3258 m_atime = date;
3259 }
3260
3261 virtual void set_mtime(time_point date)
3262 {
3263 m_mtime = date;
3264 }
3265#endif
3266
3267 protected:
3275 template <class T>
3276 inline void set(_In_ fpos_t offset, _In_ const T data)
3277 {
3278#if SET_FILE_OP_TIMES
3279 m_atime = m_mtime = time_point::now();
3280#endif
3281 assert(offset + sizeof(T) < m_size);
3282 (*reinterpret_cast<T*>(m_data + offset)) = HE2LE(data);
3283 }
3284
3285 public:
3286 inline void set(_In_ fpos_t offset, _In_ const int8_t data) { set<int8_t>(offset, data); }
3287 inline void set(_In_ fpos_t offset, _In_ const int16_t data) { set<int16_t>(offset, data); }
3288 inline void set(_In_ fpos_t offset, _In_ const int32_t data) { set<int32_t>(offset, data); }
3289 inline void set(_In_ fpos_t offset, _In_ const int64_t data) { set<int64_t>(offset, data); }
3290 inline void set(_In_ fpos_t offset, _In_ const uint8_t data) { set<uint8_t>(offset, data); }
3291 inline void set(_In_ fpos_t offset, _In_ const uint16_t data) { set<uint16_t>(offset, data); }
3292 inline void set(_In_ fpos_t offset, _In_ const uint32_t data) { set<uint32_t>(offset, data); }
3293 inline void set(_In_ fpos_t offset, _In_ const uint64_t data) { set<uint64_t>(offset, data); }
3294#if defined(_WIN64) && defined(_NATIVE_SIZE_T_DEFINED)
3295 inline void set(_In_ fpos_t offset, _In_ const size_t data) { set<size_t>(offset, data); }
3296#endif
3297 inline void set(_In_ fpos_t offset, _In_ const float data) { set<float>(offset, data); }
3298 inline void set(_In_ fpos_t offset, _In_ const double data) { set<double>(offset, data); }
3299 inline void set(_In_ fpos_t offset, _In_ const char data) { set<char>(offset, data); }
3300#ifdef _NATIVE_WCHAR_T_DEFINED
3301 inline void set(_In_ fpos_t offset, _In_ const wchar_t data) { set<wchar_t>(offset, data); }
3302#endif
3303
3311 protected:
3312 template <class T>
3313 inline void get(_In_ fpos_t offset, _Out_ T & data)
3314 {
3315 assert(offset + sizeof(T) < m_size);
3316 data = LE2HE(*(T*)(m_data + offset));
3317#if SET_FILE_OP_TIMES
3318 m_atime = time_point::now();
3319#endif
3320 }
3321
3322 public:
3323 inline void get(_In_ fpos_t offset, _Out_ int8_t & data) { get<int8_t>(offset, data); }
3324 inline void get(_In_ fpos_t offset, _Out_ int16_t & data) { get<int16_t>(offset, data); }
3325 inline void get(_In_ fpos_t offset, _Out_ int32_t & data) { get<int32_t>(offset, data); }
3326 inline void get(_In_ fpos_t offset, _Out_ int64_t & data) { get<int64_t>(offset, data); }
3327 inline void get(_In_ fpos_t offset, _Out_ uint8_t & data) { get<uint8_t>(offset, data); }
3328 inline void get(_In_ fpos_t offset, _Out_ uint16_t & data) { get<uint16_t>(offset, data); }
3329 inline void get(_In_ fpos_t offset, _Out_ uint32_t & data) { get<uint32_t>(offset, data); }
3330 inline void get(_In_ fpos_t offset, _Out_ uint64_t & data) { get<uint64_t>(offset, data); }
3331#if defined(_WIN64) && defined(_NATIVE_SIZE_T_DEFINED)
3332 inline void get(_In_ fpos_t offset, _Out_ size_t & data) { get<size_t>(offset, data); }
3333#endif
3334 inline void get(_In_ fpos_t offset, _Out_ float& data) { get<float>(offset, data); }
3335 inline void get(_In_ fpos_t offset, _Out_ double& data) { get<double>(offset, data); }
3336 inline void get(_In_ fpos_t offset, _Out_ char& data) { get<char>(offset, data); }
3337#ifdef _NATIVE_WCHAR_T_DEFINED
3338 inline void get(_In_ fpos_t offset, _Out_ wchar_t& data) { get<wchar_t>(offset, data); }
3339#endif
3340
3341 inline memory_file& operator <<(_In_ const int8_t data) { return write_data(data); }
3342 inline memory_file& operator >>(_Out_ int8_t & data) { return read_data(data); }
3343 inline memory_file& operator <<(_In_ const int16_t data) { return write_data(data); }
3344 inline memory_file& operator >>(_Out_ int16_t & data) { return read_data(data); }
3345 inline memory_file& operator <<(_In_ const int32_t data) { return write_data(data); }
3346 inline memory_file& operator >>(_Out_ int32_t & data) { return read_data(data); }
3347 inline memory_file& operator <<(_In_ const int64_t data) { return write_data(data); }
3348 inline memory_file& operator >>(_Out_ int64_t & data) { return read_data(data); }
3349 inline memory_file& operator <<(_In_ const uint8_t data) { return write_data(data); }
3350 inline memory_file& operator >>(_Out_ uint8_t & data) { return read_data(data); }
3351 inline memory_file& operator <<(_In_ const uint16_t data) { return write_data(data); }
3352 inline memory_file& operator >>(_Out_ uint16_t & data) { return read_data(data); }
3353 inline memory_file& operator <<(_In_ const uint32_t data) { return write_data(data); }
3354 inline memory_file& operator >>(_Out_ uint32_t & data) { return read_data(data); }
3355 inline memory_file& operator <<(_In_ const uint64_t data) { return write_data(data); }
3356 inline memory_file& operator >>(_Out_ uint64_t & data) { return read_data(data); }
3357#if defined(_WIN64) && defined(_NATIVE_SIZE_T_DEFINED)
3358 inline memory_file& operator <<(_In_ const size_t data) { return write_data(data); }
3359 inline memory_file& operator >>(_Out_ size_t & data) { return read_data(data); }
3360#endif
3361 inline memory_file& operator <<(_In_ const float data) { return write_data(data); }
3362 inline memory_file& operator >>(_Out_ float& data) { return read_data(data); }
3363 inline memory_file& operator <<(_In_ const double data) { return write_data(data); }
3364 inline memory_file& operator >>(_Out_ double& data) { return read_data(data); }
3365 inline memory_file& operator <<(_In_ const char data) { return write_data(data); }
3366 inline memory_file& operator >>(_Out_ char& data) { return read_data(data); }
3367#ifdef _NATIVE_WCHAR_T_DEFINED
3368 inline memory_file& operator <<(_In_ const wchar_t data) { return write_data(data); }
3369 inline memory_file& operator >>(_Out_ wchar_t& data) { return read_data(data); }
3370#endif
3371 template <class T>
3372 inline memory_file& operator <<(_In_ const T * data) { return write_str(data); }
3373 template<class _Elem, class _Traits = std::char_traits<_Elem>, class _Ax = std::allocator<_Elem>>
3374 inline memory_file& operator >>(_Inout_ std::basic_string<_Elem, _Traits, _Ax>&data) { return read_str(data); }
3375
3376 protected:
3377 uint8_t* m_data;
3379 size_t m_offset;
3380 size_t m_size;
3381 size_t m_reserved;
3382#if SET_FILE_OP_TIMES
3383 time_point
3384 m_ctime,
3385 m_atime,
3386 m_mtime;
3387#endif
3388 };
3389
3393 class fifo : public basic {
3394 public:
3395 fifo() :
3396 m_offset(0),
3397 m_size(0),
3398 m_head(nullptr),
3399 m_tail(nullptr)
3400 {}
3401
3402 virtual ~fifo()
3403 {
3404 while (m_head) {
3405 auto p = m_head;
3406 m_head = p->next;
3407 delete p;
3408 }
3409 }
3410
3411#pragma warning(suppress: 6101) // See [2] below
3412 virtual _Success_(return != 0 || length == 0) size_t read(
3413 _Out_writes_bytes_to_opt_(length, return) void* data, _In_ size_t length)
3414 {
3415 assert(data || !length);
3416 for (size_t to_read = length;;) {
3417 if (!m_head) _Unlikely_ {
3418 // [1] Code analysis misses length - to_read bytes were written to data in previous loop iterations.
3419 m_state = to_read < length || !length ? state_t::ok : state_t::eof;
3420 return length - to_read;
3421 }
3422 size_t remaining = m_head->size - m_offset;
3423 if (remaining > to_read) {
3424 memcpy(data, m_head->data + m_offset, to_read);
3425 m_offset += to_read;
3426 m_size -= to_read;
3427 m_state = state_t::ok;
3428 return length;
3429 }
3430 memcpy(data, m_head->data + m_offset, remaining);
3431 m_offset = 0;
3432 m_size -= remaining;
3433 reinterpret_cast<uint8_t*&>(data) += remaining;
3434 to_read -= remaining;
3435 auto p = m_head;
3436 m_head = p->next;
3437 delete p;
3438 }
3439 }
3440
3441 virtual _Success_(return != 0) size_t write(
3442 _In_reads_bytes_opt_(length) const void* data, _In_ size_t length)
3443 {
3444 assert(data || !length);
3445 try {
3446 std::unique_ptr<node_t> n(reinterpret_cast<node_t*>(new uint8_t[sizeof(node_t) + length]));
3447 n->next = nullptr;
3448 n->size = length;
3449 memcpy(n->data, data, length);
3450 m_size += length;
3451 if (m_head)
3452 m_tail = m_tail->next = n.release();
3453 else
3454 m_head = m_tail = n.release();
3455 m_state = state_t::ok;
3456 return length;
3457 }
3458 catch (std::bad_alloc) {
3459 m_state = state_t::fail;
3460 return 0;
3461 }
3462 }
3463
3464 virtual void close()
3465 {
3466 m_size = m_offset = 0;
3467 while (m_head) {
3468 auto p = m_head;
3469 m_head = p->next;
3470 delete p;
3471 }
3472 m_state = state_t::ok;
3473 }
3474
3478 inline size_t size() const { return m_size; };
3479
3480 protected:
3481 size_t m_offset, m_size;
3482 struct node_t {
3483 node_t* next;
3484 size_t size;
3485#pragma warning(suppress:4200)
3486 uint8_t data[0];
3487 } *m_head, * m_tail;
3488 };
3489
3493 class diag_file : public basic_file {
3494 public:
3495 diag_file(_In_count_(num_files) basic_file* const* files, _In_ size_t num_files) :
3496 basic(num_files ? files[0]->state() : state_t::fail),
3497 m_files(files, files + num_files)
3498 {
3499 }
3500
3501 virtual _Success_(return != 0 || length == 0) size_t read(
3502 _Out_writes_bytes_to_opt_(length, return) void* data, _In_ size_t length)
3503 {
3504 assert(data || !length);
3505 if (m_files.empty()) {
3506 m_state = state_t::fail;
3507 return 0;
3508 }
3509 size_t result = m_files[0]->read(data, length);
3510 _Analysis_assume_(result <= length);
3511 m_state = m_files[0]->state();
3512 if (length > m_tmp.size())
3513 m_tmp.resize(length);
3514 for (size_t i = 1, n = m_files.size(); i < n; ++i) {
3515 if (m_files[i]->read(m_tmp.data(), length) != result ||
3516 memcmp(m_tmp.data(), data, result))
3517 throw std::runtime_error("read mismatch");
3518 if (m_files[i]->state() != m_state)
3519 throw std::runtime_error("state mismatch");
3520 }
3521 return result;
3522 }
3523
3524 virtual _Success_(return != 0) size_t write(
3525 _In_reads_bytes_opt_(length) const void* data, _In_ size_t length)
3526 {
3527 if (m_files.empty()) {
3528 m_state = state_t::fail;
3529 return 0;
3530 }
3531 size_t result = m_files[0]->write(data, length);
3532 m_state = m_files[0]->state();
3533 for (size_t i = 1, n = m_files.size(); i < n; ++i) {
3534 if (m_files[i]->write(data, length) != result)
3535 throw std::runtime_error("write mismatch");
3536 if (m_files[i]->state() != m_state)
3537 throw std::runtime_error("state mismatch");
3538 }
3539 return result;
3540 }
3541
3542 virtual void flush()
3543 {
3544 if (m_files.empty()) {
3545 m_state = state_t::ok;
3546 return;
3547 }
3548 m_files[0]->flush();
3549 m_state = m_files[0]->state();
3550 for (size_t i = 1, n = m_files.size(); i < n; ++i) {
3551 m_files[i]->flush();
3552 if (m_files[i]->state() != m_state)
3553 throw std::runtime_error("state mismatch");
3554 }
3555 }
3556
3557 virtual void close()
3558 {
3559 if (m_files.empty()) {
3560 m_state = state_t::ok;
3561 return;
3562 }
3563 m_files[0]->close();
3564 m_state = m_files[0]->state();
3565 for (size_t i = 1, n = m_files.size(); i < n; ++i) {
3566 m_files[i]->close();
3567 if (m_files[i]->state() != m_state)
3568 throw std::runtime_error("state mismatch");
3569 }
3570 m_tmp.clear();
3571 m_tmp.shrink_to_fit();
3572 }
3573
3574 virtual fpos_t seek(_In_ foff_t offset, _In_ seek_t how = seek_t::beg)
3575 {
3576 if (m_files.empty()) {
3577 m_state = state_t::fail;
3578 return fpos_max;
3579 }
3580 fpos_t result = m_files[0]->seek(offset, how);
3581 m_state = m_files[0]->state();
3582 for (size_t i = 1, n = m_files.size(); i < n; ++i) {
3583 if (m_files[i]->seek(offset, how) != result)
3584 throw std::runtime_error("seek mismatch");
3585 if (m_files[i]->state() != m_state)
3586 throw std::runtime_error("state mismatch");
3587 }
3588 return result;
3589 }
3590
3591 virtual fpos_t tell() const
3592 {
3593 if (m_files.empty())
3594 return fpos_max;
3595 fpos_t result = m_files[0]->tell();
3596 for (size_t i = 1, n = m_files.size(); i < n; ++i) {
3597 if (m_files[i]->tell() != result)
3598 throw std::runtime_error("tell mismatch");
3599 }
3600 return result;
3601 }
3602
3603 virtual void lock(_In_ fpos_t offset, _In_ fsize_t length)
3604 {
3605 if (m_files.empty())
3606 m_state = state_t::fail;
3607 m_files[0]->lock(offset, length);
3608 m_state = m_files[0]->state();
3609 for (size_t i = 1, n = m_files.size(); i < n; ++i) {
3610 m_files[i]->lock(offset, length);
3611 if (m_files[i]->state() != m_state)
3612 throw std::runtime_error("state mismatch");
3613 }
3614 }
3615
3616 virtual void unlock(_In_ fpos_t offset, _In_ fsize_t length)
3617 {
3618 if (m_files.empty())
3619 m_state = state_t::fail;
3620 m_files[0]->unlock(offset, length);
3621 m_state = m_files[0]->state();
3622 for (size_t i = 1, n = m_files.size(); i < n; ++i) {
3623 m_files[i]->unlock(offset, length);
3624 if (m_files[i]->state() != m_state)
3625 throw std::runtime_error("state mismatch");
3626 }
3627 }
3628
3629 virtual fsize_t size()
3630 {
3631 if (m_files.empty()) {
3632 m_state = state_t::fail;
3633 return 0;
3634 }
3635 fsize_t result = m_files[0]->size();
3636 m_state = m_files[0]->state();
3637 for (size_t i = 1, n = m_files.size(); i < n; ++i) {
3638 if (m_files[i]->size() != result)
3639 throw std::runtime_error("size mismatch");
3640 if (m_files[i]->state() != m_state)
3641 throw std::runtime_error("state mismatch");
3642 }
3643 return result;
3644 }
3645
3646 virtual void truncate()
3647 {
3648 if (m_files.empty())
3649 m_state = state_t::fail;
3650 m_files[0]->truncate();
3651 m_state = m_files[0]->state();
3652 for (size_t i = 1, n = m_files.size(); i < n; ++i) {
3653 m_files[i]->truncate();
3654 if (m_files[i]->state() != m_state)
3655 throw std::runtime_error("state mismatch");
3656 }
3657 }
3658
3659 protected:
3660 std::vector<basic_file*> m_files;
3661 std::vector<uint8_t> m_tmp;
3662 };
3663 }
3664}
Provides read-ahead stream capability.
Definition stream.hpp:1070
virtual size_t read(_Out_writes_bytes_to_opt_(length, return) void *data, size_t length)
Reads block of data from the stream.
Definition stream.hpp:1084
Provides write-back stream capability.
Definition stream.hpp:1138
virtual void flush()
Persists volatile element data.
Definition stream.hpp:1175
virtual size_t write(_In_reads_bytes_opt_(length) const void *data, size_t length)
Writes block of data to the stream.
Definition stream.hpp:1151
Basic seekable stream operations.
Definition stream.hpp:654
virtual void skip(fsize_t amount)
Skips given amount of bytes of data on the stream.
Definition stream.hpp:697
virtual time_point ctime() const
Returns file creation time.
Definition stream.hpp:744
virtual void lock(fpos_t offset, fsize_t length)
Locks file section for exclusive access.
Definition stream.hpp:713
virtual void truncate()=0
Sets file size - truncates the remainder of file content from the current file position to the end of...
charset_id read_charset(charset_id default_charset=charset_id::default)
Attempts to detect textfile charset based on UTF16 or UTF8 BOM.
Definition stream.hpp:817
fpos_t seekbeg(fpos_t offset)
Seeks to absolute file position.
Definition stream.hpp:681
virtual std::vector< uint8_t > read_remainder(size_t max_length=SIZE_MAX)
Reads and returns remainder of the stream.
Definition stream.hpp:656
virtual void set_mtime(time_point date)
Sets file modification time.
Definition stream.hpp:786
fpos_t seekcur(foff_t offset)
Seeks to relative from current file position.
Definition stream.hpp:688
virtual time_point atime() const
Returns file access time.
Definition stream.hpp:752
virtual void set_ctime(time_point date)
Sets file create time.
Definition stream.hpp:768
virtual fsize_t size()=0
Returns file size Should the file size cannot be determined, the method returns fsize_max and it does...
virtual void unlock(fpos_t offset, fsize_t length)
Unlocks file section for exclusive access.
Definition stream.hpp:723
virtual fpos_t tell() const =0
Returns absolute file position in file or fpos_max if fails. This method does not update stream state...
virtual time_point mtime() const
Returns file modification time.
Definition stream.hpp:760
fpos_t seekend(foff_t offset)
Seeks to relative from end file position.
Definition stream.hpp:695
virtual void set_atime(time_point date)
Sets file access time.
Definition stream.hpp:777
virtual fpos_t seek(foff_t offset, seek_t how=seek_t::beg)=0
Seeks to specified relative file position.
OS data stream (file, pipe, socket...)
Definition stream.hpp:1982
virtual size_t write(_In_reads_bytes_opt_(length) const void *data, size_t length)
Writes block of data to the stream.
Definition stream.hpp:2040
virtual void flush()
Persists volatile element data.
Definition stream.hpp:2098
virtual size_t read(_Out_writes_bytes_to_opt_(length, return) void *data, size_t length)
Reads block of data from the stream.
Definition stream.hpp:1989
virtual void close()
Closes the stream.
Definition stream.hpp:2087
‍UTF-8 byte-order-mark
Definition stream.hpp:71
bool ok() const
Returns true if the stream state is clean i.e. previous operation was succesful.
Definition stream.hpp:165
size_t write_vsprintf(_Printf_format_string_params_(2) const char *format, locale_t locale, va_list params)
Writes formatted string to the stream.
Definition stream.hpp:555
size_t write_array(_In_reads_or_z_opt_(num_chars) const wchar_t *wstr, size_t num_chars, charset_id charset)
Writes array of characters to the stream.
Definition stream.hpp:405
state_t state() const
Returns stream state after last operation.
Definition stream.hpp:160
basic & read_str(std::basic_string< _Elem, _Traits, _Ax > &data)
Reads length-prefixed string from the stream.
Definition stream.hpp:429
size_t write_sprintf(_Printf_format_string_params_(2) const wchar_t *format, locale_t locale,...)
Writes formatted string to the stream.
Definition stream.hpp:541
size_t write_vsprintf(_Printf_format_string_params_(2) const wchar_t *format, locale_t locale, va_list params)
Writes formatted string to the stream.
Definition stream.hpp:568
virtual void flush()
Persists volatile element data.
Definition stream.hpp:116
virtual void skip(fsize_t amount)
Skips given amount of bytes of data on the stream.
Definition stream.hpp:132
virtual void close()
Closes the stream.
Definition stream.hpp:124
uint8_t read_byte()
Reads one byte of data.
Definition stream.hpp:200
virtual std::vector< uint8_t > read_remainder(size_t max_length=SIZE_MAX)
Reads and returns remainder of the stream.
Definition stream.hpp:174
size_t write_sprintf(_Printf_format_string_params_(2) const char *format, locale_t locale,...)
Writes formatted string to the stream.
Definition stream.hpp:527
size_t readln_and_attach(std::basic_string< wchar_t, _Traits, _Ax > &wstr, charset_id charset)
Reads stream to the end-of-line or end-of-file and append to str.
Definition stream.hpp:358
size_t readln(std::basic_string< char, _Traits, _Ax > &str)
Reads stream to the end-of-line or end-of-file.
Definition stream.hpp:296
size_t readln_and_attach(std::basic_string< _Elem, _Traits, _Ax > &str)
Reads stream to the end-of-line or end-of-file and append to str.
Definition stream.hpp:337
size_t read_array(_Out_writes_bytes_(size *count) void *array, size_t size, size_t count)
Reads an array of data from the stream.
Definition stream.hpp:373
basic & write_str(const T *data)
Writes string to the stream length-prefixed.
Definition stream.hpp:454
virtual size_t read(_Out_writes_bytes_to_opt_(length, return) void *data, size_t length)
Reads block of data from the stream.
Definition stream.hpp:86
size_t readln(std::basic_string< wchar_t, _Traits, _Ax > &wstr)
Reads stream to the end-of-line or end-of-file.
Definition stream.hpp:308
void write_charset(charset_id charset)
Writes UTF8 or UTF-16 byte-order-mark.
Definition stream.hpp:514
size_t readln(std::basic_string< wchar_t, _Traits, _Ax > &wstr, charset_id charset)
Reads stream to the end-of-line or end-of-file.
Definition stream.hpp:320
basic & write_data(const T data)
Writes one primitive data type.
Definition stream.hpp:277
fsize_t write_stream(basic &stream, fsize_t amount=fsize_max)
Writes content of another stream.
Definition stream.hpp:489
virtual size_t write(_In_reads_bytes_opt_(length) const void *data, size_t length)
Writes block of data to the stream.
Definition stream.hpp:104
size_t write_array(_In_reads_bytes_opt_(size *count) const void *array, size_t size, size_t count)
Writes an array of data to the stream.
Definition stream.hpp:391
void write_byte(uint8_t byte, fsize_t amount=1)
Writes a byte of data.
Definition stream.hpp:211
basic & read_data(T &data)
Reads one primitive data type.
Definition stream.hpp:249
Buffered read/write stream.
Definition stream.hpp:1209
virtual void flush()
Persists volatile element data.
Definition stream.hpp:1312
virtual size_t read(_Out_writes_bytes_to_opt_(length, return) void *data, size_t length)
Reads block of data from the stream.
Definition stream.hpp:1230
virtual size_t write(_In_reads_bytes_opt_(length) const void *data, size_t length)
Writes block of data to the stream.
Definition stream.hpp:1266
Buffered OS data stream (file, pipe, socket...)
Definition stream.hpp:2112
Cached file.
Definition stream.hpp:1629
virtual time_point ctime() const
Returns file creation time.
Definition stream.hpp:1861
fpos_t m_offset
Logical absolute file position.
Definition stream.hpp:1970
virtual void truncate()
Sets file size - truncates the remainder of file content from the current file position to the end of...
Definition stream.hpp:1839
virtual size_t read(_Out_writes_bytes_to_opt_(length, return) void *data, size_t length)
Reads block of data from the stream.
Definition stream.hpp:1669
virtual time_point atime() const
Returns file access time.
Definition stream.hpp:1866
virtual fsize_t size()
Returns file size Should the file size cannot be determined, the method returns fsize_max and it does...
Definition stream.hpp:1834
virtual void unlock(fpos_t offset, fsize_t length)
Unlocks file section for exclusive access.
Definition stream.hpp:1828
virtual time_point mtime() const
Returns file modification time.
Definition stream.hpp:1875
virtual void close()
Closes the stream.
Definition stream.hpp:1784
virtual void set_mtime(time_point date)
Sets file modification time.
Definition stream.hpp:1897
virtual void lock(fpos_t offset, fsize_t length)
Locks file section for exclusive access.
Definition stream.hpp:1822
virtual void flush()
Persists volatile element data.
Definition stream.hpp:1791
virtual size_t write(_In_reads_bytes_opt_(length) const void *data, size_t length)
Writes block of data to the stream.
Definition stream.hpp:1731
virtual void set_ctime(time_point date)
Sets file create time.
Definition stream.hpp:1884
virtual fpos_t tell() const
Returns absolute file position in file or fpos_max if fails. This method does not update stream state...
Definition stream.hpp:1817
virtual void set_atime(time_point date)
Sets file access time.
Definition stream.hpp:1889
virtual fpos_t seek(foff_t offset, seek_t how=seek_t::beg)
Seeks to specified relative file position.
Definition stream.hpp:1802
Cached file-system file.
Definition stream.hpp:2660
cached_file(const sys_char *filename, int mode, size_t cache_size=default_cache_size)
Opens file.
Definition stream.hpp:2676
void open(const sys_char *filename, int mode)
Opens file.
Definition stream.hpp:2690
Modifies data on the fly when reading from/writing to a source stream.
Definition stream.hpp:847
virtual void flush()
Persists volatile element data.
Definition stream.hpp:888
virtual void close()
Closes the stream.
Definition stream.hpp:882
virtual size_t read(_Out_writes_bytes_to_opt_(length, return) void *data, size_t length)
Reads block of data from the stream.
Definition stream.hpp:866
virtual size_t write(_In_reads_bytes_opt_(length) const void *data, size_t length)
Writes block of data to the stream.
Definition stream.hpp:874
Compares multiple files to perform the same.
Definition stream.hpp:3493
virtual fsize_t size()
Returns file size Should the file size cannot be determined, the method returns fsize_max and it does...
Definition stream.hpp:3629
virtual void truncate()
Sets file size - truncates the remainder of file content from the current file position to the end of...
Definition stream.hpp:3646
virtual size_t write(_In_reads_bytes_opt_(length) const void *data, size_t length)
Writes block of data to the stream.
Definition stream.hpp:3524
virtual void close()
Closes the stream.
Definition stream.hpp:3557
virtual void lock(fpos_t offset, fsize_t length)
Locks file section for exclusive access.
Definition stream.hpp:3603
virtual void unlock(fpos_t offset, fsize_t length)
Unlocks file section for exclusive access.
Definition stream.hpp:3616
virtual fpos_t seek(foff_t offset, seek_t how=seek_t::beg)
Seeks to specified relative file position.
Definition stream.hpp:3574
virtual fpos_t tell() const
Returns absolute file position in file or fpos_max if fails. This method does not update stream state...
Definition stream.hpp:3591
virtual void flush()
Persists volatile element data.
Definition stream.hpp:3542
virtual size_t read(_Out_writes_bytes_to_opt_(length, return) void *data, size_t length)
Reads block of data from the stream.
Definition stream.hpp:3501
In-memory FIFO queue.
Definition stream.hpp:3393
virtual void close()
Closes the stream.
Definition stream.hpp:3464
virtual size_t write(_In_reads_bytes_opt_(length) const void *data, size_t length)
Writes block of data to the stream.
Definition stream.hpp:3441
size_t size() const
Returns total size of pending data in the queue.
Definition stream.hpp:3478
virtual size_t read(_Out_writes_bytes_to_opt_(length, return) void *data, size_t length)
Reads block of data from the stream.
Definition stream.hpp:3412
Limits file reading/writing to a predefined window.
Definition stream.hpp:1519
virtual void truncate()
Sets file size - truncates the remainder of file content from the current file position to the end of...
Definition stream.hpp:1612
virtual void flush()
Persists volatile element data.
Definition stream.hpp:1562
virtual void skip(fsize_t amount)
Skips given amount of bytes of data on the stream.
Definition stream.hpp:1575
virtual fpos_t seek(foff_t offset, seek_t how=seek_t::beg)
Seeks to specified relative file position.
Definition stream.hpp:1568
virtual fsize_t size()
Returns file size Should the file size cannot be determined, the method returns fsize_max and it does...
Definition stream.hpp:1607
virtual size_t write(_In_reads_bytes_opt_(length) const void *data, size_t length)
Writes block of data to the stream.
Definition stream.hpp:1542
virtual void lock(fpos_t offset, fsize_t length)
Locks file section for exclusive access.
Definition stream.hpp:1587
virtual size_t read(_Out_writes_bytes_to_opt_(length, return) void *data, size_t length)
Reads block of data from the stream.
Definition stream.hpp:1528
virtual void unlock(fpos_t offset, fsize_t length)
Unlocks file section for exclusive access.
Definition stream.hpp:1597
virtual fpos_t tell() const
Returns absolute file position in file or fpos_max if fails. This method does not update stream state...
Definition stream.hpp:1581
virtual void close()
Closes the stream.
Definition stream.hpp:1556
File-system file.
Definition stream.hpp:2350
virtual time_point mtime() const
Returns file modification time.
Definition stream.hpp:2593
virtual void unlock(fpos_t offset, fsize_t length)
Unlocks file section for exclusive access.
Definition stream.hpp:2486
file(const sys_char *filename, int mode)
Opens file.
Definition stream.hpp:2360
virtual void set_ctime(time_point date)
Sets file create time.
Definition stream.hpp:2607
virtual time_point atime() const
Returns file access time.
Definition stream.hpp:2579
virtual void set_mtime(time_point date)
Sets file modification time.
Definition stream.hpp:2637
virtual void set_atime(time_point date)
Sets file access time.
Definition stream.hpp:2619
void open(const sys_char *filename, int mode)
Opens file.
Definition stream.hpp:2371
virtual void lock(fpos_t offset, fsize_t length)
Locks file section for exclusive access.
Definition stream.hpp:2463
virtual void truncate()
Sets file size - truncates the remainder of file content from the current file position to the end of...
Definition stream.hpp:2529
virtual time_point ctime() const
Returns file creation time.
Definition stream.hpp:2569
virtual fsize_t size()
Returns file size Should the file size cannot be determined, the method returns fsize_max and it does...
Definition stream.hpp:2511
virtual fpos_t seek(foff_t offset, seek_t how=seek_t::beg)
Seeks to specified relative file position.
Definition stream.hpp:2424
virtual fpos_t tell() const
Returns absolute file position in file or fpos_max if fails. This method does not update stream state...
Definition stream.hpp:2445
Limits reading from/writing to stream to a predefined number of bytes.
Definition stream.hpp:1360
fsize_t read_limit
Number of bytes left that may be read from the stream.
Definition stream.hpp:1420
virtual size_t read(_Out_writes_bytes_to_opt_(length, return) void *data, size_t length)
Reads block of data from the stream.
Definition stream.hpp:1368
virtual size_t write(_In_reads_bytes_opt_(length) const void *data, size_t length)
Writes block of data to the stream.
Definition stream.hpp:1393
fsize_t write_limit
Number of bytes left, that can be written to the stream.
Definition stream.hpp:1421
In-memory file.
Definition stream.hpp:2713
void load(const sys_char *filename, int mode)
Loads content from a file-system file.
Definition stream.hpp:2837
size_t m_size
file size
Definition stream.hpp:3380
void get(fpos_t offset, T &data)
Reads data from specified file location This does not move file pointer. It checks for data size asse...
Definition stream.hpp:3313
size_t write_stream(basic &stream, size_t amount=SIZE_MAX)
Writes content of another stream.
Definition stream.hpp:3135
uint8_t * m_data
file data
Definition stream.hpp:3377
memory_file & read_data(T &data)
Reads one primitive data type.
Definition stream.hpp:2934
virtual void close()
Closes the stream.
Definition stream.hpp:3185
virtual size_t read(_Out_writes_bytes_to_opt_(length, return) void *data, size_t length)
Reads block of data from the stream.
Definition stream.hpp:2895
virtual fpos_t tell() const
Returns absolute file position in file or fpos_max if fails. This method does not update stream state...
Definition stream.hpp:3216
size_t m_reserved
reserved file size
Definition stream.hpp:3381
memory_file(size_t size, state_t state=state_t::ok)
Creates an empty file of reserved size.
Definition stream.hpp:2734
void reserve(size_t required, bool tight=false) noexcept
Reallocates memory.
Definition stream.hpp:2808
memory_file & read_str(std::basic_string< _Elem, _Traits, _Ax > &data)
Reads length-prefixed string from the stream.
Definition stream.hpp:2974
void write_byte(uint8_t byte, size_t amount=1)
Writes a byte of data.
Definition stream.hpp:3029
void set(fpos_t offset, const T data)
Writes data to specified file location This does not move file pointer nor update file size....
Definition stream.hpp:3276
size_t m_offset
file pointer
Definition stream.hpp:3379
memory_file(const sys_char *filename, int mode)
Loads content from file-system file.
Definition stream.hpp:2791
virtual fsize_t size()
Returns file size Should the file size cannot be determined, the method returns fsize_max and it does...
Definition stream.hpp:3221
virtual fpos_t seek(foff_t offset, seek_t how=seek_t::beg)
Seeks to specified relative file position.
Definition stream.hpp:3199
virtual void truncate()
Sets file size - truncates the remainder of file content from the current file position to the end of...
Definition stream.hpp:3226
memory_file & write_data(const T data)
Writes one primitive data type.
Definition stream.hpp:3062
memory_file & write_str(const T *data)
Writes string to the stream length-prefixed.
Definition stream.hpp:3100
bool m_manage
may reallocate m_data?
Definition stream.hpp:3378
memory_file(void *data, size_t size, bool manage=false, state_t state=state_t::ok)
Creates a file based on available data.
Definition stream.hpp:2781
virtual size_t write(_In_reads_bytes_opt_(length) const void *data, size_t length)
Writes block of data to the stream.
Definition stream.hpp:3005
void save(const sys_char *filename, int mode)
Saves content to a file-system file.
Definition stream.hpp:2870
memory_file(void *data, size_t size, size_t reserved, bool manage=false, state_t state=state_t::ok)
Creates a file based on available data.
Definition stream.hpp:2758
const void * data() const
Returns pointer to data.
Definition stream.hpp:2893
Definition stream.hpp:985
enum stdex::stream::replicator::worker::op_t op
Operation to perform.
size_t num_written
Number of bytes written.
Definition stream.hpp:1032
size_t length
Byte limit of data to write.
Definition stream.hpp:1031
const void * data
Data to write.
Definition stream.hpp:1030
Replicates writing of the same data to multiple streams.
Definition stream.hpp:902
void push_back(basic *source)
Adds stream on the list.
Definition stream.hpp:921
virtual void flush()
Persists volatile element data.
Definition stream.hpp:978
void remove(basic *source)
Removes stream from the list.
Definition stream.hpp:929
virtual size_t write(_In_reads_bytes_opt_(length) const void *data, size_t length)
Writes block of data to the stream.
Definition stream.hpp:946
virtual void close()
Closes the stream.
Definition stream.hpp:973
Limits reading from/writing to stream to a predefined window.
Definition stream.hpp:1428
fpos_t write_offset
Number of bytes to discard on write.
Definition stream.hpp:1512
virtual size_t write(_In_reads_bytes_opt_(length) const void *data, size_t length)
Writes block of data to the stream.
Definition stream.hpp:1469
virtual size_t read(_Out_writes_bytes_to_opt_(length, return) void *data, size_t length)
Reads block of data from the stream.
Definition stream.hpp:1436
fpos_t read_offset
Number of bytes to skip on read.
Definition stream.hpp:1511
Operating system object (file, pipe, anything with an OS handle etc.)
Definition system.hpp:57
virtual void close()
Closes object.
Definition system.hpp:98
Numerical interval.
Definition interval.hpp:18
bool contains(T x) const
Is value in interval?
Definition interval.hpp:88
bool empty() const
Is interval empty?
Definition interval.hpp:54
T size() const
Returns interval size.
Definition interval.hpp:47
T end
interval end
Definition interval.hpp:20
T start
interval start
Definition interval.hpp:19
Definition stream.hpp:1337
Definition stream.hpp:1948
interval< fpos_t > region
valid data region
Definition stream.hpp:1956
Definition stream.hpp:3482