免费注册 查看新帖 |

Chinaunix

  平台 论坛 博客 文库
› 论坛 › 程序设计 › C/C++ › HTML5应用越来越多了，应朋友要求写了一个小应用框架
最近访问板块 发新帖
查看: 4329 | 回复: 7
打印 上一主题 下一主题

HTML5应用越来越多了，应朋友要求写了一个小应用框架 [复制链接]

zliming

论坛徽章:
0
跳转到指定楼层
1楼 [收藏(0)] [报告]
发表于 2010-08-03 11:11 |只看该作者 |倒序浏览
  1. /*
  2. * SendBuf.h
  3. *
  4. *  Created on: 2010-8-2
  5. *      Author: zliming
  6. */

  8. #ifndef SENDBUF_H_
  9. #define SENDBUF_H_

  11. #include <vector>

  13. using namespace std;

  15. /*
  16. * tr1/shared_ptr.h在mingw下怪怪，又不想用boost，头太大
  17. */

  19. class SharedBuf {
  20. public:
  21.     SharedBuf(const char* data);
  22.     SharedBuf(const SharedBuf& o);
  23.     ~SharedBuf();
  24.     void        operator=(const SharedBuf& o);
  25.     size_t      Size();
  26.     const char* GetBuffer();

  28. private:
  29.     void ReleaseRef();
  30. private:
  31.     struct Data {
  32.         char*   data;
  33.         int     refCount;
  34.         size_t  len;
  35.     } *ptr;
  36. };

  38. #endif /* SENDBUF_H_ */
复制代码
  1. /*
  2. * SendBuf.cpp
  3. *
  4. *  Created on: 2010-8-2
  5. *      Author: zliming
  6. */

  8. #include "SendBuf.h"
  9. #include <string.h>

  11. SharedBuf::SharedBuf(const char* buf) {
  12.     size_t len = strlen(buf);
  13.     ptr = new Data;
  14.     ptr->len = len + 2;
  15.     ptr->data = new char[ptr->len];
  16.     ptr->data[0] = 0;
  17.     memcpy(ptr->data + 1, buf, len);
  18.     ptr->data[len + 1] = 0xff;
  19.     ptr->refCount = 1;
  20. }
  21. SharedBuf::SharedBuf(const SharedBuf& o): ptr(0){
  22.     operator=(o);
  23. }
  24. SharedBuf::~SharedBuf(){
  25.     ReleaseRef();
  26. }
  27. size_t SharedBuf::Size(){
  28.     if(ptr){
  29.         return ptr->len;
  30.     }
  31.     return 0;
  32. }
  33. const char* SharedBuf::GetBuffer(){
  34.     if(ptr){
  35.         return ptr->data;
  36.     }
  37.     return NULL;
  38. }
  39. void SharedBuf::operator =(const SharedBuf& o){
  40.     //去掉原引用的内存
  41.     ReleaseRef();
  42.     ptr = o.ptr;
  43.     ptr->refCount ++;
  44. }
  45. void SharedBuf::ReleaseRef(){
  46.     if(ptr && --ptr->refCount <= 0){
  47.         delete[] ptr->data;
  48.         delete ptr;
  49.         ptr = 0;
  50.     }
  51. }

复制代码
  1. #ifndef __CLIENT_H__
  2. #define __CLIENT_H__

  4. #include <libZLM/LibZLM.h>

  6. struct SafeDelete
  7. {
  8.     template <typename T>
  9.     void operator ()(T& t)
  10.     {
  11.         if(t)
  12.         {
  13.             delete t;
  14.             t = NULL;
  15.         }
  16.     }
  17. };
  18. #define SAFE_DELETE SafeDelete()

  20. class Client
  21. {
  22. public:
  23.     virtual void Start() { }
  24.     virtual ~Client() { }
  25.     Socket& GetSocket(){ return _socket; }

  27. private:
  28.     Socket  _socket;
  29. };

  31. template<typename C>
  32. class Server
  33. {
  34. public:
  35.     Server(int port) : _listen("0.0.0.0", port) { }
  36.     virtual void Accept(const Error& ec, C* client)
  37.     {
  38.         if(client)
  39.         {
  40.             if(!ec)
  41.                 client->Start();
  42.             else
  43.                 delete client;
  44.         }
  45.         client = new C;
  46.         _listen.AsyncAccept(&client->GetSocket(), Bind(&Server::Accept, this, _1, client));
  47.     }
  48.     virtual ~Server(){}
  49. protected:
  50.     Listen  _listen;
  51. };

  53. #endif//__CLIENT_H__
