| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723 | 
							- /* <!-- copyright */
 
- /*
 
-  * aria2 - The high speed download utility
 
-  *
 
-  * Copyright (C) 2006 Tatsuhiro Tsujikawa
 
-  *
 
-  * This program is free software; you can redistribute it and/or modify
 
-  * it under the terms of the GNU General Public License as published by
 
-  * the Free Software Foundation; either version 2 of the License, or
 
-  * (at your option) any later version.
 
-  *
 
-  * This program is distributed in the hope that it will be useful,
 
-  * but WITHOUT ANY WARRANTY; without even the implied warranty of
 
-  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
-  * GNU General Public License for more details.
 
-  *
 
-  * You should have received a copy of the GNU General Public License
 
-  * along with this program; if not, write to the Free Software
 
-  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
-  *
 
-  * In addition, as a special exception, the copyright holders give
 
-  * permission to link the code of portions of this program with the
 
-  * OpenSSL library under certain conditions as described in each
 
-  * individual source file, and distribute linked combinations
 
-  * including the two.
 
-  * You must obey the GNU General Public License in all respects
 
-  * for all of the code used other than OpenSSL.  If you modify
 
-  * file(s) with this exception, you may extend this exception to your
 
-  * version of the file(s), but you are not obligated to do so.  If you
 
-  * do not wish to do so, delete this exception statement from your
 
-  * version.  If you delete this exception statement from all source
 
-  * files in the program, then also delete it here.
 
-  */
 
- /* copyright --> */
 
- #include "util.h"
 
- #include <signal.h>
 
- #include <limits.h>
 
- #include <stdint.h>
 
- #include <cerrno>
 
- #include <cassert>
 
- #include <cstring>
 
- #include <cstdio>
 
- #include <cstdlib>
 
- #include <sstream>
 
- #include <ostream>
 
- #include <algorithm>
 
- #include <fstream>
 
- #include <iomanip>
 
- #ifdef HAVE_LIBGCRYPT
 
- # include <gcrypt.h>
 
- #elif HAVE_OPENSSL
 
- # include <openssl/rand.h>
 
- # include "SimpleRandomizer.h"
 
- #endif // HAVE_OPENSSL
 
- #include "File.h"
 
- #include "message.h"
 
- #include "Randomizer.h"
 
- #include "a2netcompat.h"
 
- #include "DlAbortEx.h"
 
- #include "BitfieldMan.h"
 
- #include "DefaultDiskWriter.h"
 
- #include "FatalException.h"
 
- #include "FileEntry.h"
 
- #include "fmt.h"
 
- #include "A2STR.h"
 
- #include "array_fun.h"
 
- #include "bitfield.h"
 
- #include "DownloadHandlerConstants.h"
 
- #include "RequestGroup.h"
 
- #include "LogFactory.h"
 
- #include "Logger.h"
 
- #include "Option.h"
 
- #include "DownloadContext.h"
 
- #include "BufferedFile.h"
 
- #ifdef ENABLE_MESSAGE_DIGEST
 
- # include "MessageDigest.h"
 
- # include "message_digest_helper.h"
 
- #endif // ENABLE_MESSAGE_DIGEST
 
- // For libc6 which doesn't define ULLONG_MAX properly because of broken limits.h
 
- #ifndef ULLONG_MAX
 
- # define ULLONG_MAX 18446744073709551615ULL
 
- #endif // ULLONG_MAX
 
