|
jpayne@69
|
1 // Copyright (c) 2014 Sandstorm Development Group, Inc. and contributors
|
|
jpayne@69
|
2 // Licensed under the MIT License:
|
|
jpayne@69
|
3 //
|
|
jpayne@69
|
4 // Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
jpayne@69
|
5 // of this software and associated documentation files (the "Software"), to deal
|
|
jpayne@69
|
6 // in the Software without restriction, including without limitation the rights
|
|
jpayne@69
|
7 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
jpayne@69
|
8 // copies of the Software, and to permit persons to whom the Software is
|
|
jpayne@69
|
9 // furnished to do so, subject to the following conditions:
|
|
jpayne@69
|
10 //
|
|
jpayne@69
|
11 // The above copyright notice and this permission notice shall be included in
|
|
jpayne@69
|
12 // all copies or substantial portions of the Software.
|
|
jpayne@69
|
13 //
|
|
jpayne@69
|
14 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
jpayne@69
|
15 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
jpayne@69
|
16 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
jpayne@69
|
17 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
jpayne@69
|
18 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
jpayne@69
|
19 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
jpayne@69
|
20 // THE SOFTWARE.
|
|
jpayne@69
|
21
|
|
jpayne@69
|
22 /*
|
|
jpayne@69
|
23 * Compatibility layer for stdlib iostream
|
|
jpayne@69
|
24 */
|
|
jpayne@69
|
25
|
|
jpayne@69
|
26 #pragma once
|
|
jpayne@69
|
27
|
|
jpayne@69
|
28 #include "../io.h"
|
|
jpayne@69
|
29 #include <iostream>
|
|
jpayne@69
|
30
|
|
jpayne@69
|
31 KJ_BEGIN_HEADER
|
|
jpayne@69
|
32
|
|
jpayne@69
|
33 namespace kj {
|
|
jpayne@69
|
34 namespace std {
|
|
jpayne@69
|
35
|
|
jpayne@69
|
36 class StdOutputStream: public kj::OutputStream {
|
|
jpayne@69
|
37
|
|
jpayne@69
|
38 public:
|
|
jpayne@69
|
39 explicit StdOutputStream(::std::ostream& stream) : stream_(stream) {}
|
|
jpayne@69
|
40 ~StdOutputStream() noexcept(false) {}
|
|
jpayne@69
|
41
|
|
jpayne@69
|
42 virtual void write(const void* src, size_t size) override {
|
|
jpayne@69
|
43 // Always writes the full size.
|
|
jpayne@69
|
44
|
|
jpayne@69
|
45 stream_.write((char*)src, size);
|
|
jpayne@69
|
46 }
|
|
jpayne@69
|
47
|
|
jpayne@69
|
48 virtual void write(ArrayPtr<const ArrayPtr<const byte>> pieces) override {
|
|
jpayne@69
|
49 // Equivalent to write()ing each byte array in sequence, which is what the
|
|
jpayne@69
|
50 // default implementation does. Override if you can do something better,
|
|
jpayne@69
|
51 // e.g. use writev() to do the write in a single syscall.
|
|
jpayne@69
|
52
|
|
jpayne@69
|
53 for (auto piece : pieces) {
|
|
jpayne@69
|
54 write(piece.begin(), piece.size());
|
|
jpayne@69
|
55 }
|
|
jpayne@69
|
56 }
|
|
jpayne@69
|
57
|
|
jpayne@69
|
58 private:
|
|
jpayne@69
|
59 ::std::ostream& stream_;
|
|
jpayne@69
|
60
|
|
jpayne@69
|
61 };
|
|
jpayne@69
|
62
|
|
jpayne@69
|
63 class StdInputStream: public kj::InputStream {
|
|
jpayne@69
|
64
|
|
jpayne@69
|
65 public:
|
|
jpayne@69
|
66 explicit StdInputStream(::std::istream& stream) : stream_(stream) {}
|
|
jpayne@69
|
67 ~StdInputStream() noexcept(false) {}
|
|
jpayne@69
|
68
|
|
jpayne@69
|
69 virtual size_t tryRead(
|
|
jpayne@69
|
70 void* buffer, size_t minBytes, size_t maxBytes) override {
|
|
jpayne@69
|
71 // Like read(), but may return fewer than minBytes on EOF.
|
|
jpayne@69
|
72
|
|
jpayne@69
|
73 stream_.read((char*)buffer, maxBytes);
|
|
jpayne@69
|
74 return stream_.gcount();
|
|
jpayne@69
|
75 }
|
|
jpayne@69
|
76
|
|
jpayne@69
|
77 private:
|
|
jpayne@69
|
78 ::std::istream& stream_;
|
|
jpayne@69
|
79
|
|
jpayne@69
|
80 };
|
|
jpayne@69
|
81
|
|
jpayne@69
|
82 } // namespace std
|
|
jpayne@69
|
83 } // namespace kj
|
|
jpayne@69
|
84
|
|
jpayne@69
|
85 KJ_END_HEADER
|
