1. 程式人生 > 實用技巧 >hls 直播視訊下載器

hls 直播視訊下載器

-----------------------------------------------------------------------------------
#include <curl/curl.h>
#include <string>
#include <iostream>
#include <unistd.h>
#include <stdio.h>
#include <conio.h>
#include <regex>
#include <iterator>
#include <io.h>
#include 
<fstream> #include <getopt.h> #include <list> using namespace std; static char m3u8[1024]; static char g_useragent[1024]; static char g_proxy[100]; static char g_maxlen[50]; static double g_maxduration = 0.0; //static size_t totalsize = 0; static unsigned long long int totalsize = 0; static
char strtotalsize[100]; static string baseurl; static int isDEBUG = 0; static list<pthread_t>listpthreads; void putmsg(const char * msg) { if (!msg) return; FILE * m3u8h = fopen(m3u8, "ab+"); if (m3u8h) { fwrite(msg, 1, strlen(msg), m3u8h); fwrite("\r\n", 1, 2, m3u8h); fclose(m3u8h); } } typedef
struct targ { char url[1024]; char fn[1024]; } targ; size_t writeFunction(void *ptr, size_t size, size_t nmemb, string* data) { data->append((char*) ptr, size * nmemb); return size * nmemb; } size_t writeFunction2(void *ptr, size_t size, size_t nmemb, string* data) { data->append((char*) ptr, size * nmemb); return size * nmemb; } static void deleteNode(pthread_t tid) { list<pthread_t>::iterator it; for (it = listpthreads.begin(); it != listpthreads.end(); it++) { if (tid == *it) { listpthreads.erase(it); } } } static void waitThreads() { list<pthread_t>::iterator it; cout << "Waiting for the end of download threads" << endl; for (it = listpthreads.begin(); it != listpthreads.end(); it++) { //cout << " thread ID " << *it << endl; pthread_join(*it, NULL); } } static void * downts(void *arg) { targ *ta= (targ *)arg; string fullurl = ta->url;; string fn = ta->fn; //cout << fn << endl; auto curl = curl_easy_init(); if (curl) { curl_easy_setopt(curl, CURLOPT_URL, fullurl.c_str()); curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 1L); //curl_easy_setopt(curl, CURLOPT_USERPWD, "user:pass"); curl_easy_setopt(curl, CURLOPT_USERAGENT, g_useragent); curl_easy_setopt(curl, CURLOPT_MAXREDIRS, 50L); curl_easy_setopt(curl, CURLOPT_TCP_KEEPALIVE, 1L); curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 0L); curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 0L); curl_easy_setopt(curl, CURLOPT_TIMEOUT_MS, 10000L); if(strlen(g_proxy)>0) curl_easy_setopt(curl, CURLOPT_PROXY, g_proxy); string response_string; string header_string; curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, writeFunction2); curl_easy_setopt(curl, CURLOPT_WRITEDATA, & response_string); curl_easy_setopt(curl, CURLOPT_HEADERDATA, & header_string); char * url; long response_code; double elapsed; curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, & response_code); curl_easy_getinfo(curl, CURLINFO_TOTAL_TIME, & elapsed); curl_easy_getinfo(curl, CURLINFO_EFFECTIVE_URL, & url); CURLcode res = curl_easy_perform(curl); if (res != CURLE_OK) { if(isDEBUG) fprintf(stderr, "thread curl_easy_perform() failed: %s\n", curl_easy_strerror(res)); if(isDEBUG) cout << "pthread remove " << fn << endl; cout << "*** failed [" << fn << "] ***" <<endl; remove(fn.c_str()); } else { if(isDEBUG) cout << "thread response_code " << response_code << endl; if(isDEBUG) cout << "thread response_string size " << response_string.size() << endl; //cout << "elapsed " << elapsed << endl; //cout << "effective url " << url << endl; //cout << "response_string \n" << response_string << endl; //cout << "header_string \n" << header_string << endl; if( response_string.size() > 0 ) { //cout << fn << endl; FILE *tsh = fopen(fn.c_str(), "wb"); if(tsh) { fwrite(response_string.c_str(), 1, response_string.size(), tsh); fclose(tsh); } totalsize += response_string.size(); if(totalsize<1024) sprintf(strtotalsize,"%10ld ", totalsize); else if(totalsize<1024*1024) sprintf(strtotalsize,"%10.03f KB", (double)totalsize/1024); else if(totalsize<1024*1024*1024) sprintf(strtotalsize,"%10.03f MB", (double)totalsize/1024/1024); else sprintf(strtotalsize,"%10.03f GB", (double)totalsize/1024/1024/1024); } else { cout << "*** response error [" << fn << "] ***" <<endl; remove(fn.c_str()); } } } curl_easy_cleanup(curl); curl = NULL; if(isDEBUG) cout << "pthread exit" << endl; deleteNode(pthread_self()); pthread_exit(NULL); return NULL; } string getbaseurl(string url) { string text=url; string pattern = "((.*)/)"; regex express(pattern); regex_iterator<string::const_iterator> begin(text.cbegin(), text.cend(), express); for (auto iter = begin; iter != sregex_iterator(); iter++) { //cout << iter->str() << endl; return iter->str(); } } string getindex(string text) { string ret = ""; string pattern = "([0-9]{1,15})"; regex express(pattern); regex_iterator<string::const_iterator> begin(text.cbegin(), text.cend(), express); for (auto iter = begin; iter != sregex_iterator(); iter++) { ret = iter->str(); } if (ret == "" || ret.size() <= 0 ) return text; return ret; } static double duration = 0.0; static char strduration[100]; static unsigned int dlidx = 1; static int isMAX = 0; double getlen(string text) { double rd = 0.0; string ret = ""; string pattern = "([0-9.]{1,8})"; regex express(pattern); regex_iterator<string::const_iterator> begin(text.cbegin(), text.cend(), express); for (auto iter = begin; iter != sregex_iterator(); iter++) { ret = iter->str(); } if (ret == "" || ret.size() <= 0 ) return 0.0; rd = atof(ret.c_str()); duration = duration + rd; int dhh=(unsigned int)(duration/3600); int dmm=(unsigned int)(duration)%3600/60; int dss=(unsigned int)(duration)%3600%60; int dms=(unsigned int)(duration*1000)%1000; sprintf(strduration,"%4d %5.03f %02d:%02d:%02d.%03d", dlidx, rd, dhh, dmm, dss, dms); dlidx++; if(g_maxduration != 0.0 && duration >= g_maxduration) isMAX = 1; return rd; } void string_replace( string &strBig, const string &strsrc, const string &strdst) { string::size_type pos = 0; string::size_type srclen = strsrc.size(); string::size_type dstlen = strdst.size(); while( (pos=strBig.find(strsrc, pos)) != string::npos ) { strBig.replace( pos, srclen, strdst ); pos += dstlen; } } static unsigned int dlcount = 0; static unsigned int firstdl = 0; static unsigned int dladdcount = 0; static unsigned int checknodowncount = 0; void gettsurl(string str) { string fullurl; string text=str; if(dlcount==0) { string pattern = "(#EXT-X(.*)|#EXTM3U)"; regex express(pattern); regex_iterator<string::const_iterator> begin(text.cbegin(), text.cend(), express); for (auto iter = begin; iter != sregex_iterator(); iter++) { cout << iter->str() << endl; putmsg(iter->str().c_str()); } } //string pattern = "((.*).ts)"; string pattern = "(#EXTINF:(.*)|(.*).ts)"; regex express(pattern); regex_iterator<string::const_iterator> begin(text.cbegin(), text.cend(), express); string extinfo; for (auto iter = begin; iter != sregex_iterator(); iter++) { fullurl = baseurl + iter->str(); //cout << fullurl << endl; //cout << iter->str() << endl; //downts(fullurl, iter->str()); if( iter->str().substr(0,1) == "#" ) { extinfo = iter->str(); } else { string newfn = getindex(iter->str()); string_replace(newfn, "/", "-"); if(newfn.size()<=4) { char tmp[50]; sprintf(tmp, "%05d", atoi(newfn.c_str())); newfn = tmp; } newfn = newfn + ".ts"; if (access(newfn.c_str(), 0)) { getlen(extinfo); cout << " " << strduration << " " << strtotalsize << " " << newfn << endl; //cout << newfn << endl; putmsg(extinfo.c_str()); putmsg(newfn.c_str()); FILE * tsh = fopen(newfn.c_str(), "wb"); if (tsh) { fclose(tsh); } pthread_t thread; targ ta; memset( & ta, 0x00, sizeof(ta)); strcpy(ta.url, fullurl.c_str()); strcpy(ta.fn, newfn.c_str()); pthread_create( & thread, NULL, downts, & ta); listpthreads.push_back(thread); dladdcount ++; } } } dlcount++; } void help() { cout << "Usage: hlslivedl [options] -i [http://...m3u8]" << endl; cout << " -o set output filename" << endl; cout << " -p set proxy http: https: socks4: socks4a: socks5: socks5h:" << endl; cout << " -u set useragent" << endl; cout << " -t set duration seconds" << endl; cout << " -d debug mode" << endl; cout << " Press [Q] to stop download" << endl; cout << " Version 1.0.7 by NLSoft 2020.07" << endl; } int main(int argc, char** argv) { CURLcode res; static char urlm3u8[1024]; memset(urlm3u8,0,1024); memset(g_proxy,0,100); memset(strduration,0,100); memset(strtotalsize,0,100); strcpy(strtotalsize, " "); if(argc == 1) { help(); exit(0); } strcpy(m3u8, "_index.m3u8"); strcpy(g_useragent, "Mozilla/5.0 (iPhone; CPU iPhone OS 12_0 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/12.0 Mobile/15E148 Safari/604.1"); int option_index = 0; char * user_name = NULL; while ((option_index = getopt(argc, argv, "p:o:u:i:t:d")) != -1) { switch (option_index) { case 'o': strcpy(m3u8, optarg); if(isDEBUG) printf("output filename is %s\n", m3u8); break; case 'u': strcpy(g_useragent, optarg); if(isDEBUG) printf("useragent is %s\n", g_useragent); break; case 'p': strcpy(g_proxy, optarg); if(isDEBUG) printf("proxy is %s\n", g_proxy); break; case 'i': strcpy(urlm3u8, optarg); if(isDEBUG) printf("m3u8 url is %s\n", urlm3u8); break; case 't': strcpy(g_maxlen, optarg); g_maxduration = atof(g_maxlen); if(isDEBUG) printf("g_maxlen url is %s\n", g_maxlen); break; case 'd': isDEBUG = 1; if(isDEBUG) printf("select debug mode\n"); break; } } if(strlen(urlm3u8)<=0) { help(); exit(0); } FILE * m3u8h = fopen(m3u8, "wb"); if (m3u8h) { fclose(m3u8h); } curl_global_init(CURL_GLOBAL_DEFAULT); baseurl = getbaseurl(urlm3u8); if(isDEBUG) cout << baseurl << endl; auto curl = curl_easy_init(); if (curl) { while (1) { dladdcount = 0; //curl_easy_setopt(curl, CURLOPT_URL, "https://api.github.com/repos/whoshuu/cpr/contributors?anon=true&key=value"); curl_easy_setopt(curl, CURLOPT_URL, urlm3u8); curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 1L); //curl_easy_setopt(curl, CURLOPT_USERPWD, "user:pass"); curl_easy_setopt(curl, CURLOPT_USERAGENT, g_useragent); curl_easy_setopt(curl, CURLOPT_MAXREDIRS, 50L); curl_easy_setopt(curl, CURLOPT_TCP_KEEPALIVE, 1L); curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 0L); curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 0L); curl_easy_setopt(curl, CURLOPT_TIMEOUT_MS, 5000L); if(strlen(g_proxy)>0) curl_easy_setopt(curl, CURLOPT_PROXY, g_proxy); string response_string; string header_string; curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, writeFunction); curl_easy_setopt(curl, CURLOPT_WRITEDATA, & response_string); curl_easy_setopt(curl, CURLOPT_HEADERDATA, & header_string); char * url; long response_code; double elapsed; curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, & response_code); curl_easy_getinfo(curl, CURLINFO_TOTAL_TIME, & elapsed); curl_easy_getinfo(curl, CURLINFO_EFFECTIVE_URL, & url); res = curl_easy_perform(curl); if (res != CURLE_OK) { if(isDEBUG) fprintf(stderr, "main curl_easy_perform() failed: %s\n", curl_easy_strerror(res)); } else { if(isDEBUG) cout << "main response_code " << response_code << endl; //cout << "elapsed " << elapsed << endl; //cout << "effective url " << url << endl; //cout << "response_string \n" << response_string << endl; //cout << "header_string \n" << header_string << endl; if( response_code == 0 ) continue; if( response_code == 200 ) { // skip first ts if(firstdl != 0) gettsurl(response_string); firstdl ++; } else { waitThreads(); if(isDEBUG) cout << "#EXT-X-ENDLIST\r\n" << endl; putmsg("#EXT-X-ENDLIST"); if(isDEBUG) cout << "main response_code " << response_code << "\n" << endl; break; }; } char exitflag = '\0'; if (_kbhit()) { exitflag = _getch(); if (exitflag == 'q' || exitflag == 'Q') { waitThreads(); if(isDEBUG) cout << "#EXT-X-ENDLIST\r\n" << endl; putmsg("#EXT-X-ENDLIST"); break; } } if(dladdcount>0) checknodowncount = 0; else checknodowncount ++; //printf("dladdcount %d checknodowncount %ld\n", dladdcount, checknodowncount); // No files were downloaded in 60 seconds if(checknodowncount>=60 || isMAX) { waitThreads(); if(isDEBUG) cout << "#EXT-X-ENDLIST\r\n" << endl; putmsg("#EXT-X-ENDLIST"); break; } usleep(1000 * 1000); //sleep(1); } } curl_easy_cleanup(curl); curl = NULL; curl_global_cleanup(); if(isDEBUG) cout << "main exit" << endl; return 0; } ----------------------------------------------------------------------------------- echo g++ hlslivedl.cpp g++ -w hlslivedl.cpp -o hlslivedl.exe -DCURL_STATICLIB -ID:/MSYS/local/include -LD:/MSYS/local/lib -lcurl -lssl -lcrypto -lssl -lcrypto -lgdi32 -lwldap32 -lz -lws2_32 -lpsapi -lws2_32 -lmswsock -lshlwapi -lcrypt32 -static echo strip hlslivedl.exe strip hlslivedl.exe -----------------------------------------------------------------------------------