开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > P2P编程 > 查看源码
Layer1.cpp
资源名称：p2p_vod.rar　[点击查看]
上传用户：liguizhu
上传日期：2015-11-01
资源大小：2422k
文件大小：10k
源码类别：

P2P编程

开发平台：

Visual C++

Layer1.cpp：源码内容
  1. /*
  2.  *  Openmysee
  3.  *
  4.  *  This program is free software; you can redistribute it and/or modify
  5.  *  it under the terms of the GNU General Public License as published by
  6.  *  the Free Software Foundation; either version 2 of the License, or
  7.  *  (at your option) any later version.
  8.  *
  9.  *  This program is distributed in the hope that it will be useful,
  10.  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.  *  GNU General Public License for more details.
  13.  *
  14.  *  You should have received a copy of the GNU General Public License
  15.  *  along with this program; if not, write to the Free Software
  16.  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17.  *
  18.  */
  19. #include "stdafx.h"
  20. #include "Communicator.h"
  21. #include "Layer1.h"
  22. #include "Http_Fetcher.h"
  24. // 在所有服务器都已经支持切换编码之前，需要从GTV文件中读取媒体类型数据
  25. //#define READ_GTV_MEDIA
  27. namespace NPLayer1 {
  28. Layer1::Layer1() {
  29. comm = NULL;
  30. }
  32. Layer1::~Layer1() {
  33. delete comm;
  34. }
  36. P2P_RETURN_TYPE Layer1::Request(
  37. LPCSTR url,  // in, 要播放的gtv地址或者路径
  38. const HWND notifyWnd,  // in, layer1发送消息的窗口
  39. const UINT notifyCode, // in, layer1发送消息的msg ID
  40. const UINT16 channelID //in, layer1发送消息的目标channel ID, , 将附加到WPARAM里面的HIWORD，
  41. )
  42. {
  43. if(comm)
  44. return PRT_DUP_INIT;
  45. if(!url || strlen(url) >= MAX_PATH_EX)
  46. return PRT_BAD_URL;
  48. string resname;
  49. string hashcode;
  50. string spList;
  51. string trackerList;
  52. UINT blockSize = BLOCK_SIZE;
  53. float bitRate = 40.0f;
  55. char szFinal[MAX_PATH_EX];
  56.     strcpy(szFinal, url);
  58. #ifdef READ_GTV_MEDIA
  59.     UINT attachDataSize = 0;
  60.     char* attachData = NULL;
  61. #endif
  63. // gaov协议格式：gaov://tsip:tsport/spip:spport/res md5/res name(optional)/bitrate(optional)/
  64. if(strnicmp(szFinal, "gaov://", 7) == 0) {
  65. string temp = szFinal;
  66. temp.erase(0, 7);
  67. size_t index = temp.find('/');
  68. trackerList = temp.substr(0, index);
  69. if(index == -1)
  70. return PRT_BAD_URL;
  71. temp.erase(0, index+1);
  72. index = temp.find('/');
  73. spList = temp.substr(0, index);
  74. if(index == -1)
  75. return PRT_BAD_URL;
  76. temp.erase(0, index+1);
  77. index = temp.find('/');
  78. hashcode = temp.substr(0, index);
  79. if(index != -1) {
  80. temp.erase(0, index+1);
  81. index= temp.find('/');
  82. resname = temp.substr(0, index);
  83. if(index != -1) {
  84. temp.erase(0, index +1);
  85. index = temp.find('/');
  86. bitRate = static_cast<float>(atof(temp.substr(0, index).data()));
  87. if(index != -1) {
  88. temp.erase(0, index+1);
  89. }
  90. }
  91. }
  93. // 如果协议中没有包含资源名，则显示hashcoder，因为后面会检查resname是否为空 -_-
  94. if(resname.empty())
  95. resname = hashcode;
  96. }
  97. else {
  98. // 1. 判断是否GTV文件
  99. char* index = strrchr(szFinal, '.');
  100. if(index != NULL) {
  101. index++;
  102. if(strnicmp(index, "gtv", 3) != 0)
  103. return PRT_BAD_URL;
  104. }
  105. else
  106. return PRT_BAD_URL;
  108. // 2. 下载GTV文件
  109. bool isHttpGTV = false;
  110. // 如果是"HTTP://"打头的gtv，就说明需要下载
  111. if(strnicmp(szFinal, "http://", 7) == 0) {
  112. // 通过HTTP下载的文件内容保存在这里
  113. char buf[MAX_PATH_EX];
  114. if(GetTempPath(MAX_PATH_EX, buf) == 0)
  115. return PRT_SYS;
  116. itoa(GetTickCount(), buf+strlen(buf), 10);
  117. strcat(buf, ".gtv");
  119. HANDLE localFile = CreateFile(buf, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
  120. if(localFile == INVALID_HANDLE_VALUE)
  121. return PRT_SYS;
  122. // HTTP下载类
  123. HttpFetcher fetcher;
  124. UINT fileSize = 0;
  125. UINT bytesRead = fetcher.http_fetch(szFinal, localFile, fileSize, bytesRead, 0.0f/*no limit*/);
  126. CloseHandle(localFile);
  127. if(bytesRead == UINT_MAX || bytesRead == 0)
  128. return PRT_BAD_URL;
  130. // 将szFinal设置为下载完毕的GTV文件的路径
  131. strcpy(szFinal, buf);
  132. isHttpGTV = true;
  133. }
  135. // 3. 读取GTV文件的数据
  136. char temp[4096];
  137. HANDLE hFile = CreateFile(szFinal, GENERIC_READ, 0, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
  138. if(hFile == INVALID_HANDLE_VALUE)
  139. return PRT_SYS;
  140. DWORD TTT = 0;
  141. if(!ReadFile(hFile, temp, 4096, &TTT, NULL) || TTT == 0) {
  142. CloseHandle(hFile);
  143. return PRT_SYS;
  144. }
  145. CloseHandle(hFile);
  147. if(isHttpGTV)
  148. remove(szFinal); // 删除临时gtv文件
  150. // 4. 解析数据
  151. istrstream* is = NULL;
  152. is = new istrstream(temp);
  153. if(is->fail())
  154. return PRT_SYS;
  156. string line;
  157. string name;
  158. string value;
  159. string inSection;
  160. size_t posEqual;
  161. while (getline(*is,line)) {
  162. if ( !line.length()) continue;
  163. if ( line[0] == '#') continue; // 注释行
  164. if ( line[0] == ';') continue; // 注释行
  166. if ( line[0] == '[') {
  167. inSection=line.substr(1,line.find(']')-1);
  168. continue;
  169. }
  171. posEqual=line.find('=');
  172. name = line.substr(0,posEqual);
  173. value = line.substr(posEqual+1);
  174. size_t index = value.find('r');
  175. if(index != -1)
  176. value = value.substr(0, index);
  177. index = value.find('n');
  178. if(index != -1)
  179. value = value.substr(0, index);
  181. if(name == "ChannelName")
  182. resname = value;
  183. else if(name == "ResourceHash")
  184. hashcode = value;
  185. else if(name == "TrackServer")
  186. trackerList = value;
  187. else if(name == "SuperPeer")
  188. spList = value;
  189. else if(name == "BlockSize")
  190. blockSize = atoi(value.data());
  191. else if(name == "BitRate")
  192. bitRate = static_cast<float>(atof(value.data()));
  193. #ifdef READ_GTV_MEDIA
  194. else if(name == "DataLength") {
  195. attachDataSize = atoi(value.data());
  196. if(attachDataSize > 1024)
  197. break;
  198. attachData = new char[attachDataSize];
  199. }
  200. else if(name == "Data") {
  201. if(attachDataSize > 0) {
  202. char* index = strstr(temp, "Data=");
  203. memcpy(attachData, index+5, attachDataSize); 
  204. }
  205. }
  206. #endif
  207. }
  208. delete is;
  209. is = NULL;
  210. }
  212. // 检查数据
  213. if(resname.empty() || hashcode.empty() || spList.empty() || trackerList.empty() || blockSize != BLOCK_SIZE)
  214. return PRT_BAD_URL;
  216. // 5. 创建实例，并初始化
  217. comm = new Communicator();
  218. if(!comm)
  219. return PRT_ALLOC;
  221. comm->noitfyWindow = notifyWnd;
  222. comm->notifyCode = notifyCode;
  223. comm->channelID = channelID;
  225. P2P_RETURN_TYPE ret = comm->Init(trackerList);
  226. if(ret < PRT_OK)
  227. return ret;
  229. // 6. 开始下载
  230. ret = comm->Request(resname.data(), hashcode.data(), spList.data(), bitRate);
  231. if(ret < PRT_OK)
  232. return ret;
  234. #ifdef READ_GTV_MEDIA
  235. MediaInterval mi;
  236. mi.start = 0;
  237. mi.size = 1000000000;
  238. char* tmpPt = attachData;
  239.   memcpy(&mi.videoType, tmpPt, sizeof(MediaType));
  240. tmpPt += sizeof(MediaType);
  241. mi.videoData = new BYTE[mi.videoType.cbFormat];
  242. if(mi.videoType.cbFormat > 200)
  243. return PRT_BAD_MEDIAFORMAT;
  244. memcpy(mi.videoData, tmpPt, mi.videoType.cbFormat);
  245. tmpPt += mi.videoType.cbFormat;
  246. memcpy(&mi.audioType, tmpPt, sizeof(MediaType));
  247. tmpPt += sizeof(MediaType);
  248. mi.audioData = new BYTE[mi.audioType.cbFormat];
  249. if(mi.audioType.cbFormat > 200)
  250. return PRT_BAD_MEDIAFORMAT;
  251. memcpy(mi.audioData, tmpPt, mi.audioType.cbFormat);
  252. tmpPt += mi.audioType.cbFormat;
  254. delete [] attachData;
  256. comm->currRes->AddMediaInterval(mi);
  257. #endif
  259. return PRT_OK;
  260. }
  262. // 停止当前下载
  263. void Layer1::Stop() {
  264. if(!comm)
  265. return;
  267. comm->logFile.StatusOut("Incoming stop...");
  268. comm->StopCurrentRes();
  269. }
  271. // 缓冲完成
  272. void Layer1::BufferFinished() {
  273. if(!comm || !comm->currRes)
  274. return;
  275. comm->currRes->EnlargeBuffer();
  276. }
  278. // 获取当前资源的传输信息
  279. void Layer1::GetTransferInfo(
  280.  TransferInfo& ti // out, 传输信息
  281.  )
  282. {
  283. if(!comm)
  284. return;
  285. comm->p2pMgr.GetTransferInfo(ti);
  286. }
  288. // 获取当前资源Hash
  289. BOOL Layer1::GetResourceHash(
  290.  LPTSTR buf,  // out, 存储Hash的缓冲区
  291.  UINT& len // in/out, 传入缓冲区的大小，返回Hash的长度
  292.  )
  293. {
  294. if(!comm || !comm->currRes)
  295. return 0;
  296. if(!buf)
  297. return FALSE;
  298. string hashcode = comm->currRes->GetHashCode();
  299. if(len <= hashcode.length())
  300. return FALSE;
  301. len = hashcode.length();
  302. memcpy(buf, hashcode.data(), len);
  303. buf[len] = 0;
  304. return TRUE;
  305. }
  307. // 获取当前资源名
  308. BOOL Layer1::GetFileName(
  309.  LPTSTR buf,  // out, 存储名字的缓冲区
  310.  UINT& len // in/out, 传入缓冲区的大小，返回名字的长度
  311.  )
  312. {
  313. if(!comm || !comm->currRes)
  314. return 0;
  315. if(!buf)
  316. return FALSE;
  317. /// 三种不同版本的客户端，能够拿到不同的信息，所以要分别处理
  318. string resname = comm->currRes->GetResName();
  319. string programname = comm->currRes->GetProgramName();
  320. string channelname = comm->currRes->GetChannelName();
  321. UINT32 progtime = comm->currRes->GetProgramTimeInSeconds();
  322. if(programname.empty()) {
  323. if(len <= resname.length())
  324. return FALSE;
  325. len = resname.length();
  326. sprintf(buf, "%s", resname.data());
  327. return true;
  328. }
  329. if(channelname.empty()) {
  330. if(len <= resname.length()+programname.length()+64)
  331. return FALSE;
  332. len = resname.length()+programname.length()+64;
  333. LONGLONG curTime = comm->currRes->GetOriginTime()/10000000;
  334. sprintf(buf, "%s -> %s %.2d:%.2d:%.2d", resname.data(), programname.data(), 
  335. (int)curTime/3600, (int)curTime%3600/60, (int)curTime%60);
  336. return TRUE;
  337. }
  338. if(len <= channelname.length()+programname.length()+channelname.length()+64)
  339. return FALSE;
  340. len = channelname.length()+programname.length()+channelname.length()+64;
  341. LONGLONG curTime = comm->currRes->GetOriginTime()/10000000;
  342. sprintf(buf, "%s -> %s %.2d:%.2d:%.2d", channelname.data(), programname.data(), 
  343. (int)curTime/3600, (int)curTime%3600/60, (int)curTime%60);
  344. if(progtime)
  345. sprintf(buf+strlen(buf), "/%.2d:%.2d:%.2d", (int)progtime/3600, (int)progtime%3600/60, (int)progtime%60);
  346. return TRUE;
  347. }
  349. // 获取当前资源的缓冲百分比
  350. int Layer1::GetBufferPercent() {
  351. if(!comm || !comm->currRes)
  352. return -1;
  353. return comm->currRes->GetBufferPercent();
  354. }
  356. // 获取当前频道的码率
  357. float Layer1::GetChannelBitRate() {
  358. if(!comm || !comm->currRes)
  359. return 0.0f;
  360. return comm->currRes->GetBitRate();
  361. }
  363. // 获取Sample，可能得到普通Sample或者MediaType(参考struct SampleHeader的注释)
  364. P2P_RETURN_TYPE Layer1::GetData(
  365. SampleHeader& header, // out, 数据头
  366. PBYTE& pData, // out, 存储数据的缓冲区
  367. const UINT maxSize, // in, 缓冲区的长度
  368. const bool bAudio, // in, 获取音频还是视频
  369. const bool bKeySample // in, 是否寻找关键帧
  370. )
  371. {
  372. if(!comm || !comm->currRes)
  373. return PRT_NOT_INIT;
  374. return comm->currRes->GetData(header, pData, maxSize, bAudio, bKeySample);
  375. }
  376. }