src/examples/cpp03/chat/chat_message.hpp | src/examples/cpp11/chat/chat_message.hpp |
⋮ | ⋮ |
1 | // | 1 | // |
2 | //·chat_message.hpp | 2 | //·chat_message.hpp |
3 | //·~~~~~~~~~~~~~~~~ | 3 | //·~~~~~~~~~~~~~~~~ |
4 | // | 4 | // |
5 | //·Copyright·(c)·2003-2018·Christopher·M.·Kohlhoff·(chris·at·kohlhoff·dot·com) | 5 | //·Copyright·(c)·2003-2018·Christopher·M.·Kohlhoff·(chris·at·kohlhoff·dot·com) |
6 | // | 6 | // |
7 | //·Distributed·under·the·Boost·Software·License,·Version·1.0.·(See·accompanying | 7 | //·Distributed·under·the·Boost·Software·License,·Version·1.0.·(See·accompanying |
8 | //·file·LICENSE_1_0.txt·or·copy·at·http://www.boost.org/LICENSE_1_0.txt) | 8 | //·file·LICENSE_1_0.txt·or·copy·at·http://www.boost.org/LICENSE_1_0.txt) |
9 | // | 9 | // |
10 | | 10 | |
11 | #ifndef·CHAT_MESSAGE_HPP | 11 | #ifndef·CHAT_MESSAGE_HPP |
12 | #define·CHAT_MESSAGE_HPP | 12 | #define·CHAT_MESSAGE_HPP |
13 | | 13 | |
14 | #include·<cstdio> | 14 | #include·<cstdio> |
15 | #include·<cstdlib> | 15 | #include·<cstdlib> |
16 | #include·<cstring> | 16 | #include·<cstring> |
17 | | 17 | |
18 | class·chat_message | 18 | class·chat_message |
19 | { | 19 | { |
20 | public: | 20 | public: |
21 | ··enum·{·header_length·=·4·}; | 21 | ··enum·{·header_length·=·4·}; |
22 | ··enum·{·max_body_length·=·512·}; | 22 | ··enum·{·max_body_length·=·512·}; |
23 | | 23 | |
24 | ··chat_message() | 24 | ··chat_message() |
25 | ····:·body_length_(0) | 25 | ····:·body_length_(0) |
26 | ··{ | 26 | ··{ |
27 | ··} | 27 | ··} |
28 | | 28 | |
29 | ··const·char*·data()·const | 29 | ··const·char*·data()·const |
30 | ··{ | 30 | ··{ |
31 | ····return·data_; | 31 | ····return·data_; |
32 | ··} | 32 | ··} |
33 | | 33 | |
34 | ··char*·data() | 34 | ··char*·data() |
35 | ··{ | 35 | ··{ |
36 | ····return·data_; | 36 | ····return·data_; |
37 | ··} | 37 | ··} |
38 | | 38 | |
39 | ··size_t·length()·const | 39 | ··std::size_t·length()·const |
40 | ··{ | 40 | ··{ |
41 | ····return·header_length·+·body_length_; | 41 | ····return·header_length·+·body_length_; |
42 | ··} | 42 | ··} |
43 | | 43 | |
44 | ··const·char*·body()·const | 44 | ··const·char*·body()·const |
45 | ··{ | 45 | ··{ |
46 | ····return·data_·+·header_length; | 46 | ····return·data_·+·header_length; |
47 | ··} | 47 | ··} |
48 | | 48 | |
49 | ··char*·body() | 49 | ··char*·body() |
50 | ··{ | 50 | ··{ |
51 | ····return·data_·+·header_length; | 51 | ····return·data_·+·header_length; |
52 | ··} | 52 | ··} |
53 | | 53 | |
54 | ··size_t·body_length()·const | 54 | ··std::size_t·body_length()·const |
55 | ··{ | 55 | ··{ |
56 | ····return·body_length_; | 56 | ····return·body_length_; |
57 | ··} | 57 | ··} |
58 | | 58 | |
59 | ··void·body_length(size_t·new_length) | 59 | ··void·body_length(std::size_t·new_length) |
60 | ··{ | 60 | ··{ |
61 | ····body_length_·=·new_length; | 61 | ····body_length_·=·new_length; |
62 | ····if·(body_length_·>·max_body_length) | 62 | ····if·(body_length_·>·max_body_length) |
63 | ······body_length_·=·max_body_length; | 63 | ······body_length_·=·max_body_length; |
64 | ··} | 64 | ··} |
65 | | 65 | |
66 | ··bool·decode_header() | 66 | ··bool·decode_header() |
67 | ··{ | 67 | ··{ |
68 | ····using·namespace·std;·//·For·strncat·and·atoi. | |
69 | ····char·header[header_length·+·1]·=·""; | 68 | ····char·header[header_length·+·1]·=·""; |
70 | ····strncat(header,·data_,·header_length); | 69 | ····std::strncat(header,·data_,·header_length); |
71 | ····body_length_·=·atoi(header); | 70 | ····body_length_·=·std::atoi(header); |
72 | ····if·(body_length_·>·max_body_length) | 71 | ····if·(body_length_·>·max_body_length) |
73 | ····{ | 72 | ····{ |
74 | ······body_length_·=·0; | 73 | ······body_length_·=·0; |
75 | ······return·false; | 74 | ······return·false; |
76 | ····} | 75 | ····} |
77 | ····return·true; | 76 | ····return·true; |
78 | ··} | 77 | ··} |
79 | | 78 | |
80 | ··void·encode_header() | 79 | ··void·encode_header() |
81 | ··{ | 80 | ··{ |
82 | ····using·namespace·std;·//·For·sprintf·and·memcpy. | |
83 | ····char·header[header_length·+·1]·=·""; | 81 | ····char·header[header_length·+·1]·=·""; |
84 | ····sprintf(header,·"%4d",·static_cast<int>(body_length_)); | 82 | ····std::sprintf(header,·"%4d",·static_cast<int>(body_length_)); |
85 | ····memcpy(data_,·header,·header_length); | 83 | ····std::memcpy(data_,·header,·header_length); |
86 | ··} | 84 | ··} |
87 | | 85 | |
88 | private: | 86 | private: |
89 | ··char·data_[header_length·+·max_body_length]; | 87 | ··char·data_[header_length·+·max_body_length]; |
90 | ··size_t·body_length_; | 88 | ··std::size_t·body_length_; |
91 | }; | 89 | }; |
92 | | 90 | |
93 | #endif·//·CHAT_MESSAGE_HPP | 91 | #endif·//·CHAT_MESSAGE_HPP |