陳碩 (giantchen_AT_gmail)

Blog.csdn.net/Solstice

這是《Muduo 網(wǎng)絡(luò)編程示例》系列的第四篇文章。

Muduo 全系列文章列表： http://blog.csdn.net/Solstice/category/779646.aspx

Python Twisted 是一款非常好的網(wǎng)絡(luò)庫，它也采用 Reactor 作為網(wǎng)絡(luò)編程的基本模型，所以從使用上與 muduo 頗有相似之處。（當(dāng)然，muduo 沒有 deferreds）Finger 是 twisted 文檔的一個經(jīng)典例子，本文展示如何用 muduo 來實現(xiàn)最簡單的 finger 服務(wù)端。限于篇幅，只實現(xiàn) finger01~07。代碼位于 examples/twisted/finger 。

1 拒絕連接

什么都不做，程序空等。finger01.cc

1: #include <muduo/net/EventLoop.h> 2:? 3: using namespace muduo; 4: using namespace muduo::net; 5:? 6: int main() 7: { 8: EventLoop loop; 9: loop.loop(); 10: }

2 接受新連接

在 1079 端口偵聽新連接，接受連接之后什么都不做，程序空等。muduo 會自動丟棄收到的數(shù)據(jù)。finger02.cc

1: #include <muduo/net/EventLoop.h> 2: #include <muduo/net/TcpServer.h> 3:? 4: using namespace muduo; 5: using namespace muduo::net; 6:? 7: int main() 8: { 9: EventLoop loop; 10: TcpServer server(&loop, InetAddress(1079), "Finger"); 11: server.start(); 12: loop.loop(); 13: }

3 主動斷開連接

接受新連接之后主動斷開。finger03.cc

以下省略頭文件和 namespace。

1: void onConnection(const TcpConnectionPtr& conn) 2: { 3: if (conn->connected()) 4: { 5: conn->shutdown(); 6: } 7: } 8:? 9: int main() 10: { 11: EventLoop loop; 12: TcpServer server(&loop, InetAddress(1079), "Finger"); 13: server.setConnectionCallback(onConnection); 14: server.start(); 15: loop.loop(); 16: }

4 讀取用戶名，然后斷開連接

如果讀到一行以 \r\n 結(jié)尾的消息，就斷開連接。finger04.cc

注意這段代碼有安全問題，如果惡意客戶端不斷發(fā)送數(shù)據(jù)而不換行，會撐爆服務(wù)端的內(nèi)存。另外，Buffer::findCRLF() 是線性查找，如果客戶端每次發(fā)一個字節(jié)，服務(wù)端的時間復(fù)雜度為 O(N^2)，會消耗 CPU 資源。

1: void onMessage(const TcpConnectionPtr& conn, 2: Buffer* buf, 3: Timestamp receiveTime) 4: { 5: if (buf->findCRLF()) 6: { 7: conn->shutdown(); 8: } 9: } 10:? 11: int main() 12: { 13: EventLoop loop; 14: TcpServer server(&loop, InetAddress(1079), "Finger"); 15: server.setMessageCallback(onMessage); 16: server.start(); 17: loop.loop(); 18: }

5. 讀取用戶名、輸出錯誤信息、然后斷開連接

如果讀到一行以 \r\n 結(jié)尾的消息，就發(fā)送一條出錯信息，然后斷開連接。finger05.cc

安全問題同上。

1: void onMessage(const TcpConnectionPtr& conn, 2: Buffer* buf, 3: Timestamp receiveTime) 4: { 5: if (buf->findCRLF()) 6: { 7: conn->send("No such user\r\n"); 8: conn->shutdown(); 9: } 10: } 11:? 12: int main() 13: { 14: EventLoop loop; 15: TcpServer server(&loop, InetAddress(1079), "Finger"); 16: server.setMessageCallback(onMessage); 17: server.start(); 18: loop.loop(); 19: }

6. 從空的 UserMap 里查找用戶

從一行消息中拿到用戶名（第 22 行），在 UserMap 里查找，然后返回結(jié)果。finger06.cc

安全問題同上。

1: typedef std::map<string, string> UserMap; 2: UserMap users; 3:? 4: string getUser(const string& user) 5: { 6: string result = "No such user"; 7: UserMap::iterator it = users.find(user); 8: if (it != users.end()) 9: { 10: result = it->second; 11: } 12: return result; 13: } 14:? 15: void onMessage(const TcpConnectionPtr& conn, 16: Buffer* buf, 17: Timestamp receiveTime) 18: { 19: const char* crlf = buf->findCRLF(); 20: if (crlf) 21: { 22: string user(buf->peek(), crlf); 23: conn->send(getUser(user) + "\r\n"); 24: buf->retrieveUntil(crlf + 2); 25: conn->shutdown(); 26: } 27: } 28:? 29: int main() 30: { 31: EventLoop loop; 32: TcpServer server(&loop, InetAddress(1079), "Finger"); 33: server.setMessageCallback(onMessage); 34: server.start(); 35: loop.loop(); 36: }

7. 往 UserMap 里添加一個用戶

與前面幾乎完全一樣，只多了第 31 行。finger07.cc

1: typedef std::map<string, string> UserMap; 2: UserMap users; 3:? 4: string getUser(const string& user) 5: { 6: string result = "No such user"; 7: UserMap::iterator it = users.find(user); 8: if (it != users.end()) 9: { 10: result = it->second; 11: } 12: return result; 13: } 14:? 15: void onMessage(const TcpConnectionPtr& conn, 16: Buffer* buf, 17: Timestamp receiveTime) 18: { 19: const char* crlf = buf->findCRLF(); 20: if (crlf) 21: { 22: string user(buf->peek(), crlf); 23: conn->send(getUser(user) + "\r\n"); 24: buf->retrieveUntil(crlf + 2); 25: conn->shutdown(); 26: } 27: } 28:? 29: int main() 30: { 31: users["schen"] = "Happy and well"; 32: EventLoop loop; 33: TcpServer server(&loop, InetAddress(1079), "Finger"); 34: server.setMessageCallback(onMessage); 35: server.start(); 36: loop.loop(); 37: }

以上就是全部內(nèi)容，可以用 telnet 扮演客戶端來測試我們的簡單 finger 服務(wù)端。

Telnet 測試

在一個命令行窗口運(yùn)行

$ ./bin/twisted_finger07

另一個命令行運(yùn)行

$ telnet localhost 1079
Trying ::1...
Trying 127.0.0.1...
Connected to localhost.
Escape character is '^]'.
muduo
No such user
Connection closed by foreign host.

再試一次

$ telnet localhost 1079
Trying ::1...
Trying 127.0.0.1...
Connected to localhost.
Escape character is '^]'.
schen
Happy and well
Connection closed by foreign host.

冒煙測試過關(guān)。

(待續(xù))

轉(zhuǎn)載于:https://www.cnblogs.com/Solstice/archive/2011/02/23/1963197.html

總結(jié)

以上是生活随笔為你收集整理的Muduo 网络编程示例之四：Twisted Finger的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。