← Back to branch summary

~peekabot-devs/peekabot/peekabot.org 

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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197

<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
<html xmlns="http://www.w3.org/1999/xhtml">
<head>
<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
<title>peekabot: src/Connection.hh Source File</title>
<link href="tabs.css" rel="stylesheet" type="text/css"/>
<link href="doxygen.css" rel="stylesheet" type="text/css"/>
</head>
<body>
<!-- Generated by Doxygen 1.7.1 -->
<div class="navigation" id="top">
  <div class="tabs">
    <ul class="tablist">
      <li><a href="main.html"><span>Main&nbsp;Page</span></a></li>
      <li><a href="pages.html"><span>Related&nbsp;Pages</span></a></li>
      <li><a href="annotated.html"><span>Classes</span></a></li>
      <li class="current"><a href="files.html"><span>Files</span></a></li>
    </ul>
  </div>
  <div class="tabs2">
    <ul class="tablist">
      <li><a href="files.html"><span>File&nbsp;List</span></a></li>
    </ul>
  </div>
<div class="header">
  <div class="headertitle">
<h1>src/Connection.hh</h1>  </div>
</div>
<div class="contents">
<div class="fragment"><pre class="fragment"><a name="l00001"></a>00001 <span class="comment">/*</span>
<a name="l00002"></a>00002 <span class="comment"> * Copyright Staffan Gimåker 2010.</span>
<a name="l00003"></a>00003 <span class="comment"> *</span>
<a name="l00004"></a>00004 <span class="comment"> * ---</span>
<a name="l00005"></a>00005 <span class="comment"> *</span>
<a name="l00006"></a>00006 <span class="comment"> * Distributed under the Boost Software License, Version 1.0.</span>
<a name="l00007"></a>00007 <span class="comment"> * (See accompanying file LICENSE_1_0.txt or copy at</span>
<a name="l00008"></a>00008 <span class="comment"> * http://www.boost.org/LICENSE_1_0.txt)</span>
<a name="l00009"></a>00009 <span class="comment"> */</span>
<a name="l00010"></a>00010 
<a name="l00011"></a>00011 <span class="preprocessor">#ifndef PEEKABOT_CONNECTION_HH_INCLUDED</span>
<a name="l00012"></a>00012 <span class="preprocessor"></span><span class="preprocessor">#define PEEKABOT_CONNECTION_HH_INCLUDED</span>
<a name="l00013"></a>00013 <span class="preprocessor"></span>
<a name="l00014"></a>00014 
<a name="l00015"></a>00015 <span class="preprocessor">#include &lt;string&gt;</span>
<a name="l00016"></a>00016 <span class="preprocessor">#include &lt;vector&gt;</span>
<a name="l00017"></a>00017 <span class="preprocessor">#include &lt;boost/asio.hpp&gt;</span>
<a name="l00018"></a>00018 <span class="preprocessor">#include &lt;boost/cstdint.hpp&gt;</span>
<a name="l00019"></a>00019 <span class="preprocessor">#include &lt;boost/function.hpp&gt;</span>
<a name="l00020"></a>00020 <span class="preprocessor">#include &lt;boost/shared_ptr.hpp&gt;</span>
<a name="l00021"></a>00021 <span class="preprocessor">#include &lt;boost/shared_array.hpp&gt;</span>
<a name="l00022"></a>00022 <span class="preprocessor">#include &lt;boost/date_time/posix_time/ptime.hpp&gt;</span>
<a name="l00023"></a>00023 
<a name="l00024"></a>00024 <span class="preprocessor">#include &quot;serialization/MemAdapters.hh&quot;</span>
<a name="l00025"></a>00025 <span class="preprocessor">#include &quot;Protocol.hh&quot;</span>
<a name="l00026"></a>00026 
<a name="l00027"></a>00027 
<a name="l00028"></a>00028 <span class="keyword">namespace </span>peekabot
<a name="l00029"></a>00029 {
<a name="l00030"></a>00030     <span class="keyword">class </span>Action;
<a name="l00031"></a>00031 
<a name="l00032"></a>00032     <span class="keyword">class </span>Connection
<a name="l00033"></a>00033     {
<a name="l00034"></a>00034         <span class="keyword">struct </span>ActionWriteData
<a name="l00035"></a>00035         {
<a name="l00036"></a>00036             boost::uint8_t m_ctrl_byte;
<a name="l00037"></a>00037             boost::uint32_t m_uncomp_len;
<a name="l00038"></a>00038             boost::uint32_t m_comp_len;
<a name="l00039"></a>00039             MemSerializationBuffer m_buf;
<a name="l00040"></a>00040         };
<a name="l00041"></a>00041 
<a name="l00042"></a>00042     <span class="keyword">public</span>:
<a name="l00043"></a>00043         <span class="keyword">virtual</span> ~Connection() {}
<a name="l00044"></a>00044 
<a name="l00045"></a>00045         <span class="keywordtype">void</span> connect(<span class="keyword">const</span> std::string &amp;host, <span class="keyword">const</span> std::string &amp;port);
<a name="l00046"></a>00046 
<a name="l00047"></a>00047         <span class="keywordtype">void</span> close();
<a name="l00048"></a>00048 
<a name="l00049"></a>00049         boost::asio::io_service &amp;get_io_service();
<a name="l00050"></a>00050 
<a name="l00051"></a>00051         boost::asio::ip::tcp::socket &amp;get_socket();
<a name="l00052"></a>00052 
<a name="l00053"></a>00053         <span class="keyword">const</span> boost::asio::ip::tcp::socket &amp;get_socket() <span class="keyword">const</span>;
<a name="l00054"></a>00054 
<a name="l00055"></a>00055         <span class="keywordtype">bool</span> is_authenticated() <span class="keyword">const</span>;
<a name="l00056"></a>00056 
<a name="l00057"></a>00057         <span class="keywordtype">bool</span> is_open() <span class="keyword">const</span>;
<a name="l00058"></a>00058 
<a name="l00059"></a>00059         <span class="keywordtype">bool</span> is_operational() <span class="keyword">const</span>;
<a name="l00060"></a>00060 
<a name="l00061"></a>00061         boost::posix_time::time_duration get_uptime() <span class="keyword">const</span>;
<a name="l00062"></a>00062 
<a name="l00063"></a>00063         boost::uint64_t get_bytes_sent() <span class="keyword">const</span>;
<a name="l00064"></a>00064 
<a name="l00065"></a>00065         boost::uint64_t get_bytes_received() <span class="keyword">const</span>;
<a name="l00066"></a>00066 
<a name="l00067"></a>00067         <span class="keywordtype">void</span> get_throughput(<span class="keywordtype">float</span> &amp;in, <span class="keywordtype">float</span> &amp;out) <span class="keyword">const</span>;
<a name="l00068"></a>00068 
<a name="l00069"></a>00069         <span class="keywordtype">void</span> get_remote_peekabot_version(
<a name="l00070"></a>00070             boost::uint8_t &amp;major,
<a name="l00071"></a>00071             boost::uint8_t &amp;minor,
<a name="l00072"></a>00072             boost::uint8_t &amp;rev,
<a name="l00073"></a>00073             boost::uint8_t &amp;rc);
<a name="l00074"></a>00074 
<a name="l00075"></a>00075         std::string get_remote_address() <span class="keyword">const</span>;
<a name="l00076"></a>00076 
<a name="l00077"></a>00077         <span class="keyword">virtual</span> <span class="keywordtype">void</span> start() = 0;
<a name="l00078"></a>00078 
<a name="l00079"></a>00079         <span class="keywordtype">void</span> authenticate();
<a name="l00080"></a>00080 
<a name="l00081"></a>00081         <span class="keywordtype">void</span> write_action(<span class="keyword">const</span> boost::shared_ptr&lt;Action&gt; &amp;action);
<a name="l00082"></a>00082 
<a name="l00083"></a>00083     <span class="keyword">protected</span>:
<a name="l00084"></a>00084         Connection(boost::asio::io_service &amp;io_service);
<a name="l00085"></a>00085 
<a name="l00086"></a>00086         <span class="comment">//</span>
<a name="l00087"></a>00087 
<a name="l00088"></a>00088         <span class="keywordtype">void</span> read_action();
<a name="l00089"></a>00089 
<a name="l00090"></a>00090         <span class="comment">//</span>
<a name="l00091"></a>00091 
<a name="l00092"></a>00092         <span class="keyword">virtual</span> boost::shared_ptr&lt;Connection&gt; ptr() = 0;
<a name="l00093"></a>00093 
<a name="l00094"></a>00094         <span class="keyword">virtual</span> <span class="keywordtype">void</span> action_read(
<a name="l00095"></a>00095             <span class="keyword">const</span> boost::shared_ptr&lt;Action&gt; &amp;action) = 0;
<a name="l00096"></a>00096 
<a name="l00097"></a>00097         <span class="keyword">virtual</span> <span class="keywordtype">void</span> on_authenticated() = 0;
<a name="l00098"></a>00098 
<a name="l00099"></a>00099         <span class="keyword">virtual</span> <span class="keywordtype">void</span> on_authentication_failed(
<a name="l00100"></a>00100             protocol::AuthenticationResult reason) = 0;
<a name="l00101"></a>00101 
<a name="l00102"></a>00102     <span class="keyword">private</span>:
<a name="l00103"></a>00103         <span class="keywordtype">void</span> handle_auth_write(<span class="keyword">const</span> boost::system::error_code &amp;e);
<a name="l00104"></a>00104 
<a name="l00105"></a>00105         <span class="comment">//</span>
<a name="l00106"></a>00106 
<a name="l00107"></a>00107         <span class="keywordtype">void</span> authenticate_();
<a name="l00108"></a>00108 
<a name="l00109"></a>00109         <span class="keywordtype">void</span> handle_auth_data_read(
<a name="l00110"></a>00110             <span class="keyword">const</span> boost::system::error_code &amp;e,
<a name="l00111"></a>00111             boost::shared_ptr&lt;boost::uint32_t&gt; unique_id,
<a name="l00112"></a>00112             boost::shared_ptr&lt;boost::uint32_t&gt; protocol_version);
<a name="l00113"></a>00113 
<a name="l00114"></a>00114         <span class="keywordtype">void</span> handle_auth_res_read(
<a name="l00115"></a>00115             <span class="keyword">const</span> boost::system::error_code &amp;e);
<a name="l00116"></a>00116 
<a name="l00117"></a>00117         <span class="comment">//</span>
<a name="l00118"></a>00118 
<a name="l00119"></a>00119         <span class="keywordtype">void</span> read_action_();
<a name="l00120"></a>00120 
<a name="l00121"></a>00121         <span class="keywordtype">void</span> handle_action_pre_header_read(
<a name="l00122"></a>00122             <span class="keyword">const</span> boost::system::error_code &amp;e,
<a name="l00123"></a>00123             boost::shared_ptr&lt;boost::uint8_t&gt; ctrl_byte);
<a name="l00124"></a>00124 
<a name="l00125"></a>00125         <span class="keywordtype">void</span> handle_action_header_read(
<a name="l00126"></a>00126             <span class="keyword">const</span> boost::system::error_code &amp;e,
<a name="l00127"></a>00127             boost::shared_ptr&lt;boost::uint32_t&gt; uncomp_len,
<a name="l00128"></a>00128             boost::shared_ptr&lt;boost::uint32_t&gt; comp_len);
<a name="l00129"></a>00129 
<a name="l00130"></a>00130         <span class="keywordtype">void</span> handle_action_body_read(
<a name="l00131"></a>00131             <span class="keyword">const</span> boost::system::error_code &amp;e,
<a name="l00132"></a>00132             boost::shared_array&lt;boost::uint8_t&gt; buf,
<a name="l00133"></a>00133             boost::uint32_t uncomp_len,
<a name="l00134"></a>00134             boost::uint32_t comp_len);
<a name="l00135"></a>00135 
<a name="l00136"></a>00136         <span class="comment">//</span>
<a name="l00137"></a>00137 
<a name="l00138"></a>00138         <span class="keywordtype">void</span> write_action_(boost::shared_ptr&lt;Action&gt; action);
<a name="l00139"></a>00139 
<a name="l00140"></a>00140         <span class="keywordtype">void</span> handle_action_write(
<a name="l00141"></a>00141             <span class="keyword">const</span> boost::system::error_code &amp;e,
<a name="l00142"></a>00142             boost::shared_ptr&lt;ActionWriteData&gt; awd);
<a name="l00143"></a>00143 
<a name="l00144"></a>00144     <span class="keyword">private</span>:
<a name="l00145"></a>00145         boost::asio::ip::tcp::socket m_socket;
<a name="l00146"></a>00146         boost::asio::deadline_timer m_auth_timeout;
<a name="l00147"></a>00147 
<a name="l00148"></a>00148         <span class="keywordtype">bool</span> m_is_authenticated;
<a name="l00149"></a>00149         boost::posix_time::ptime m_up_since;
<a name="l00150"></a>00150         boost::uint64_t m_bytes_sent;
<a name="l00151"></a>00151         boost::uint64_t m_bytes_recv;
<a name="l00152"></a>00152         boost::uint8_t m_remote_is_big_endian;
<a name="l00153"></a>00153         boost::uint32_t m_remote_peekabot_version;
<a name="l00154"></a>00154 
<a name="l00155"></a>00155         boost::uint8_t m_auth_res_local;
<a name="l00156"></a>00156         boost::uint8_t m_auth_res_remote;
<a name="l00157"></a>00157     };
<a name="l00158"></a>00158 }
<a name="l00159"></a>00159 
<a name="l00160"></a>00160 
<a name="l00161"></a>00161 <span class="preprocessor">#endif // PEEKABOT_CONNECTION_HH_INCLUDED</span>
</pre></div></div>
</div>
<hr class="footer"/><address class="footer"><small>Generated on Sun Jan 30 2011 for peekabot by&nbsp;
<a href="http://www.doxygen.org/index.html">
<img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.7.1 </small></address>
</body>
</html>