复制代码
  1. /*
  2. * websocket.h
  3. *
  4. *  Created on: 2010-7-29
  5. *      Author: zliming
  6. */

  8. #ifndef WEBSOCKET_H_
  9. #define WEBSOCKET_H_

  11. #include "Socket.h"

  13. class WebSocket : public Client {
  14. public:
  15.     void            Start();
  16.     virtual void    DoCommand(Record& record);

  18. private:
  19.     void            Do(const Error& err, int len);
  20.     void            OnRead(const Error& err, int len);

  22. private:
  23.     char            buf[0x8000];
  24.     size_t          read_bytes;
  25.     int             step;
  26.     char*           buf_offset;
  27. };

  29. #endif /* WEBSOCKET_H_ */
复制代码
  1. //============================================================================
  2. // Name        : websocket.cpp
  3. // Author      : zliming
  4. //============================================================================

  6. #include "websocket.h"
  7. #include "Handshaker.h"
  8. #include <stdlib.h>
  9. #include <string.h>
  10. #include <memory.h>
  11. #include <stdio.h>
  12. #include <string>
  13. using namespace std;

  15. static char flash_policy_req[] = "<policy-file-request/>";
  16. static char flash_policy_ret[] = "<cross-domain-policy>\n"
  17.     "<allow-access-from domain='*' to-ports='*'/>\n"
  18.     "</cross-domain-policy>";
  19. static char http_end_flag[] = "\r\n\r\n";

  21. void WebSocket::Start() {
  22.     step = 1;
  23.     read_bytes = 0;
  24.     Do(Error(), 0);
  25. }

  27. void WebSocket::Do(const Error& err, int len) {
  28.     switch (step) {
  29.     case 1:
  30.         step = 2;
  31.         return GetSocket().AsyncRead(buf, sizeof(flash_policy_req), Bind(
  32.                 &WebSocket::Do, this, _1, _2));
  33.     case 2:
  34.         //如果是通过Flash桥来连接的，会先读求连接策略
  35.         if (memcmp(flash_policy_req, buf, sizeof(flash_policy_req)) == 0) {
  36.             step = 3;
  37.             return GetSocket().AsyncWrite(flash_policy_ret,
  38.                     sizeof(flash_policy_ret),
  39.                     Bind(&WebSocket::Do, this, _1, _2));
  40.             case 3:
  41.             return delete this;
  42.         } else {
  43.             case 4:
  44.             read_bytes += len;
  45.             buf[read_bytes] = 0;
  46.             buf_offset = strstr(buf, http_end_flag);
  47.             if (buf_offset) {
  48.                 //HTTP头读完，还要读8个字节(key3)
  49.                 if (buf + read_bytes - (buf_offset + 4) < 8) {
  50.                     step = 5;
  51.                     return GetSocket().AsyncRead(buf + read_bytes, 8 - (buf
  52.                             + read_bytes - (buf_offset + 4)), Bind(
  53.                             &WebSocket::Do, this, _1, _2));
  54.                     case 5:
  55.                     read_bytes += len;
  56.                     buf[read_bytes] = 0;
  57.                 }
  58.                 //全体需要的数据完成
  59.                 Handshaker handshaker(buf, read_bytes);
  60.                 const string& result = handshaker.GetResult();
  61.                 if(result.empty()){
  62.                     return delete this;
  63.                 }
  64.                 memcpy(buf, result.c_str(), result.length());
  65.                 memcpy(buf + result.length(), handshaker.GetChallenge(), 16);

  67.                 read_bytes = 0;
  68.                 return GetSocket().AsyncWrite(buf, result.length() + 16,
  69.                         Bind(&WebSocket::OnRead, this, _1, 0));
  70.             } else {
  71.                 step = 4;
  72.                 return GetSocket().AsyncReadSome(buf + read_bytes, sizeof(buf)
  73.                         - read_bytes - 1, Bind(&WebSocket::Do, this, _1, _2));
  74.             }
  75.         }
  76.     }
  77. }
  78. void WebSocket::OnRead(const Error& err, int len){
  79.     if(!err){
  80.         read_bytes += len;
  81.         char* find, *prog = buf;
  82.         while((find = (char*)memchr(prog, 0xff, buf + read_bytes - prog)) != NULL){
  83.             //websocket发送的数据为: 00 data ff
  84.             *find = 0;
  85.             //这个是Json数据的操作类，和javascript通讯，json是不二选择
  86.             Record record;
  87.             if(record.FromString(prog + 1, find - prog - 1)){
  88.                 this->DoCommand(record);
  89.             }
  90.             //处理完成，跳过已经处理部分数据
  91.             prog = find + 1;
  92.         }
  93.         if(prog != buf){
  94.             //已经处理完的数据丢弃
  95.             memmove(buf, prog, buf + read_bytes - prog);
  96.             read_bytes = buf + read_bytes - prog;
  97.         }
  98.         if(sizeof(buf) - read_bytes - 1 > 0){
  99.             //读新的数据进来
  100.             return GetSocket().AsyncReadSome(buf + read_bytes,
  101.                     sizeof(buf) - read_bytes - 1,
  102.                     Bind(&WebSocket::OnRead, this, _1, _2));
  103.         }
  104.     }
  105.     delete this;
  106. }
  107. void WebSocket::DoCommand(Record& record){

  109. }

