blob: 486cfc8165c86090791963a05bbe7861879909a6 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
#pragma once
#include <util/generic/string.h>
#include <util/stream/output.h>
#include <util/stream/zerocopy.h>
namespace NClickHouse {
/**
* Class which reads and decodes binary data which is composed of varint-
* encoded integers and fixed-width pieces.
*/
class TCodedInputStream {
public:
TCodedInputStream() = default;
/// Create a CodedInputStream that reads from the given ZeroCopyInput.
explicit TCodedInputStream(IZeroCopyInput* input);
// Read an unsigned integer with Varint encoding, truncating to 32 bits.
// Reading a 32-bit value is equivalent to reading a 64-bit one and casting
// it to uint32, but may be more efficient.
bool ReadVarint32(ui32* value);
// Read an unsigned integer with Varint encoding.
bool ReadVarint64(ui64* value);
// Read raw bytes, copying them into the given buffer.
bool ReadRaw(void* buffer, size_t size);
// Like ReadRaw, but reads into a string.
//
// Implementation Note: ReadString() grows the string gradually as it
// reads in the data, rather than allocating the entire requested size
// upfront. This prevents denial-of-service attacks in which a client
// could claim that a string is going to be MAX_INT bytes long in order to
// crash the server because it can't allocate this much space at once.
bool ReadString(TString* buffer, int size);
// Skips a number of bytes. Returns false if an underlying read error
// occurs.
bool Skip(size_t count);
private:
IZeroCopyInput* Input_;
};
class TCodedOutputStream {
public:
TCodedOutputStream() = default;
/// Create a CodedInputStream that writes to the given ZeroCopyOutput.
explicit TCodedOutputStream(IOutputStream* output);
void Flush();
// Write raw bytes, copying them from the given buffer.
void WriteRaw(const void* buffer, int size);
/// Write an unsigned integer with Varint encoding.
void WriteVarint64(const ui64 value);
private:
IOutputStream* Output_;
};
}
|
