Chinaunix首页 | 论坛 | 博客

bellcat2008bellcat2008.blog.chinaunix.net

首页　| 　博文目录　| 　关于我

bellcat2008

  • 博客访问： 518005
  • 博文数量： 137
  • 博客积分： 3170
  • 博客等级： 中校
  • 技术积分： 1455
  • 用 户 组： 普通用户
  • 注册时间： 2009-03-17 11:47
文章分类

全部博文（137）

文章存档

2015年（2）

2013年（1）

2012年（6）

2011年（5）

2010年（62）

2009年（61）

我的朋友
最近访客
推荐博文
相关博文
xxxxxxxxx

分类： LINUX

2015-11-06 17:43:12

点击(此处)折叠或打开

  1. #include <lua.h>
  2. #include <lauxlib.h>
  3. #include <lualib.h>
  4. #include <stdlib.h>
  5. #include <string.h>
  6. #include "skynet_malloc.h"
  7. #include <arpa/inet.h>
  8. #include <time.h>

  10. #include "cJSON.h"
  11. //#define __DEBUG__L_

  13. static int dbmsg(const char* fmt,...) {
  14. #ifdef __DEBUG__L_
  15.     va_list ap;
  16.     va_start(ap, fmt);
  17.     fprintf(stderr, "[LOG]");
  18.     vfprintf(stderr, fmt, ap);
  19.     va_end(ap);
  20. #endif    
  21.     return 0;
  22. }

  24. static int checkdata(cJSON *json){
  25.     cJSON *jtmp, *jtmp1;

  27.     jtmp = cJSON_GetObjectItem(json, "sender");
  28.     if(jtmp == NULL || jtmp->type != cJSON_Number){
  29.         return 1;
  30.     }

  32.     jtmp = cJSON_GetObjectItem(json, "event");
  33.     if(jtmp == NULL || jtmp->type != cJSON_Number){
  34.         return 2;
  35.     }

  37.     jtmp1= cJSON_GetObjectItem(json, "receiver");
  38.     if(jtmp1== NULL || jtmp1->type != cJSON_Object){
  39.         return 3;
  40.     }

  42.     jtmp = cJSON_GetObjectItem(jtmp1, "type");
  43.     if(jtmp == NULL || jtmp->type != cJSON_Number){
  44.         return 4;
  45.     }

  47.     jtmp = cJSON_GetObjectItem(jtmp1, "id");
  48.     if(jtmp == NULL || jtmp->type != cJSON_Number){
  49.         return 5;
  50.     }

  52.     jtmp = cJSON_GetObjectItem(json, "data");
  53.     if(jtmp == NULL){
  54.         return 6;
  55.     }

  57.     return 0;
  58. }

  60. static int encode(lua_State *L){
  61.     if (lua_isnoneornil(L,1)) {
  62.         return 0;
  63.     }

  65.     size_t sz;
  66.     const char * jsonbuf = luaL_checklstring(L,1, &sz);

  68.     cJSON *json = NULL;

  70.     json = cJSON_Parse(jsonbuf);
  71.     if(!json){
  72.         cJSON_Delete(json);
  73.         return luaL_error(L, "json 格式错误 %s", jsonbuf);
  74.     }
  75.     sz = checkdata(json);
  76.     if(sz != 0){
  77.         cJSON_Delete(json);
  78.         return luaL_error(L, "json 数据有问题 %d", sz);
  79.     }


  82.   int version = luaL_checkinteger(L,3);
  83.     dbmsg("%d==>", version);

  85.     cJSON *p;

  87.     char sendbuf[8192]; // 这个长度不够用 怎么办，使用 lua 数组？

  89.     int pos=0;
  90.     int ts;
  91.     int32_t big_int =htonl(version<<24);
  92.     if(version == 1){
  93.         memcpy(sendbuf+pos*sizeof(int32_t), &big_int, sizeof(int32_t));
  94.         pos++;
  95.     }

  97.     p = cJSON_GetObjectItem(json, "sender");
  98.     big_int = htonl(p->valueint);
  99.     memcpy(sendbuf+pos*sizeof(int32_t), &big_int, sizeof(int32_t));
  100.     pos++;
  101.     
  102.     p = cJSON_GetObjectItem(json, "receiver");
  103.     p = cJSON_GetObjectItem(p, "id");
  104.     big_int = htonl(p->valueint);
  105.     memcpy(sendbuf+pos*sizeof(int32_t), &big_int, sizeof(int32_t));
  106.     pos++;

  108.     p = cJSON_GetObjectItem(json, "receiver");
  109.     p = cJSON_GetObjectItem(p, "type");
  110.     big_int = htonl(p->valueint);
  111.     memcpy(sendbuf+pos*sizeof(int32_t), &big_int, sizeof(int32_t));
  112.     pos++;

  114.     p = cJSON_GetObjectItem(json, "event");
  115.     big_int = htonl(p->valueint);
  116.     memcpy(sendbuf+pos*sizeof(int32_t), &big_int, sizeof(int32_t));
  117.     pos++;

  119.     p = cJSON_GetObjectItem(json, "timestamp");
  120.     if(p == NULL){
  121.         ts = time(NULL);
  122.     }else{
  123.         ts = p->valueint;
  124.     }

  126.     if(version == 1){
  127.         big_int = htonl(ts);
  128.         memcpy(sendbuf+pos*sizeof(int32_t), &big_int, sizeof(int32_t));
  129.         pos++;
  130.     }

  132.     sz = pos*sizeof(int32_t);

  134.     p = cJSON_GetObjectItem(json, "data");
  135.     char *ptr = cJSON_PrintUnformatted(p);

  137.     int32_t dlen = (ptr==NULL?0:strlen(ptr));
  138.     
  139.     int i;
  140.     for(i=0; i<(dlen-4); i+=sizeof(int32_t)){
  141.         big_int = htonl(*(int32_t*)(ptr+i)) ^ 0xa5a5a5a5;
  142.         memcpy(sendbuf+sz, &big_int, sizeof(int32_t));
  143.         sz += sizeof(int32_t);
  144.     }

  146.     for(;i<dlen;i++){
  147.         char a = *(ptr+i)^0xa5;
  148.         memcpy(sendbuf+sz, &a, sizeof(char));
  149.         sz += 1;
  150.     }

  152.     free(ptr);
  153.     
  154.     lua_pushlstring(L,sendbuf, sz);
  155.     lua_pushnumber(L, ts);
  156.             
  157.     cJSON_Delete(json);

  159.     return 2;
  160. }

  162. static int decode(lua_State *L){

  164.     if (lua_isnoneornil(L,1)) {
  165.     return 0;
  166.   }
  167.   char * data = lua_touserdata(L,1);
  168.   int sz = luaL_checkinteger(L,2);

  170.     dbmsg("%s %d\n", data+20, sz);

  172.     int32_t ackid = ntohl(*(int32_t*)(data));

  174.     int version = ackid >> 24;
  175.     ackid = ackid & 0x00ffffff;

  177.     int32_t senderid = ntohl(*(int32_t*)(data+sizeof(int32_t)));
  178.     int32_t receiverid = ntohl(*(int32_t*)(data+2*sizeof(int32_t)));
  179.     int32_t receivertype = ntohl(*(int32_t*)(data+3*sizeof(int32_t)));
  180.     int32_t event = ntohl(*(int32_t*)(data+4*sizeof(int32_t)));
  181.     int32_t respts = ntohl(*(int32_t*)(data+5*sizeof(int32_t)));

  183.     int32_t timestamp =0;
  184.     if(version == 1){
  185.         timestamp = time(NULL);
  186. //        timestamp = ntohl(*(int32_t*)(data+5*sizeof(int32_t)));
  187.     }


  190. dbmsg("srte:%d %d %d %d\n", senderid, receiverid, receivertype, event);

  192.     char json[8192]={0};

  194.     strcpy(json, "{\"sender\":");

  196.   sprintf(json+strlen(json), "%d,", senderid);
  197.   strcat(json, "\"receiver\":{\"id\":");
  198.   sprintf(json+strlen(json), "%d,", receiverid);
  199.   strcat(json, "\"type\":");
  200.   sprintf(json+strlen(json), "%d},", receivertype);
  201.   strcat(json, "\"event\":");
  202.   sprintf(json+strlen(json), "%d,", event);

  204.     if(version == 1){
  205.         strcat(json, "\"timestamp\":");
  206.         sprintf(json+strlen(json), "%u,", timestamp);
  207.     }

  209.   strcat(json,"\"data\":");
  210.  // memcpy(json+strlen(json), data+5*sizeof(int32_t), sz-5*sizeof(int32_t));

  212.     int i = 0;
  213.     int len = sz;

  215.     dbmsg("==>%d\n", version);
  216.     if(version == 0){
  217.         i = 5*sizeof(int32_t);
  218.     }else if(version == 1){
  219.         i = 6*sizeof(int32_t);
  220.     }

  222.   for (; i<(len-4); i+=sizeof(int32_t) ){
  223.         dbmsg("%X\n", *(int32_t*)(data+i));
  224.     int32_t big_int = ntohl(*(int32_t*)(data+i)) ^ 0xa5a5a5a5;
  225.     dbmsg("%X %s\n", big_int, (char*)&big_int);
  226.         memcpy(json+strlen(json), &big_int, sizeof(int32_t));
  227.   }

  229.   for(;i<len && i<(sizeof(json)-10); i++){
  230.     char a = *(data+i) ^ 0xa5;
  231.     memcpy(json+strlen(json), &a, 1);
  232.   }
  233.   strcat(json,"}");

  235.     if (i>=(sizeof(json)-10)){
  236.         event = -1;    
  237.     }
  238.     
  239. //    lua_newtable(L);
  240.   if(receivertype == 1){
  241.     lua_pushlstring(L,"group", 5);
  242.   }else {
  243.     lua_pushlstring(L,"send", 4);
  244.   }
  245.     if(event == 92){
  246.         lua_pushlstring(L, "response", 8);
  247.         memset(json, 0, sizeof(json));
  248.         sprintf(json, "%u", respts);
  249.     }

  251. /*
  252.     if (event == 90){
  253.         lua_pushlstring(L,"login", 5);

  255.     }else if(event == 91){
  256.         lua_pushlstring(L,"heartbeat", 9);

  258.     }else if(event == 92){
  259.         lua_pushlstring(L, "response", 8);
  260.         memset(json, 0, sizeof(json));
  261.         sprintf(json, "%u", respts);
  262.     }else if(receivertype == 1){
  263.         lua_pushlstring(L,"group", 5);    
  264.     }else if(event >= 1000 && event <= 4000){
  265.         lua_pushlstring(L,"send", 4);    
  266.     }else{
  267.         lua_pushlstring(L, "error", 5);
  268.     }
  269.  */

  271.     lua_pushnumber(L, ackid);

  273.     dbmsg("1:%d\n", senderid);
  274.     lua_pushnumber(L, senderid);
  275.     dbmsg("2:%d\n", receiverid);
  276.     lua_pushnumber(L, receiverid);
  277.     
  278.     dbmsg("3:%s\n", json);
  279.   lua_pushlstring(L,json, strlen(json));

  281.     dbmsg("4:%d\n", version);

  283.   lua_pushnumber(L, version);

  285.   lua_pushnumber(L, event);
  286. //    lua_rawseti(L, -2, 1);
  287. //
  288. // 这里不必 skynet_free ， 因为数据是 skynet_redirect 过来的

  290.   return 7;
  291. }


  294. static const struct luaL_Reg chatencode[] = {

  296.   {"encode" , encode},
  297.   {"decode" , decode},

  299.   {NULL, NULL}

  301. };

  303. int luaopen_chatencode(lua_State *L){

  305.   luaL_newlib(L, chatencode);

  307.     return 1;
  308. }


阅读(3685) | 评论(0) | 转发(0) |
0

上一篇：ffmpeg抓取视频，进一步，推送为rtmp流过程分析

下一篇：没有了

给主人留下些什么吧！~~
关于我们 | 关于IT168 | 联系方式 | 广告合作 | 法律声明 | 免费注册

Copyright 2001-2010 ChinaUnix.net All Rights Reserved 北京皓辰网域网络信息技术有限公司. 版权所有

感谢所有关心和支持过ChinaUnix的朋友们

16024965号-6