- namespace aria2 {
 
- #ifdef __MINGW32__
 
- namespace {
 
- int utf8ToWChar(wchar_t* out, size_t outLength, const char* src)
 
- {
 
-   return MultiByteToWideChar(CP_UTF8, 0, src, -1, out, outLength);
 
- }
 
- } // namespace
 
- namespace {
 
- int ansiToWChar(wchar_t* out, size_t outLength, const char* src)
 
- {
 
-   return MultiByteToWideChar(CP_ACP, 0, src, -1, out, outLength);
 
- }
 
- } // namespace
 
- namespace {
 
- int wCharToUtf8(char* out, size_t outLength, const wchar_t* src)
 
- {
 
-   return WideCharToMultiByte(CP_UTF8, 0, src, -1, out, outLength, 0, 0);
 
- }
 
- } // namespace
 
- namespace {
 
- int wCharToAnsi(char* out, size_t outLength, const wchar_t* src)
 
- {
 
-   return WideCharToMultiByte(CP_ACP, 0, src, -1, out, outLength, 0, 0);
 
- }
 
- } // namespace
 
- std::wstring utf8ToWChar(const char* src)
 
- {
 
-   int len = utf8ToWChar(0, 0, src);
 
-   if(len == 0) {
 
-     abort();
 
-   }
 
-   array_ptr<wchar_t> buf(new wchar_t[len]);
 
-   len = utf8ToWChar(buf, len, src);
 
-   if(len == 0) {
 
-     abort();
 
-   } else {
 
-     std::wstring dest(buf);
 
-     return dest;
 
-   }
 
- }
 
- std::wstring utf8ToWChar(const std::string& src)
 
- {
 
-   return utf8ToWChar(src.c_str());
 
- }
 
- std::string utf8ToNative(const std::string& src)
 
- {
 
-   std::wstring wsrc = utf8ToWChar(src);
 
-   int len = wCharToAnsi(0, 0, wsrc.c_str());
 
-   if(len == 0) {
 
-     abort();
 
-   }
 
-   array_ptr<char> buf(new char[len]);
 
-   len = wCharToAnsi(buf, len, wsrc.c_str());
 
-   if(len == 0) {
 
-     abort();
 
-   } else {
 
-     std::string dest(buf);
 
-     return dest;
 
-   }
 
- }
 
- std::string wCharToUtf8(const std::wstring& wsrc)
 
- {
 
-   int len = wCharToUtf8(0, 0, wsrc.c_str());
 
-   if(len == 0) {
 
-     abort();
 
-   }
 
-   array_ptr<char> buf(new char[len]);
 
-   len = wCharToUtf8(buf, len, wsrc.c_str());
 
-   if(len == 0) {
 
-     abort();
 
-   } else {
 
-     std::string dest(buf);
 
-     return dest;
 
-   }
 
- }
 
- std::string nativeToUtf8(const std::string& src)
 
- {
 
-   int len = ansiToWChar(0, 0, src.c_str());
 
-   if(len == 0) {
 
-     abort();
 
-   }
 
-   array_ptr<wchar_t> buf(new wchar_t[len]);
 
-   len = ansiToWChar(buf, len, src.c_str());
 
-   if(len == 0) {
 
-     abort();
 
-   } else {
 
-     return wCharToUtf8(std::wstring(buf));
 
-   }
 
- }
 
- #endif // __MINGW32__
 
- namespace util {
 
- const std::string DEFAULT_STRIP_CHARSET("\r\n\t ");
 
- std::string strip(const std::string& str, const std::string& chars)
 
- {
 
-   return stripIter(str.begin(), str.end(), chars);
 
- }
 
- void divide
 
- (std::pair<std::string, std::string>& hp, const std::string& src, char delim)
 
- {
 
-   std::string::const_iterator first = src.begin();
 
-   std::string::const_iterator last = src.end();
 
-   std::string::const_iterator dpos = std::find(first, last, delim);
 
-   if(dpos == last) {
 
-     hp.first = strip(src);
 
-     hp.second = A2STR::NIL;
 
-   } else {
 
-     hp.first = stripIter(first, dpos);
 
-     hp.second = stripIter(dpos+1, last);
 
-   }
 
- }
 
- std::string itos(int64_t value, bool comma)
 
- {
 
-   bool flag = false;
 
-   std::string str;
 
-   if(value < 0) {
 
-     if(value == INT64_MIN) {
 
-       if(comma) {
 
-         str = "-9,223,372,036,854,775,808";
 
-       } else {
 
-         str = "-9223372036854775808";
 
-       }
 
-       return str;
 
-     }
 
-     flag = true;
 
-     value = -value;
 
-   }
 
-   str = uitos(value, comma);
 
-   if(flag) {
 
-     str.insert(str.begin(), '-');
 
-   }
 
-   return str;
 
- }
 
- int64_t difftv(struct timeval tv1, struct timeval tv2) {
 
-   if((tv1.tv_sec < tv2.tv_sec) ||
 
-      ((tv1.tv_sec == tv2.tv_sec) && (tv1.tv_usec < tv2.tv_usec))) {
 
-     return 0;
 
-   }
 
-   return ((int64_t)(tv1.tv_sec-tv2.tv_sec)*1000000+
 
-           tv1.tv_usec-tv2.tv_usec);
 
- }
 
- int32_t difftvsec(struct timeval tv1, struct timeval tv2) {
 
-   if(tv1.tv_sec < tv2.tv_sec) {
 
-     return 0;
 
-   }
 
-   return tv1.tv_sec-tv2.tv_sec;
 
- }
 
- bool startsWith(const std::string& target, const std::string& part) {
 
-   if(target.size() < part.size()) {
 
-     return false;
 
-   }
 
-   if(part.empty()) {
 
-     return true;
 
-   }
 
-   if(target.find(part) == 0) {
 
-     return true;
 
-   } else {
 
-     return false;
 
-   }
 
- }
 
- bool endsWith(const std::string& target, const std::string& part) {
 
-   if(target.size() < part.size()) {
 
-     return false;
 
-   }
 
-   if(part.empty()) {
 
-     return true;
 
-   }
 
-   if(target.rfind(part) == target.size()-part.size()) {
 
-     return true;
 
-   } else {
 
-     return false;
 
-   }
 
- }
 
- std::string replace(const std::string& target, const std::string& oldstr, const std::string& newstr) {
 
-   if(target.empty() || oldstr.empty()) {
 
-     return target;
 
-   }
 
-   std::string result;
 
-   std::string::size_type p = 0;
 
-   std::string::size_type np = target.find(oldstr);
 
-   while(np != std::string::npos) {
 
-     result += target.substr(p, np-p);
 
-     result += newstr;
 
-     p = np+oldstr.size();
 
-     np = target.find(oldstr, p);
 
-   }
 
-   result += target.substr(p);
 
-   return result;
 
- }
 
- bool isAlpha(const char c)
 
- {
 
-   return ('A' <= c && c <= 'Z') || ('a' <= c && c <= 'z');
 
- }
 
- bool isDigit(const char c)
 
- {
 
-   return '0' <= c && c <= '9';
 
- }
 
- bool isHexDigit(const char c)
 
- {
 
-   return isDigit(c) || ('A' <= c && c <= 'F') || ('a' <= c && c <= 'f');
 
- }
 
- bool isHexDigit(const std::string& s)
 
- {
 
-   for(std::string::const_iterator i = s.begin(), eoi = s.end(); i != eoi; ++i) {
 
-     if(!isHexDigit(*i)) {
 
-       return false;
 
-     }
 
-   }
 
-   return true;
 
- }
 
- bool inRFC3986ReservedChars(const char c)
 
- {
 
-   static const char reserved[] = {
 
-     ':' , '/' , '?' , '#' , '[' , ']' , '@',
 
-     '!' , '$' , '&' , '\'' , '(' , ')',
 
-     '*' , '+' , ',' , ';' , '=' };
 
-   return std::find(vbegin(reserved), vend(reserved), c) != vend(reserved);
 
- }
 
- bool inRFC3986UnreservedChars(const char c)
 
- {
 
-   static const char unreserved[] = { '-', '.', '_', '~' };
 
-   return isAlpha(c) || isDigit(c) ||
 
-     std::find(vbegin(unreserved), vend(unreserved), c) != vend(unreserved);
 
- }
 
- bool inRFC2978MIMECharset(const char c)
 
- {
 
-   static const char chars[] = {
 
-     '!', '#', '$', '%', '&',
 
-     '\'', '+', '-', '^', '_',
 
-     '`', '{', '}', '~'
 
-   };
 
-   return isAlpha(c) || isDigit(c) ||
 
-     std::find(vbegin(chars), vend(chars), c) != vend(chars);
 
- }
 
- bool inRFC2616HttpToken(const char c)
 
- {
 
-   static const char chars[] = {
 
-     '!', '#', '$', '%', '&', '\'', '*', '+', '-', '.',
 
-     '^', '_', '`', '|', '~'
 
-   };
 
-   return isAlpha(c) || isDigit(c) ||
 
-     std::find(vbegin(chars), vend(chars), c) != vend(chars);
 
- }
 
- namespace {
 
- bool isUtf8Tail(unsigned char ch)
 
- {
 
-   return in(ch, 0x80u, 0xbfu);
 
- }
 
- } // namespace
 
- bool isUtf8(const std::string& str)
 
- {
 
-   for(std::string::const_iterator s = str.begin(), eos = str.end(); s != eos;
 
-       ++s) {
 
-     unsigned char firstChar = *s;
 
-     // See ABNF in http://tools.ietf.org/search/rfc3629#section-4
 
-     if(in(firstChar, 0x20u, 0x7eu) ||
 
-        firstChar == 0x09u || firstChar == 0x0au ||firstChar == 0x0du) {
 
-       // UTF8-1 (without ctrl chars)
 
-     } else if(in(firstChar, 0xc2u, 0xdfu)) {
 
-       // UTF8-2
 
-       if(++s == eos || !isUtf8Tail(*s)) {
 
-         return false;
 
-       }
 
-     } else if(0xe0u == firstChar) {
 
-       // UTF8-3
 
-       if(++s == eos || !in(static_cast<unsigned char>(*s), 0xa0u, 0xbfu) ||
 
-          ++s == eos || !isUtf8Tail(*s)) {
 
-         return false;
 
-       }
 
-     } else if(in(firstChar, 0xe1u, 0xecu) || in(firstChar, 0xeeu, 0xefu)) {
 
-       // UTF8-3
 
-       if(++s == eos || !isUtf8Tail(*s) ||
 
-          ++s == eos || !isUtf8Tail(*s)) {
 
-         return false;
 
-       }
 
-     } else if(0xedu == firstChar) {
 
-       // UTF8-3
 
-       if(++s == eos || !in(static_cast<unsigned char>(*s), 0x80u, 0x9fu) ||
 
-          ++s == eos || !isUtf8Tail(*s)) {
 
-         return false;
 
-       } 
 
-     } else if(0xf0u == firstChar) {
 
-       // UTF8-4
 
-       if(++s == eos || !in(static_cast<unsigned char>(*s), 0x90u, 0xbfu) ||
 
-          ++s == eos || !isUtf8Tail(*s) ||
 
-          ++s == eos || !isUtf8Tail(*s)) {
 
-         return false;
 
-       }
 
-     } else if(in(firstChar, 0xf1u, 0xf3u)) {
 
-       // UTF8-4
 
-       if(++s == eos || !isUtf8Tail(*s) ||
 
-          ++s == eos || !isUtf8Tail(*s) ||
 
-          ++s == eos || !isUtf8Tail(*s)) {
 
-         return false;
 
-       }
 
-     } else if(0xf4u == firstChar) {
 
-       // UTF8-4
 
-       if(++s == eos || !in(static_cast<unsigned char>(*s), 0x80u, 0x8fu) ||
 
-          ++s == eos || !isUtf8Tail(*s) ||
 
-          ++s == eos || !isUtf8Tail(*s)) {
 
-         return false;
 
-       }
 
-     } else {
 
-       return false;
 
-     }
 
-   }
 
-   return true;
 
- }
 
- std::string percentEncode(const unsigned char* target, size_t len)
 
- {
 
-   std::string dest;
 
-   for(size_t i = 0; i < len; ++i) {
 
-     if(inRFC3986UnreservedChars(target[i])) {
 
-       dest += target[i];
 
-     } else {
 
-       dest.append(fmt("%%%02X", target[i]));
 
-     }
 
-   }
 
-   return dest;
 
- }
 
- std::string percentEncode(const std::string& target)
 
- {
 
-   return percentEncode(reinterpret_cast<const unsigned char*>(target.c_str()),
 
-                        target.size());
 
- }
 
- std::string percentEncodeMini(const std::string& src)
 
- {
 
-   std::string result;
 
-   for(std::string::const_iterator i = src.begin(), eoi = src.end(); i != eoi;
 
-       ++i) {
 
-     // Non-Printable ASCII and non-ASCII chars + some ASCII chars.
 
-     unsigned char c = *i;
 
-     if(in(c, 0x00u, 0x20u) || c >= 0x7fu ||
 
-        // Chromium escapes following characters. Firefox4 escapes
 
-        // more.
 
-        c == '"' || c == '<' || c == '>') {
 
-       result += fmt("%%%02X", c);
 
-     } else {
 
-       result += c;
 
-     }
 
-   }
 
-   return result;
 
- }
 
- std::string torrentPercentEncode(const unsigned char* target, size_t len) {
 
-   std::string dest;
 
-   for(size_t i = 0; i < len; ++i) {
 
-     if(isAlpha(target[i]) || isDigit(target[i])) {
 
-       dest += target[i];
 
-     } else {
 
-       dest.append(fmt("%%%02X", target[i]));
 
-     }
 
-   }
 
-   return dest;
 
- }
 
- std::string torrentPercentEncode(const std::string& target)
 
- {
 
-   return torrentPercentEncode
 
-     (reinterpret_cast<const unsigned char*>(target.c_str()), target.size());
 
- }
 
- std::string percentDecode(const std::string& target) {
 
-   std::string result;
 
-   for(std::string::const_iterator itr = target.begin(), eoi = target.end();
 
-       itr != eoi; ++itr) {
 
-     if(*itr == '%') {
 
-       if(itr+1 != target.end() && itr+2 != target.end() &&
 
-          isHexDigit(*(itr+1)) && isHexDigit(*(itr+2))) {
 
-         result += parseInt(std::string(itr+1, itr+3), 16);
 
-         itr += 2;
 
-       } else {
 
-         result += *itr;
 
-       }
 
-     } else {
 
-       result += *itr;
 
-     }
 
-   }
 
-   return result;
 
- }
 
- std::string toHex(const unsigned char* src, size_t len) {
 
-   std::string out(len*2, '\0');
 
-   std::string::iterator o = out.begin();
 
-   const unsigned char* last = src+len;
 
-   for(const unsigned char* i = src; i != last; ++i) {
 
-     *o = (*i >> 4);
 
-     *(o+1) = (*i)&0x0fu;
 
-     for(int j = 0; j < 2; ++j) {
 
-       if(*o < 10) {
 
-         *o += '0';
 
-       } else {
 
-         *o += 'a'-10;
 
-       }
 
-       ++o;
 
-     }
 
-   }
 
-   return out;
 
- }
 
- std::string toHex(const char* src, size_t len)
 
- {
 
-   return toHex(reinterpret_cast<const unsigned char*>(src), len);
 
- }
 
- std::string toHex(const std::string& src)
 
- {
 
-   return toHex(reinterpret_cast<const unsigned char*>(src.c_str()), src.size());
 
- }
 
- namespace {
 
- unsigned int hexCharToUInt(unsigned char ch)
 
- {
 
-   if('a' <= ch && ch <= 'f') {
 
-     ch -= 'a';
 
-     ch += 10;
 
-   } else if('A' <= ch && ch <= 'F') {
 
-     ch -= 'A';
 
-     ch += 10;
 
-   } else if('0' <= ch && ch <= '9') {
 
-     ch -= '0';
 
-   } else {
 
-     ch = 255;
 
-   }
 
-   return ch;
 
- }
 
- } // namespace
 
- std::string fromHex(const std::string& src)
 
- {
 
-   std::string dest;
 
-   if(src.size()%2) {
 
-     return dest;
 
-   }
 
-   for(size_t i = 0, eoi = src.size(); i < eoi; i += 2) {
 
-     unsigned char high = hexCharToUInt(src[i]);
 
-     unsigned char low = hexCharToUInt(src[i+1]);
 
-     if(high == 255 || low == 255) {
 
-       dest.clear();
 
-       return dest;
 
-     }
 
-     dest += (high*16+low);
 
-   }
 
-   return dest;
 
- }
 
- FILE* openFile(const std::string& filename, const std::string& mode) {
 
-   FILE* file = fopen(filename.c_str(), mode.c_str());
 
-   return file;
 
- }
 
- bool isPowerOf(int num, int base) {
 
-   if(base <= 0) { return false; }
 
-   if(base == 1) { return true; }
 
-   while(num%base == 0) {
 
-     num /= base;
 
-     if(num == 1) {
 
-       return true;
 
-     }
 
-   }
 
-   return false;
 
- }
 
- std::string secfmt(time_t sec) {
 
-   std::string str;
 
-   if(sec >= 3600) {
 
-     str = itos(sec/3600);
 
-     str += "h";
 
-     sec %= 3600;
 
-   }
 
-   if(sec >= 60) {
 
-     int min = sec/60;
 
-     if(min < 10) {
 
-       str += "0";
 
-     }
 
-     str += itos(min);
 
-     str += "m";
 
-     sec %= 60;
 
-   }
 
-   if(sec < 10) {
 
-     str += "0";
 
-   }
 
-   str += itos(sec);
 
-   str += "s";
 
-   return str;
 
- }
 
- int getNum(const char* buf, int offset, size_t length) {
 
-   char* temp = new char[length+1];
 
-   memcpy(temp, buf+offset, length);
 
-   temp[length] = '\0';
 
-   int x = strtol(temp, 0, 10);
 
-   delete [] temp;
 
-   return x;
 
- }
 
- int32_t parseInt(const std::string& s, int32_t base)
 
- {
 
-   int64_t v = parseLLInt(s, base);
 
-   if(v < INT32_MIN || INT32_MAX < v) {
 
-     throw DL_ABORT_EX(fmt(MSG_STRING_INTEGER_CONVERSION_FAILURE,
 
-                           s.c_str()));
 
-   }
 
-   return v;
 
- }
 
- bool parseIntNoThrow(int32_t& result, const std::string& s, int base)
 
- {
 
-   // Without trim, strtol("  -1  ",..) emits error.
 
-   std::string trimed = strip(s);
 
-   if(trimed.empty()) {
 
-     return false;
 
-   }
 
-   char* stop;
 
-   errno = 0;
 
-   long int v = strtol(trimed.c_str(), &stop, base);
 
-   if(*stop != '\0') {
 
-     return false;
 
-   } else if(((v == LONG_MAX || v == LONG_MIN) && (errno == ERANGE)) ||
 
-             v < INT32_MIN || INT32_MAX < v) {
 
-     return false;
 
-   }
 
-   result = v;
 
-   return true;
 
- }
 
- uint32_t parseUInt(const std::string& s, int base)
 
- {
 
-   uint64_t v = parseULLInt(s, base);
 
-   if(UINT32_MAX < v) {
 
-     throw DL_ABORT_EX(fmt(MSG_STRING_INTEGER_CONVERSION_FAILURE,
 
-                           s.c_str()));
 
-   }
 
-   return v;
 
- }
 
- bool parseUIntNoThrow(uint32_t& result, const std::string& s, int base)
 
- {
 
-   // Without trim, strtol("  -1  ",..) emits error.
 
-   std::string trimed = strip(s);
 
-   if(trimed.empty()) {
 
-     return false;
 
-   }
 
-   // We don't allow negative number.
 
-   if(trimed[0] == '-') {
 
-     return false;
 
-   }
 
-   char* stop;
 
-   errno = 0;
 
-   unsigned long int v = strtoul(trimed.c_str(), &stop, base);
 
-   if(*stop != '\0') {
 
-     return false;
 
-   } else if(((v == ULONG_MAX) && (errno == ERANGE)) || (v > UINT32_MAX)) {
 
-     return false;
 
-   }
 
-   result = v;
 
-   return true;
 
- }
 
- int64_t parseLLInt(const std::string& s, int32_t base)
 
- {
 
-   std::string trimed = strip(s);
 
-   if(trimed.empty()) {
 
-     throw DL_ABORT_EX(fmt(MSG_STRING_INTEGER_CONVERSION_FAILURE,
 
-                           "empty string"));
 
-   }
 
-   char* stop;
 
-   errno = 0;
 
-   int64_t v = strtoll(trimed.c_str(), &stop, base);
 
-   if(*stop != '\0') {
 
-     throw DL_ABORT_EX(fmt(MSG_STRING_INTEGER_CONVERSION_FAILURE,
 
-                           trimed.c_str()));
 
-   } else if(((v == INT64_MIN) || (v == INT64_MAX)) && (errno == ERANGE)) {
 
-     throw DL_ABORT_EX(fmt(MSG_STRING_INTEGER_CONVERSION_FAILURE,
 
-                           trimed.c_str()));
 
-   }
 
-   return v;
 
- }
 
- bool parseLLIntNoThrow(int64_t& result, const std::string& s, int base)
 
- {
 
-   // Without trim, strtol("  -1  ",..) emits error.
 
-   std::string trimed = strip(s);
 
-   if(trimed.empty()) {
 
-     return false;
 
-   }
 
-   char* stop;
 
-   errno = 0;
 
-   int64_t v = strtoll(trimed.c_str(), &stop, base);
 
-   if(*stop != '\0') {
 
-     return false;
 
-   } else if(((v == INT64_MIN) || (v == INT64_MAX)) && (errno == ERANGE)) {
 
-     return false;
 
-   }
 
-   result = v;
 
-   return true;
 
- }
 
- uint64_t parseULLInt(const std::string& s, int base)
 
- {
 
-   std::string trimed = strip(s);
 
-   if(trimed.empty()) {
 
-     throw DL_ABORT_EX(fmt(MSG_STRING_INTEGER_CONVERSION_FAILURE,
 
-                           "empty string"));
 
-   }
 
-   // We don't allow negative number.
 
-   if(trimed[0] == '-') {
 
-     throw DL_ABORT_EX(fmt(MSG_STRING_INTEGER_CONVERSION_FAILURE,
 
-                           trimed.c_str()));
 
-   }
 
-   char* stop;
 
-   errno = 0;
 
-   uint64_t v = strtoull(trimed.c_str(), &stop, base);
 
-   if(*stop != '\0') {
 
-     throw DL_ABORT_EX(fmt(MSG_STRING_INTEGER_CONVERSION_FAILURE,
 
-                           trimed.c_str()));
 
-   } else if((v == ULLONG_MAX) && (errno == ERANGE)) {
 
-     throw DL_ABORT_EX(fmt(MSG_STRING_INTEGER_CONVERSION_FAILURE,
 
-                           trimed.c_str()));
 
-   }
 
-   return v;
 
- }
 
- IntSequence parseIntRange(const std::string& src)
 
- {
 
-   IntSequence::Values values;
 
-   std::string temp = src;
 
-   while(temp.size()) {
 
-     std::pair<std::string, std::string> p;
 
-     divide(p, temp, ',');
 
-     temp = p.second;
 
-     if(p.first.empty()) {
 
-       continue;
 
-     }
 
-     if(p.first.find("-") == std::string::npos) {
 
-       int32_t v = parseInt(p.first.c_str());
 
-       values.push_back(IntSequence::Value(v, v+1));
 
-     } else {
 
-       std::pair<std::string, std::string> vp;
 
-       divide(vp, p.first.c_str(), '-');
 
-       if(vp.first.empty() || vp.second.empty()) {
 
-         throw DL_ABORT_EX(fmt(MSG_INCOMPLETE_RANGE, p.first.c_str()));
 
-       }
 
-       int32_t v1 = parseInt(vp.first.c_str());
 
-       int32_t v2 = parseInt(vp.second.c_str());
 
-       values.push_back(IntSequence::Value(v1, v2+1));
 
-     } 
 
-   }
 
-   return values;
 
- }
 
- namespace {
 
- void computeHeadPieces
 
- (std::vector<size_t>& indexes,
 
-  const std::vector<SharedHandle<FileEntry> >& fileEntries,
 
-  size_t pieceLength,
 
-  uint64_t head)
 
- {
 
-   if(head == 0) {
 
-     return;
 
-   }
 
-   for(std::vector<SharedHandle<FileEntry> >::const_iterator fi =
 
-         fileEntries.begin(), eoi = fileEntries.end(); fi != eoi; ++fi) {
 
-     if((*fi)->getLength() == 0) {
 
-       continue;
 
-     }
 
-     size_t lastIndex =
 
-       ((*fi)->getOffset()+std::min(head, (*fi)->getLength())-1)/pieceLength;
 
-     for(size_t index = (*fi)->getOffset()/pieceLength;
 
-         index <= lastIndex; ++index) {
 
-       indexes.push_back(index);
 
-     }
 
-   }
 
- }
 
- } // namespace
 
- namespace {
 
- void computeTailPieces
 
- (std::vector<size_t>& indexes,
 
-  const std::vector<SharedHandle<FileEntry> >& fileEntries,
 
-  size_t pieceLength,
 
-  uint64_t tail)
 
- {
 
-   if(tail == 0) {
 
-     return;
 
-   }
 
-   for(std::vector<SharedHandle<FileEntry> >::const_iterator fi =
 
-         fileEntries.begin(), eoi = fileEntries.end(); fi != eoi; ++fi) {
 
-     if((*fi)->getLength() == 0) {
 
-       continue;
 
-     }
 
-     uint64_t endOffset = (*fi)->getLastOffset();
 
-     size_t fromIndex =
 
-       (endOffset-1-(std::min(tail, (*fi)->getLength())-1))/pieceLength;
 
-     for(size_t index = fromIndex; index <= (endOffset-1)/pieceLength;
 
-         ++index) {
 
-       indexes.push_back(index);
 
-     }
 
-   }
 
- }
 
- } // namespace
 
- void parsePrioritizePieceRange
 
- (std::vector<size_t>& result, const std::string& src,
 
-  const std::vector<SharedHandle<FileEntry> >& fileEntries,
 
-  size_t pieceLength,
 
-  uint64_t defaultSize)
 
- {
 
-   std::vector<size_t> indexes;
 
-   std::vector<std::string> parts;
 
-   split(src, std::back_inserter(parts), ",", true);
 
-   for(std::vector<std::string>::const_iterator i = parts.begin(),
 
-         eoi = parts.end(); i != eoi; ++i) {
 
-     if((*i) == "head") {
 
-       computeHeadPieces(indexes, fileEntries, pieceLength, defaultSize);
 
-     } else if(util::startsWith(*i, "head=")) {
 
-       std::string sizestr = std::string((*i).begin()+(*i).find("=")+1,
 
-                                         (*i).end());
 
-       computeHeadPieces(indexes, fileEntries, pieceLength,
 
-                         std::max((int64_t)0, getRealSize(sizestr)));
 
-     } else if((*i) == "tail") {
 
-       computeTailPieces(indexes, fileEntries, pieceLength, defaultSize);
 
-     } else if(util::startsWith(*i, "tail=")) {
 
-       std::string sizestr = std::string((*i).begin()+(*i).find("=")+1,
 
-                                         (*i).end());
 
-       computeTailPieces(indexes, fileEntries, pieceLength,
 
-                         std::max((int64_t)0, getRealSize(sizestr)));
 
-     } else {
 
-       throw DL_ABORT_EX(fmt("Unrecognized token %s", (*i).c_str()));
 
-     }
 
-   }
 
-   std::sort(indexes.begin(), indexes.end());
 
-   indexes.erase(std::unique(indexes.begin(), indexes.end()), indexes.end());
 
-   result.insert(result.end(), indexes.begin(), indexes.end());
 
- }
 
- // Converts ISO/IEC 8859-1 string to UTF-8 string.  If there is a
 
- // character not in ISO/IEC 8859-1, returns empty string.
 
- std::string iso8859ToUtf8(const std::string& src)
 
- {
 
-   std::string dest;
 
-   for(std::string::const_iterator itr = src.begin(), eoi = src.end();
 
-       itr != eoi; ++itr) {
 
-     unsigned char c = *itr;
 
-     if(0xa0u <= c) {
 
-       if(c <= 0xbfu) {
 
-         dest += 0xc2u;
 
-       } else {
 
-         dest += 0xc3u;
 
-       }
 
-       dest += c&(~0x40u);
 
-     } else if(0x80u <= c && c <= 0x9fu) {
 
-       return A2STR::NIL;
 
-     } else {
 
-       dest += c;
 
-     }
 
-   }
 
-   return dest;
 
- }
 
- namespace {
 
- template<typename OutputIterator>
 
- void parseParam(OutputIterator out, const std::string& header)
 
- {
 
-   for(std::string::const_iterator i = header.begin(), eoi = header.end();
 
-       i != eoi;) {
 
-     std::string::const_iterator paramFirst = i;
 
-     std::string::const_iterator paramLast = paramFirst;
 
-     for(; paramLast != eoi && *paramLast != '=' && *paramLast != ';';
 
-         ++paramLast);
 
-     std::string param;
 
-     if(paramLast == eoi || *paramLast == ';') {
 
-       // No value, parmname only
 
-       param = std::string(paramFirst, paramLast);
 
-     } else {
 
-       for(; paramLast != eoi && *paramLast != '"' && *paramLast != ';';
 
-           ++paramLast);
 
-       if(paramLast != eoi && *paramLast == '"') {
 
-         // quoted-string
 
-         ++paramLast;
 
-         for(; paramLast != eoi && *paramLast != '"'; ++paramLast);
 
-         if(paramLast != eoi) {
 
-           ++paramLast;
 
-         }
 
-         param = std::string(paramFirst, paramLast);
 
-         for(; paramLast != eoi && *paramLast != ';'; ++paramLast);
 
-       } else {
 
-         param = std::string(paramFirst, paramLast);
 
-       }
 
-     }
 
-     param = strip(param);
 
-     *out++ = param;
 
-     if(paramLast == eoi) {
 
-       break;
 
-     }
 
-     i = paramLast;
 
-     ++i;
 
-   }
 
- }
 
- } // namespace
 
- std::string getContentDispositionFilename(const std::string& header)
 
- {
 
-   std::string filename;
 
-   std::vector<std::string> params;
 
-   parseParam(std::back_inserter(params), header);
 
-   for(std::vector<std::string>::const_iterator i = params.begin(),
 
-         eoi = params.end(); i != eoi; ++i) {
 
-     const std::string& param = *i;
 
-     static const std::string keyName = "filename";
 
-     if(!startsWith(toLower(param), keyName) || param.size() == keyName.size()) {
 
-       continue;
 
-     }
 
-     std::string::const_iterator markeritr = param.begin()+keyName.size();
 
-     if(*markeritr == '*') {
 
-       // See RFC2231 Section4 and draft-reschke-rfc2231-in-http.
 
-       // Please note that this function doesn't do charset conversion
 
-       // except that if iso-8859-1 is specified, it is converted to
 
-       // utf-8.
 
-       ++markeritr;
 
-       for(; markeritr != param.end() && *markeritr == ' '; ++markeritr);
 
-       if(markeritr == param.end() || *markeritr != '=') {
 
-         continue;
 
-       }
 
-       std::pair<std::string, std::string> paramPair;
 
-       divide(paramPair, param, '=');
 
-       std::string value = paramPair.second;
 
-       std::vector<std::string> extValues;
 
-       split(value, std::back_inserter(extValues), "'", false, true);
 
-       if(extValues.size() != 3) {
 
-         continue;
 
-       }
 
-       bool bad = false;
 
-       const std::string& charset = extValues[0];
 
-       for(std::string::const_iterator j = charset.begin(), eoi = charset.end();
 
-           j != eoi; ++j) {
 
-         // Since we first split parameter by ', we can safely assume
 
-         // that ' is not included in charset.
 
-         if(!inRFC2978MIMECharset(*j)) {
 
-           bad = true;
 
-           break;
 
-         }
 
-       }
 
-       if(bad) {
 
-         continue;
 
-       }
 
-       bad = false;
 
-       value = extValues[2];
 
-       for(std::string::const_iterator j = value.begin(), eoi = value.end();
 
-           j != eoi; ++j){
 
-         if(*j == '%') {
 
-           if(j+1 != value.end() && isHexDigit(*(j+1)) &&
 
-              j+2 != value.end() && isHexDigit(*(j+2))) {
 
-             j += 2;
 
-           } else {
 
-             bad = true;
 
-             break;
 
-           }
 
-         } else {
 
-           if(*j == '*' || *j == '\'' || !inRFC2616HttpToken(*j)) {
 
-             bad = true;
 
-             break;
 
-           }
 
-         }
 
-       }
 
-       if(bad) {
 
-         continue;
 
-       }
 
-       value = percentDecode(value);
 
-       if(toLower(extValues[0]) == "iso-8859-1") {
 
-         value = iso8859ToUtf8(value);
 
-       }
 
-       if(!detectDirTraversal(value) &&
 
-          value.find(A2STR::SLASH_C) == std::string::npos) {
 
-         filename = value;
 
-       }
 
-       if(!filename.empty()) {
 
-         break;
 
-       }
 
-     } else {
 
-       for(; markeritr != param.end() && *markeritr == ' '; ++markeritr);
 
-       if(markeritr == param.end() || *markeritr != '=') {
 
-         continue;
 
-       }
 
-       std::pair<std::string, std::string> paramPair;
 
-       divide(paramPair, param, '=');
 
-       std::string value = paramPair.second;
 
-       if(value.empty()) {
 
-         continue;
 
-       }
 
-       std::string::iterator filenameLast;
 
-       if(*value.begin() == '\'' || *value.begin() == '"') {
 
-         char qc = *value.begin();
 
-         for(filenameLast = value.begin()+1;
 
-             filenameLast != value.end() && *filenameLast != qc;
 
-             ++filenameLast);
 
-       } else {
 
-         filenameLast = value.end();
 
-       }
 
-       static const std::string TRIMMED("\r\n\t '\"");
 
-       value = percentDecode(std::string(value.begin(), filenameLast));
 
-       value = strip(value, TRIMMED);
 
-       value.erase(std::remove(value.begin(), value.end(), '\\'), value.end());
 
-       if(!detectDirTraversal(value) &&
 
-          value.find(A2STR::SLASH_C) == std::string::npos) {
 
-         filename = value;
 
-       }
 
-       // continue because there is a chance we can find filename*=...
 
-     }
 
-   }
 
-   return filename;
 
- }
 
- std::string randomAlpha(size_t length, const RandomizerHandle& randomizer) {
 
-   static const char *random_chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
 
-   std::string str;
 
-   for(size_t i = 0; i < length; ++i) {
 
-     size_t index = randomizer->getRandomNumber(strlen(random_chars));
 
-     str += random_chars[index];
 
-   }
 
-   return str;
 
- }
 
- std::string toUpper(const std::string& src) {
 
-   std::string temp = src;
 
-   std::transform(temp.begin(), temp.end(), temp.begin(), ::toupper);
 
-   return temp;
 
- }
 
- std::string toLower(const std::string& src) {
 
-   std::string temp = src;
 
-   std::transform(temp.begin(), temp.end(), temp.begin(), ::tolower);
 
-   return temp;
 
- }
 
- void uppercase(std::string& s)
 
- {
 
-   std::transform(s.begin(), s.end(), s.begin(), ::toupper);
 
- }
 
- void lowercase(std::string& s)
 
- {
 
-   std::transform(s.begin(), s.end(), s.begin(), ::tolower);
 
- }
 
- bool isNumericHost(const std::string& name)
 
- {
 
-   struct addrinfo hints;
 
-   struct addrinfo* res;
 
-   memset(&hints, 0, sizeof(hints));
 
-   hints.ai_family = AF_UNSPEC;
 
-   hints.ai_flags = AI_NUMERICHOST;
 
-   if(getaddrinfo(name.c_str(), 0, &hints, &res)) {
 
-     return false;
 
-   }
 
-   freeaddrinfo(res);
 
-   return true;
 
- }
 
- void setGlobalSignalHandler(int sig, void (*handler)(int), int flags) {
 
- #ifdef HAVE_SIGACTION
 
-   struct sigaction sigact;
 
-   sigact.sa_handler = handler;
 
-   sigact.sa_flags = flags;
 
-   sigemptyset(&sigact.sa_mask);
 
-   sigaction(sig, &sigact, NULL);
 
- #else
 
-   signal(sig, handler);
 
- #endif // HAVE_SIGACTION
 
- }
 
- std::string getHomeDir()
 
- {
 
-   const char* p = getenv("HOME");
 
-   if(p) {
 
-     return p;
 
-   } else {
 
- #ifdef __MINGW32__
 
-     p = getenv("USERPROFILE");
 
-     if(p) {
 
-       return p;
 
-     } else {
 
-       p = getenv("HOMEDRIVE");
 
-       if(p) {
 
-         std::string homeDir = p;
 
-         p = getenv("HOMEPATH");
 
-         if(p) {
 
-           homeDir += p;
 
-           return homeDir;
 
-         }
 
-       }
 
-     }
 
- #endif
 
-     return A2STR::NIL;
 
-   }
 
- }
 
- int64_t getRealSize(const std::string& sizeWithUnit)
 
- {
 
-   std::string::size_type p = sizeWithUnit.find_first_of("KM");
 
-   std::string size;
 
-   int32_t mult = 1;
 
-   if(p == std::string::npos) {
 
-     size = sizeWithUnit;
 
-   } else {
 
-     if(sizeWithUnit[p] == 'K') {
 
-       mult = 1024;
 
-     } else if(sizeWithUnit[p] == 'M') {
 
-       mult = 1024*1024;
 
-     }
 
-     size = sizeWithUnit.substr(0, p);
 
-   }
 
-   int64_t v = parseLLInt(size);
 
-   if(v < 0) {
 
-     throw DL_ABORT_EX(fmt("Negative value detected: %s", sizeWithUnit.c_str()));
 
-   } else if(INT64_MAX/mult < v) {
 
-     throw DL_ABORT_EX(fmt(MSG_STRING_INTEGER_CONVERSION_FAILURE,
 
-                           "overflow/underflow"));
 
-   }
 
-   return v*mult;
 
- }
 
- std::string abbrevSize(int64_t size)
 
- {
 
-   if(size < 1024) {
 
-     return itos(size, true);
 
-   }
 
-   char units[] = { 'K', 'M' };
 
-   size_t numUnit = sizeof(units)/sizeof(char);
 
-   size_t i = 0;
 
-   int r = size&0x3ffu;
 
-   size >>= 10;
 
-   for(; i < numUnit-1 && size >= 1024; ++i) {
 
-     r = size&0x3ffu;
 
-     size >>= 10;
 
-   }
 
-   std::string result = itos(size, true);
 
-   result += A2STR::DOT_C;
 
-   result += itos(r*10/1024);
 
-   result += units[i];
 
-   result += "i";
 
-   return result;
 
- }
 
- void sleep(long seconds) {
 
- #ifdef HAVE_SLEEP
 
-   ::sleep(seconds);
 
- #elif defined(HAVE_USLEEP)
 
-   ::usleep(seconds * 1000000);
 
- #elif defined(HAVE_WINSOCK2_H)
 
-   ::Sleep(seconds * 1000);
 
- #else
 
- #error no sleep function is available (nanosleep?)
 
- #endif
 
- }
 
- void usleep(long microseconds) {
 
- #ifdef HAVE_USLEEP
 
-   ::usleep(microseconds);
 
- #elif defined(HAVE_WINSOCK2_H)
 
-   LARGE_INTEGER current, freq, end;
 
-   static enum {GET_FREQUENCY, GET_MICROSECONDS, SKIP_MICROSECONDS} state = GET_FREQUENCY;
 
-   if (state == GET_FREQUENCY) {
 
-     if (QueryPerformanceFrequency(&freq))
 
-       state = GET_MICROSECONDS;
 
-     else
 
-       state = SKIP_MICROSECONDS;
 
-   }
 
-         
 
-   long msec = microseconds / 1000;
 
-   microseconds %= 1000;    
 
-   if (state == GET_MICROSECONDS && microseconds) {
 
-     QueryPerformanceCounter(&end);
 
-     end.QuadPart += (freq.QuadPart * microseconds) / 1000000;
 
-     while (QueryPerformanceCounter(¤t) && (current.QuadPart <= end.QuadPart))
 
-       /* noop */ ;
 
-   }
 
-   if (msec)
 
-     Sleep(msec);
 
- #else
 
- #error no usleep function is available (nanosleep?)
 
- #endif
 
- }
 
- bool isNumber(const std::string& what)
 
- {
 
-   if(what.empty()) {
 
-     return false;
 
-   }
 
-   for(std::string::const_iterator i = what.begin(), eoi = what.end();
 
-       i != eoi; ++i) {
 
-     if(!isDigit(*i)) {
 
-       return false;
 
-     }
 
-   }
 
-   return true;
 
- }
 
- bool isLowercase(const std::string& what)
 
- {
 
-   if(what.empty()) {
 
-     return false;
 
-   }
 
-   for(uint32_t i = 0, eoi = what.size(); i < eoi; ++i) {
 
-     if(!('a' <= what[i] && what[i] <= 'z')) {
 
-       return false;
 
-     }
 
-   }
 
-   return true;
 
- }
 
- bool isUppercase(const std::string& what)
 
- {
 
-   if(what.empty()) {
 
-     return false;
 
-   }
 
-   for(uint32_t i = 0, eoi = what.size(); i < eoi; ++i) {
 
-     if(!('A' <= what[i] && what[i] <= 'Z')) {
 
-       return false;
 
-     }
 
-   }
 
-   return true;
 
- }
 
- unsigned int alphaToNum(const std::string& alphabets)
 
- {
 
-   if(alphabets.empty()) {
 
-     return 0;
 
-   }
 
-   char base;
 
-   if(islower(alphabets[0])) {
 
-     base = 'a';
 
-   } else {
 
-     base = 'A';
 
-   }
 
-   uint64_t num = 0;
 
-   for(size_t i = 0, eoi = alphabets.size(); i < eoi; ++i) {
 
-     unsigned int v = alphabets[i]-base;
 
-     num = num*26+v;
 
-     if(num > UINT32_MAX) {
 
-       return 0;
 
-     }
 
-   }
 
-   return num;
 
- }
 
- void mkdirs(const std::string& dirpath)
 
- {
 
-   File dir(dirpath);
 
-   if(!dir.mkdirs()) {
 
-     int errNum = errno;
 
-     if(!dir.isDir()) {
 
-       throw DL_ABORT_EX3
 
-         (errNum,
 
-          fmt(EX_MAKE_DIR, dir.getPath().c_str(),
 
-              safeStrerror(errNum).c_str()),
 
-          error_code::DIR_CREATE_ERROR);
 
-     }
 
-   }
 
- }
 
- void convertBitfield(BitfieldMan* dest, const BitfieldMan* src)
 
- {
 
-   size_t numBlock = dest->countBlock();
 
-   for(size_t index = 0; index < numBlock; ++index) {
 
-     if(src->isBitSetOffsetRange((uint64_t)index*dest->getBlockLength(),
 
-                                 dest->getBlockLength())) {
 
-       dest->setBit(index);
 
-     }
 
-   }
 
- }
 
- std::string toString(const BinaryStreamHandle& binaryStream)
 
- {
 
-   std::stringstream strm;
 
-   char data[2048];
 
-   while(1) {
 
-     int32_t dataLength = binaryStream->readData
 
-       (reinterpret_cast<unsigned char*>(data), sizeof(data), strm.tellp());
 
-     strm.write(data, dataLength);
 
-     if(dataLength == 0) {
 
-       break;
 
-     }
 
-   }
 
-   return strm.str();
 
- }
 
- #ifdef HAVE_POSIX_MEMALIGN
 
- /**
 
-  * In linux 2.6, alignment and size should be a multiple of 512.
 
-  */
 
- void* allocateAlignedMemory(size_t alignment, size_t size)
 
- {
 
-   void* buffer;
 
-   int res;
 
-   if((res = posix_memalign(&buffer, alignment, size)) != 0) {
 
-     throw FATAL_EXCEPTION
 
-       (fmt("Error in posix_memalign: %s",
 
-            util::safeStrerror(res).c_str()));
 
-   }
 
-   return buffer;
 
- }
 
- #endif // HAVE_POSIX_MEMALIGN
 
- std::pair<std::string, uint16_t>
 
- getNumericNameInfo(const struct sockaddr* sockaddr, socklen_t len)
 
- {
 
-   char host[NI_MAXHOST];
 
-   char service[NI_MAXSERV];
 
-   int s = getnameinfo(sockaddr, len, host, NI_MAXHOST, service, NI_MAXSERV,
 
-                       NI_NUMERICHOST|NI_NUMERICSERV);
 
-   if(s != 0) {
 
-     throw DL_ABORT_EX(fmt("Failed to get hostname and port. cause: %s",
 
-                           gai_strerror(s)));
 
-   }
 
-   return std::pair<std::string, uint16_t>(host, atoi(service)); // TODO
 
- }
 
- std::string htmlEscape(const std::string& src)
 
- {
 
-   std::string dest;
 
-   for(std::string::const_iterator i = src.begin(), eoi = src.end();
 
-       i != eoi; ++i) {
 
-     char ch = *i;
 
-     if(ch == '<') {
 
-       dest += "<";
 
-     } else if(ch == '>') {
 
-       dest += ">";
 
-     } else if(ch == '&') {
 
-       dest += "&";
 
-     } else if(ch == '\'') {
 
-       dest += "'";
 
-     } else if(ch == '"') {
 
-       dest += """;
 
-     } else {
 
-       dest += ch;
 
-     }
 
-   }
 
-   return dest;
 
- }
 
- std::map<size_t, std::string>::value_type
 
- parseIndexPath(const std::string& line)
 
- {
 
-   std::pair<std::string, std::string> p;
 
-   divide(p, line, '=');
 
-   size_t index = parseUInt(p.first);
 
-   if(p.second.empty()) {
 
-     throw DL_ABORT_EX(fmt("Path with index=%u is empty.",
 
-                           static_cast<unsigned int>(index)));
 
-   }
 
-   return std::map<size_t, std::string>::value_type(index, p.second);
 
- }
 
- std::map<size_t, std::string> createIndexPathMap(std::istream& i)
 
- {
 
-   std::map<size_t, std::string> indexPathMap;
 
-   std::string line;
 
-   while(getline(i, line)) {
 
-     indexPathMap.insert(indexPathMap.begin(), parseIndexPath(line));
 
-   }
 
-   return indexPathMap;
 
- }
 
- void generateRandomData(unsigned char* data, size_t length)
 
- {
 
- #ifdef HAVE_LIBGCRYPT
 
-   gcry_randomize(data, length, GCRY_STRONG_RANDOM);
 
- #elif HAVE_OPENSSL
 
-   if(RAND_bytes(data, length) != 1) {
 
-     for(size_t i = 0; i < length; ++i) {
 
-       data[i] = SimpleRandomizer::getInstance()->getRandomNumber(UINT8_MAX+1);
 
-     }
 
-   }
 
- #else
 
-   std::ifstream i("/dev/urandom", std::ios::binary);
 
-   i.read(reinterpret_cast<char*>(data), length);
 
- #endif // HAVE_OPENSSL
 
- }
 
- bool saveAs
 
- (const std::string& filename, const std::string& data, bool overwrite)
 
- {
 
-   if(!overwrite && File(filename).exists()) {
 
-     return false;
 
-   }
 
-   std::string tempFilename = strconcat(filename, "__temp");
 
-   {
 
-     BufferedFile fp(tempFilename, BufferedFile::WRITE);
 
-     if(!fp) {
 
-       return false;
 
-     }
 
-     if(fp.write(data.data(), data.size()) != data.size()) {
 
-       return false;
 
-     }
 
-     if(fp.close() == EOF) {
 
-       return false;
 
-     }
 
-   }
 
-   return File(tempFilename).renameTo(filename);
 
- }
 
- std::string applyDir(const std::string& dir, const std::string& relPath)
 
- {
 
-   if(dir.empty()) {
 
-     return strconcat(A2STR::DOT_C, A2STR::SLASH_C, relPath);
 
-   } else if(dir == A2STR::SLASH_C) {
 
-     return strconcat(A2STR::SLASH_C, relPath);
 
-   } else {
 
-     return strconcat(dir, A2STR::SLASH_C, relPath);
 
-   }
 
- }
 
- std::string fixTaintedBasename(const std::string& src)
 
- {
 
-   static std::string SLASH_REP = "%2F";
 
-   return escapePath(replace(src, A2STR::SLASH_C, SLASH_REP));
 
- }
 
- void generateRandomKey(unsigned char* key)
 
- {
 
- #ifdef ENABLE_MESSAGE_DIGEST
 
-   unsigned char bytes[40];
 
-   generateRandomData(bytes, sizeof(bytes));
 
-   message_digest::digest(key, 20, MessageDigest::sha1(), bytes, sizeof(bytes));
 
- #else // !ENABLE_MESSAGE_DIGEST
 
-   generateRandomData(key, 20);
 
- #endif // !ENABLE_MESSAGE_DIGEST
 
- }
 
- // Returns true is given numeric ipv4addr is in Private Address Space.
 
- //
 
- // From Section.3 RFC1918
 
- // 10.0.0.0        -   10.255.255.255  (10/8 prefix)
 
- // 172.16.0.0      -   172.31.255.255  (172.16/12 prefix)
 
- // 192.168.0.0     -   192.168.255.255 (192.168/16 prefix)
 
- bool inPrivateAddress(const std::string& ipv4addr)
 
- {
 
-   if(util::startsWith(ipv4addr, "10.") ||
 
-      util::startsWith(ipv4addr, "192.168.")) {
 
-     return true;
 
-   }
 
-   if(util::startsWith(ipv4addr, "172.")) {
 
-     for(int i = 16; i <= 31; ++i) {
 
-       if(util::startsWith(ipv4addr, "172."+util::itos(i)+".")) {
 
-         return true;
 
-       }
 
-     }
 
-   }
 
-   return false;
 
- }
 
- bool detectDirTraversal(const std::string& s)
 
- {
 
-   for(std::string::const_iterator i = s.begin(), eoi = s.end(); i != eoi; ++i) {
 
-     unsigned char c = *i;
 
-     if(in(c, 0x00u, 0x1fu) || c == 0x7fu) {
 
-       return true;
 
-     }
 
-   }
 
-   static std::string A2_DS = "./";
 
-   static std::string A2_DDS = "../";
 
-   static std::string A2_SD = "/.";
 
-   static std::string A2_SDD = "/..";
 
-   static std::string A2_SDDS = "/../";
 
-   static std::string A2_SDS = "/./";
 
-   static std::string A2_DD = "..";
 
-   return s == A2STR::DOT_C ||
 
-     s == A2_DD ||
 
-     util::startsWith(s, A2STR::SLASH_C) ||
 
-     util::startsWith(s, A2_DS) ||
 
-     util::startsWith(s, A2_DDS) ||
 
-     s.find(A2_SDDS) != std::string::npos ||
 
-     s.find(A2_SDS) != std::string::npos ||
 
-     util::endsWith(s, A2STR::SLASH_C) ||
 
-     util::endsWith(s, A2_SD) ||
 
-     util::endsWith(s, A2_SDD);
 
- }
 
- std::string escapePath(const std::string& s)
 
- {
 
-   // We don't escape '/' because we use it as a path separator.
 
- #ifdef __MINGW32__
 
-   static const char WIN_INVALID_PATH_CHARS[] =
 
-     { '"', '*', ':', '<', '>', '?', '\\', '|' };
 
- #endif // __MINGW32__
 
-   std::string d;
 
-   for(std::string::const_iterator i = s.begin(), eoi = s.end(); i != eoi; ++i) {
 
-     unsigned char c = *i;
 
-     if(in(c, 0x00u, 0x1fu) || c == 0x7fu
 
- #ifdef __MINGW32__
 
-        || std::find(vbegin(WIN_INVALID_PATH_CHARS),
 
-                     vend(WIN_INVALID_PATH_CHARS),
 
-                     c) != vend(WIN_INVALID_PATH_CHARS)
 
- #endif // __MINGW32__
 
-        ){
 
-       d += fmt("%%%02X", c);
 
-     } else {
 
-       d += *i;
 
-     }
 
-   }
 
-   return d;
 
- }
 
- bool getCidrPrefix(struct in_addr& in, const std::string& ip, int bits)
 
- {
 
-   struct in_addr t;
 
-   if(inet_aton(ip.c_str(), &t) == 0) {
 
-     return false;
 
-   }
 
-   int lastindex = bits/8;
 
-   if(lastindex < 4) {
 
-     char* p = reinterpret_cast<char*>(&t.s_addr);
 
-     const char* last = p+4;
 
-     p += lastindex;    
 
-     if(bits%8 != 0) {
 
-       *p &= bitfield::lastByteMask(bits);
 
-       ++p;
 
-     }
 
-     for(; p != last; ++p) {
 
-       *p &= 0;
 
-     }
 
-   }
 
-   in = t;
 
-   return true;
 
- }
 
- bool inSameCidrBlock(const std::string& ip1, const std::string& ip2, int bits)
 
- {
 
-   struct in_addr in1;
 
-   struct in_addr in2;
 
-   if(!getCidrPrefix(in1, ip1, bits) || !getCidrPrefix(in2, ip2, bits)) {
 
-     return false;
 
-   }
 
-   return in1.s_addr == in2.s_addr;
 
- }
 
- void removeMetalinkContentTypes(const SharedHandle<RequestGroup>& group)
 
- {
 
-   removeMetalinkContentTypes(group.get());
 
- }
 
- void removeMetalinkContentTypes(RequestGroup* group)
 
- {
 
-   for(std::vector<std::string>::const_iterator i =
 
- 	DownloadHandlerConstants::getMetalinkContentTypes().begin(),
 
-         eoi = DownloadHandlerConstants::getMetalinkContentTypes().end();
 
-       i != eoi; ++i) {
 
-     group->removeAcceptType(*i);
 
-   }
 
- }
 
- namespace {
 
- void executeHook
 
- (const std::string& command,
 
-  a2_gid_t gid,
 
-  size_t numFiles,
 
-  const std::string& firstFilename)
 
- {
 
-   const std::string gidStr = util::itos(gid);
 
-   const std::string numFilesStr = util::uitos(numFiles);
 
- #ifndef __MINGW32__
 
-   A2_LOG_INFO(fmt("Executing user command: %s %s %s %s",
 
-                   command.c_str(),
 
-                   gidStr.c_str(),
 
-                   numFilesStr.c_str(),
 
-                   firstFilename.c_str()));
 
-   pid_t cpid = fork();
 
-   if(cpid == -1) {
 
-     A2_LOG_ERROR("fork() failed. Cannot execute user command.");
 
-   } else if(cpid == 0) {
 
-     execl(command.c_str(),
 
-           command.c_str(),
 
-           gidStr.c_str(),
 
-           numFilesStr.c_str(),
 
-           firstFilename.c_str(),
 
-           reinterpret_cast<char*>(0));
 
-     perror(("Could not execute user command: "+command).c_str());
 
-     exit(EXIT_FAILURE);
 
-   }
 
- #else
 
-   PROCESS_INFORMATION pi;
 
-   STARTUPINFO si;
 
-   memset(&si, 0, sizeof (si));
 
-   si.cb = sizeof(STARTUPINFO);
 
-   memset(&pi, 0, sizeof (pi));
 
-   bool batch = util::endsWith(util::toLower(command), ".bat");
 
-   std::string cmdline;
 
-   std::string cmdexe;
 
-   if(batch) {
 
-     const char* p = getenv("windir");
 
-     if(p) {
 
-       cmdexe = std::string(p);
 
-       cmdexe += "\\system32\\cmd.exe";
 
-     } else {
 
-       A2_LOG_INFO("Failed to get windir environment variable."
 
-                   " Executing batch file will fail.");
 
-       // TODO Might be useless.
 
-       cmdexe = "cmd.exe";
 
-     }
 
-     cmdline += "/C \"";
 
-   }
 
-   strappend(cmdline, "\"", command, "\"");
 
-   strappend(cmdline, " ", gidStr, " ", numFilesStr, " \"", firstFilename, "\"");
 
-   if(batch) {
 
-     cmdline += "\"";
 
-   }
 
-   A2_LOG_INFO(fmt("Executing user command: %s", cmdline.c_str()));
 
-   DWORD rc = CreateProcess(
 
-                            batch ? cmdexe.c_str() : NULL,
 
-                            (LPSTR)cmdline.c_str(),
 
-                            NULL,
 
-                            NULL,
 
-                            true,
 
-                            NULL,
 
-                            NULL,
 
-                            0,
 
-                            &si,
 
-                            &pi);
 
-   if(!rc) {
 
-     A2_LOG_ERROR("CreateProcess() failed. Cannot execute user command.");
 
-   }
 
- #endif 
 
- }
 
- } // namespace
 
- void executeHookByOptName
 
- (const SharedHandle<RequestGroup>& group, const Option* option,
 
-  const std::string& opt)
 
- {
 
-   executeHookByOptName(group.get(), option, opt);
 
- }
 
- void executeHookByOptName
 
- (const RequestGroup* group, const Option* option, const std::string& opt)
 
- {
 
-   if(!option->blank(opt)) {
 
-     const SharedHandle<DownloadContext> dctx = group->getDownloadContext();
 
-     std::string firstFilename;
 
-     size_t numFiles = 0;
 
-     if(!group->inMemoryDownload()) {
 
-       SharedHandle<FileEntry> file = dctx->getFirstRequestedFileEntry();
 
-       if(file) {
 
-         firstFilename = file->getPath();
 
-       }
 
-       numFiles = dctx->countRequestedFileEntry();
 
-     }
 
-     executeHook(option->get(opt), group->getGID(), numFiles, firstFilename);
 
-   }
 
- }
 
- std::string createSafePath
 
- (const std::string& dir, const std::string& filename)
 
- {
 
-   return util::applyDir
 
-     (dir,
 
-      util::isUtf8(filename)?
 
-      util::fixTaintedBasename(filename):
 
-      util::escapePath(util::percentEncode(filename)));
 
- }
 
- std::string encodeNonUtf8(const std::string& s)
 
- {
 
-   return util::isUtf8(s)?s:util::percentEncode(s);
 
- }
 
- std::string makeString(const char* str)
 
- {
 
-   if(str) {
 
-     return str;
 
-   } else {
 
-     return A2STR::NIL;
 
-   }
 
- }
 
- std::string safeStrerror(int errNum)
 
- {
 
-   return makeString(strerror(errNum));
 
- }
 
- } // namespace util
 
- } // namespace aria2
 
 
  |