复制代码
zliming

论坛徽章:
0
2楼 [报告]
发表于 2010-08-03 11:12 |只看该作者
  1. /*
  2. * Handshaker.h
  3. *
  4. *  Created on: 2010-7-29
  5. *      Author: zliming
  6. */

  8. #ifndef HANDSHAKER_H_
  9. #define HANDSHAKER_H_

  11. #include <string>
  12. using namespace std;

  14. class Handshaker{
  15. public:
  16.     const string&   GetResult();
  17.     const char*     GetChallenge();
  18.     Handshaker(char* req, size_t req_len);

  20. private:
  21.     string          GetValue(const char* key);
  22.     unsigned        GetNumber(const char* value);
  23. private:
  24.     string          result;
  25.     char*           req;
  26.     unsigned char   challenge[16];
  27. };

  29. #endif /* HANDSHAKER_H_ */
复制代码
  1. /*
  2. * Handshaker.cpp
  3. *
  4. *  Created on: 2010-7-29
  5. *      Author: zliming
  6. */

  8. #include "Handshaker.h"
  9. #include "md5.h"
  10. #include <string.h>
  11. #include <stdio.h>
  12. #include <algorithm>

  14. unsigned BigEndian(unsigned i){
  15.     return (i << 24) | (i >> 24) |
  16.             ((i & 0xff00) << 8) | ((i & 0xff0000) >> 8);
  17. }

  19. Handshaker::Handshaker(char* req, size_t req_len){
  20.     this->req = req;
  21.     //必须是GET请求
  22.     if(memcmp(req, "GET", strlen("GET")) != 0  ||
  23.             GetValue("Upgrade:") != "websocket" ||
  24.             GetValue("Connection:")!= "upgrade"){
  25.         return;
  26.     }
  27.     string origin = GetValue("Origin:");
  28.     string key1 = GetValue("Sec-WebSocket-Key1:");
  29.     string key2 = GetValue("Sec-WebSocket-Key2:");
  30.     if(origin.empty() || key1.empty() || key2.empty()){
  31.         return;
  32.     }
  33.     char* key3 = req + req_len - 8;
  34.     string uri(req + 4, strchr(req + 4, ' ') - (req + 4));
  35.     string location = "ws://" + GetValue("Host:") + uri;
  36.     unsigned n1 = GetNumber(key1.c_str());
  37.     unsigned n2 = GetNumber(key2.c_str());
  38.     unsigned sc1 = count(key1.begin(), key1.end(), ' ');
  39.     unsigned sc2 = count(key2.begin(), key2.end(), ' ');
  40.     if(sc1 == 0 || sc2 == 0 || n1 % sc1 != 0 || n2 % sc2 != 0){
  41.         return;
  42.     }
  43.     unsigned part1 = BigEndian(n1 / sc1);
  44.     unsigned part2 = BigEndian(n2 / sc2);
  45.     memcpy(challenge, &part1, sizeof(unsigned));
  46.     memcpy(challenge + sizeof(unsigned), &part2, sizeof(unsigned));
  47.     memcpy(challenge + sizeof(unsigned) * 2, key3, 8);
  48.     MD5_CTX ctx;
  49.     MD5_Init(&ctx);
  50.     MD5_Update(&ctx, challenge, sizeof(challenge));
  51.     MD5_Final(challenge, &ctx);

  53.     //返回
  54.     result ="HTTP/1.1 101 WebSocket Protocol Handshake\r\n"
  55.             "Upgrade: WebSocket\r\n"
  56.             "Connection: Upgrade\r\n"
  57.             "Sec-WebSocket-Location: " + location + "\r\n"
  58.             "Sec-WebSocket-Origin: " + origin + "\r\n";

  60.     string proto = GetValue("Sec-WebSocket-Protocol:");
  61.     if(!proto.empty()){
  62.         result += "Sec-WebSocket-Protocol: " + proto + "\r\n";
  63.     }
  64.     result += "\r\n";
  65. }

  67. const char* Handshaker::GetChallenge(){
  68.     return (const char*)challenge;
  69. }

  71. const string& Handshaker::GetResult(){
  72.     return result;
  73. }

  75. unsigned Handshaker::GetNumber(const char* value){
  76.     unsigned ret = 0;
  77.     while(*value){
  78.         if(*value >= '0' && *value <= '9'){
  79.             ret = ret * 10 + *value - '0';
  80.         }
  81.         value++;
  82.     }
  83.     return ret;
  84. }

  86. string Handshaker::GetValue(const char* key){
  87.     string ret;
  88.     char* find = strstr(req, key);
  89.     if(find){
  90.         size_t len = strlen(key);
  91.         find += len + 1;
  92.         ret.append(find, strchr(find, '\r') - find);
  93.         transform(ret.begin(), ret.end(), ret.begin(), ::tolower);
  94.     }
  95.     return ret;
  96. }
