Util.h 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168
  1. /* <!-- copyright */
  2. /*
  3. * aria2 - The high speed download utility
  4. *
  5. * Copyright (C) 2006 Tatsuhiro Tsujikawa
  6. *
  7. * This program is free software; you can redistribute it and/or modify
  8. * it under the terms of the GNU General Public License as published by
  9. * the Free Software Foundation; either version 2 of the License, or
  10. * (at your option) any later version.
  11. *
  12. * This program is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15. * GNU General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU General Public License
  18. * along with this program; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. *
  21. * In addition, as a special exception, the copyright holders give
  22. * permission to link the code of portions of this program with the
  23. * OpenSSL library under certain conditions as described in each
  24. * individual source file, and distribute linked combinations
  25. * including the two.
  26. * You must obey the GNU General Public License in all respects
  27. * for all of the code used other than OpenSSL. If you modify
  28. * file(s) with this exception, you may extend this exception to your
  29. * version of the file(s), but you are not obligated to do so. If you
  30. * do not wish to do so, delete this exception statement from your
  31. * version. If you delete this exception statement from all source
  32. * files in the program, then also delete it here.
  33. */
  34. /* copyright --> */
  35. #ifndef _D_UTIL_H_
  36. #define _D_UTIL_H_
  37. #include "common.h"
  38. #include "a2time.h"
  39. #include "FileEntry.h"
  40. #include "IntSequence.h"
  41. #include <utility>
  42. #include <deque>
  43. #include <ostream>
  44. class Randomizer;
  45. typedef SharedHandle<Randomizer> RandomizerHandle;
  46. class BitfieldMan;
  47. #define STRTOLL(X) strtoll(X, (char**)NULL, 10)
  48. #define START_INDEX(OFFSET, PIECE_LENGTH) ((OFFSET)/(PIECE_LENGTH))
  49. #define END_INDEX(OFFSET, LENGTH, PIECE_LENGTH) (((OFFSET)+(LENGTH)-1)/(PIECE_LENGTH))
  50. class Util {
  51. public:
  52. static void split(pair<string, string>& hp, const string& src, char delim);
  53. static pair<string, string> split(const string& src, const string& delims);
  54. static string llitos(int64_t value, bool comma = false);
  55. static string ullitos(uint64_t value, bool comma = false);
  56. static string itos(int32_t value, bool comma = false);
  57. static string uitos(uint32_t value, bool comma = false);
  58. static string itos(int16_t value, bool comma = false);
  59. static string uitos(uint16_t value, bool comma = false);
  60. /**
  61. * Computes difference in micro-seconds between tv1 and tv2,
  62. * assuming tv1 is newer than tv2.
  63. * If tv1 is older than tv2, then this method returns 0.
  64. */
  65. static int64_t difftv(struct timeval tv1, struct timeval tv2);
  66. static int32_t difftvsec(struct timeval tv1, struct timeval tv2);
  67. /**
  68. * Take a string src which is a deliminated list and add its elements
  69. * into result. result is not cleared before conversion begins.
  70. */
  71. static void slice(Strings& result, const string& src, char delim, bool trim = false);
  72. static string trim(const string& src, const string& trimCharset = "\r\n\t ");
  73. static bool startsWith(const string& target, const string& part);
  74. static bool endsWith(const string& target, const string& part);
  75. static string replace(const string& target, const string& oldstr, const string& newstr);
  76. static string urlencode(const unsigned char* target, int32_t len);
  77. static string urlencode(const string& target)
  78. {
  79. return urlencode((const unsigned char*)target.c_str(), target.size());
  80. }
  81. static bool shouldUrlencode(const char c);
  82. static string urldecode(const string& target);
  83. static string torrentUrlencode(const unsigned char* target, int32_t len);
  84. static string toHex(const unsigned char* src, int32_t len);
  85. static FILE* openFile(const string& filename, const string& mode);
  86. static void fileCopy(const string& destFile, const string& src);
  87. static void rangedFileCopy(const string& destFile, const string& src, int64_t srcOffset, int64_t length);
  88. static bool isPowerOf(int32_t num, int32_t base);
  89. static string secfmt(int32_t sec);
  90. static int32_t expandBuffer(char** pbuf, int32_t curLength, int32_t newLength);
  91. static void unfoldRange(const string& src, Integers& range);
  92. static int32_t parseInt(const string& s, int32_t base = 10);
  93. static int64_t parseLLInt(const string& s, int32_t base = 10);
  94. static IntSequence parseIntRange(const string& src);
  95. // this function temporarily put here
  96. static string getContentDispositionFilename(const string& header);
  97. static int32_t countBit(uint32_t n);
  98. static string randomAlpha(int32_t length, const RandomizerHandle& randomizer);
  99. static string toUpper(const string& src);
  100. static string toLower(const string& src);
  101. static bool isNumbersAndDotsNotation(const string& name);
  102. static void setGlobalSignalHandler(int32_t signal, void (*handler)(int), int32_t flags);
  103. static void indexRange(int32_t& startIndex, int32_t& endIndex,
  104. int64_t offset,
  105. int32_t srcLength, int32_t destLength);
  106. static string getHomeDir();
  107. static int64_t getRealSize(const string& sizeWithUnit);
  108. static string abbrevSize(int64_t size);
  109. static time_t httpGMT(const string& httpTimeFormat);
  110. static void toStream(ostream& os, const FileEntries& entries);
  111. static void sleep(long seconds);
  112. static void usleep(long microseconds);
  113. static bool isNumber(const string& what);
  114. static bool isLowercase(const string& what);
  115. static bool isUppercase(const string& what);
  116. static int32_t alphaToNum(const string& alphabets);
  117. static void mkdirs(const string& dirpath);
  118. static void convertBitfield(BitfieldMan* dest, const BitfieldMan* src);
  119. };
  120. #endif // _D_UTIL_H_