复制代码
  1. /*
  2. * Chat.h
  3. *
  4. *  Created on: 2010-8-1
  5. *      Author: zliming
  6. */

  8. #ifndef CHAT_H_
  9. #define CHAT_H_

  11. #include "websocket.h"
  12. #include "SendBuf.h"
  13. #include <string>

  15. class Chat : public WebSocket {
  16. public:
  17.     void DoCommand(Record& record);
  18.     void SendChat(SharedBuf chat);
  19.     ~Chat();

  21. private:
  22.     void FinishSend(SharedBuf);
  23. private:
  24.     string username;
  25. };

  27. #endif /* CHAT_H_ */
复制代码
  1. /*
  2. * Chat.cpp
  3. *
  4. *  Created on: 2010-8-1
  5. *      Author: zliming
  6. */

  8. #include "Chat.h"
  9. #include <string.h>
  10. #include <map>

  12. //只有一个IORUN，不需要互斥
  13. map<string, Chat*> users;

  15. void Chat::DoCommand(Record& record) {
  16.     if (memcmp("login", (const char*)record["cmd"], sizeof("login")) == 0) {
  17.         Record ret;
  18.         ret["cmd"] = "login";
  19.         if (users.find((const char*) record["value"]) != users.end()) {
  20.             ret["value"] = "user online";
  21.         } else {
  22.             ret["value"] = "ok";
  23.             username = (const char*) record["value"];
  24.             users[username] = this;
  25.         }
  26.         SharedBuf buf(ret.ToString());
  27.         SendChat(buf);
  28.     } else if (!username.empty() &&
  29.             memcmp("chat", (const char*)record["cmd"], sizeof("chat")) == 0) {
  30.         //用户在线，发聊天包时加上用户名
  31.         record["user"] = username.c_str();
  32.         SharedBuf buf(record.ToString());
  33.         for (map<string, Chat*>::iterator it = users.begin(); it != users.end(); ++it) {
  34.             it->second->SendChat(buf);
  35.         }
  36.     }
  37. }
  38. void Chat::SendChat(SharedBuf chat) {
  39.     //在发送完成处理中引用这个SharedBuf，不让它过早释放
  40.     GetSocket().AsyncWrite((char*)chat.GetBuffer(), chat.Size(),
  41.             Bind(&Chat::FinishSend, this, chat));
  42. }
  43. void Chat::FinishSend(SharedBuf){

  45. }
  46. Chat::~Chat() {
  47.     if(!username.empty()){
  48.         users.erase(username);
  49.         //通知用户下线
  50.         Record ret;
  51.         ret["cmd"] = "logout";
  52.         ret["value"] = username.c_str();
  53.         SharedBuf buf(ret.ToString());
  54.         for (map<string, Chat*>::iterator it = users.begin(); it != users.end(); ++it) {
  55.             it->second->SendChat(buf);
  56.         }
  57.     }
  58. }
复制代码
实战分享：从技术角度谈机器学习入门| 【大话IT】RadonDB低门槛向MySQL集群下战书 | ChinaUnix打赏功能已上线！ | 新一代分布式关系型数据库RadonDB知多少？
zliming

论坛徽章:
0
3楼 [报告]
发表于 2010-08-03 11:19 |只看该作者
demo地址
http://zliming.3322.org/
实战分享：从技术角度谈机器学习入门| 【大话IT】RadonDB低门槛向MySQL集群下战书 | ChinaUnix打赏功能已上线！ | 新一代分布式关系型数据库RadonDB知多少？
zhanglistar

论坛徽章:
0
4楼 [报告]
发表于 2010-08-03 11:46 |只看该作者
回复 1# zliming


    什么应用的?能否给个说明
实战分享：从技术角度谈机器学习入门| 【大话IT】RadonDB低门槛向MySQL集群下战书 | ChinaUnix打赏功能已上线！ | 新一代分布式关系型数据库RadonDB知多少？
xiapan83

论坛徽章:
0
5楼 [报告]
发表于 2010-08-03 12:00 |只看该作者
能否给个说明
aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa
实战分享：从技术角度谈机器学习入门| 【大话IT】RadonDB低门槛向MySQL集群下战书 | ChinaUnix打赏功能已上线！ | 新一代分布式关系型数据库RadonDB知多少？
zliming

论坛徽章:
0
6楼 [报告]
发表于 2010-08-03 13:11 |只看该作者
html5的websocket，使js可以用tcp长连接通信的一个东西

这里用这个做一个聊天演示
实战分享：从技术角度谈机器学习入门| 【大话IT】RadonDB低门槛向MySQL集群下战书 | ChinaUnix打赏功能已上线！ | 新一代分布式关系型数据库RadonDB知多少？
ecjtubaowp

论坛徽章:
1
CU十二周年纪念徽章 日期:2013-10-24 15:41:34
7楼 [报告]
发表于 2010-08-03 13:15 |只看该作者
那还得要用支持html5的浏览器吗？
实战分享：从技术角度谈机器学习入门| 【大话IT】RadonDB低门槛向MySQL集群下战书 | ChinaUnix打赏功能已上线！ | 新一代分布式关系型数据库RadonDB知多少？
zliming

论坛徽章:
0
8楼 [报告]
发表于 2010-08-03 13:22 |只看该作者
嗯，现在html5浏览器才刚起步，不过这个东西在不支持html5的浏览器中调用flash模拟websocket连接
实战分享：从技术角度谈机器学习入门| 【大话IT】RadonDB低门槛向MySQL集群下战书 | ChinaUnix打赏功能已上线！ | 新一代分布式关系型数据库RadonDB知多少？
返回列表 发新帖
高级模式
B Color Image Link Quote Code Smilies
您需要登录后才可以回帖 登录 | 注册

本版积分规则 发表回复

  

北京盛拓优讯信息技术有限公司. 版权所有 京ICP备16024965号-6 北京市公安局海淀分局网监中心备案编号：11010802020122 niuxiaotong@pcpop.com 17352615567
未成年举报专区
中国互联网协会会员  联系我们：huangweiwei@itpub.net
感谢所有关心和支持过ChinaUnix的朋友们 转载本站内容请注明原作者名及出处

清除 Cookies - ChinaUnix - Archiver - WAP - TOP