1 // Copyright (c) 2009-2010 Satoshi Nakamoto
2 // Copyright (c) 2009-2014 The Bitcoin Core developers
3 // Distributed under the MIT software license, see the accompanying
4 // file COPYING or http://www.opensource.org/licenses/mit-license.php.
7 * Server/client environment: argument handling, config file parsing,
8 * logging, thread wrappers
10 #ifndef BITCOIN_UTIL_H
11 #define BITCOIN_UTIL_H
13 #if defined(HAVE_CONFIG_H)
14 #include "config/bitcoin-config.h"
18 #include "tinyformat.h"
27 #include <boost/filesystem/path.hpp>
28 #include <boost/signals2/signal.hpp>
29 #include <boost/thread/exceptions.hpp>
31 /** Signals for translation. */
32 class CTranslationInterface
35 /** Translate a message to the native language of the user. */
36 boost::signals2::signal
<std::string (const char* psz
)> Translate
;
39 extern std::map
<std::string
, std::string
> mapArgs
;
40 extern std::map
<std::string
, std::vector
<std::string
> > mapMultiArgs
;
42 extern bool fPrintToConsole
;
43 extern bool fPrintToDebugLog
;
45 extern std::string strMiscWarning
;
46 extern bool fLogTimestamps
;
48 extern volatile bool fReopenDebugLog
;
49 extern CTranslationInterface translationInterface
;
52 * Translation function: Call Translate signal on UI interface, which returns a boost::optional result.
53 * If no translation slot is registered, nothing is returned, and simply return the input.
55 inline std::string
_(const char* psz
)
57 boost::optional
<std::string
> rv
= translationInterface
.Translate(psz
);
58 return rv
? (*rv
) : psz
;
61 void SetupEnvironment();
62 bool SetupNetworking();
64 /** Return true if log accepts specified category */
65 bool LogAcceptCategory(const char* category
);
66 /** Send a string to the log output */
67 int LogPrintStr(const std::string
&str
);
69 #define LogPrintf(...) LogPrint(NULL, __VA_ARGS__)
72 * When we switch to C++11, this can be switched to variadic templates instead
73 * of this macro-based construction (see tinyformat.h).
75 #define MAKE_ERROR_AND_LOG_FUNC(n) \
76 /** Print to debug.log if -debug=category switch is given OR category is NULL. */ \
77 template<TINYFORMAT_ARGTYPES(n)> \
78 static inline int LogPrint(const char* category, const char* format, TINYFORMAT_VARARGS(n)) \
80 if(!LogAcceptCategory(category)) return 0; \
81 return LogPrintStr(tfm::format(format, TINYFORMAT_PASSARGS(n))); \
83 /** Log error and return false */ \
84 template<TINYFORMAT_ARGTYPES(n)> \
85 static inline bool error(const char* format, TINYFORMAT_VARARGS(n)) \
87 LogPrintStr("ERROR: " + tfm::format(format, TINYFORMAT_PASSARGS(n)) + "\n"); \
91 TINYFORMAT_FOREACH_ARGNUM(MAKE_ERROR_AND_LOG_FUNC
)
94 * Zero-arg versions of logging and error, these are not covered by
95 * TINYFORMAT_FOREACH_ARGNUM
97 static inline int LogPrint(const char* category
, const char* format
)
99 if(!LogAcceptCategory(category
)) return 0;
100 return LogPrintStr(format
);
102 static inline bool error(const char* format
)
104 LogPrintStr(std::string("ERROR: ") + format
+ "\n");
108 void PrintExceptionContinue(const std::exception
*pex
, const char* pszThread
);
109 void ParseParameters(int argc
, const char*const argv
[]);
110 void FileCommit(FILE *fileout
);
111 bool TruncateFile(FILE *file
, unsigned int length
);
112 int RaiseFileDescriptorLimit(int nMinFD
);
113 void AllocateFileRange(FILE *file
, unsigned int offset
, unsigned int length
);
114 bool RenameOver(boost::filesystem::path src
, boost::filesystem::path dest
);
115 bool TryCreateDirectory(const boost::filesystem::path
& p
);
116 boost::filesystem::path
GetDefaultDataDir();
117 const boost::filesystem::path
&GetDataDir(bool fNetSpecific
= true);
118 void ClearDatadirCache();
119 boost::filesystem::path
GetConfigFile();
121 boost::filesystem::path
GetPidFile();
122 void CreatePidFile(const boost::filesystem::path
&path
, pid_t pid
);
124 void ReadConfigFile(std::map
<std::string
, std::string
>& mapSettingsRet
, std::map
<std::string
, std::vector
<std::string
> >& mapMultiSettingsRet
);
126 boost::filesystem::path
GetSpecialFolderPath(int nFolder
, bool fCreate
= true);
128 boost::filesystem::path
GetTempPath();
130 void ShrinkDebugFile();
131 void runCommand(const std::string
& strCommand
);
133 inline bool IsSwitchChar(char c
)
136 return c
== '-' || c
== '/';
143 * Return string argument or default value
145 * @param strArg Argument to get (e.g. "-foo")
146 * @param default (e.g. "1")
147 * @return command-line argument or default value
149 std::string
GetArg(const std::string
& strArg
, const std::string
& strDefault
);
152 * Return integer argument or default value
154 * @param strArg Argument to get (e.g. "-foo")
155 * @param default (e.g. 1)
156 * @return command-line argument (0 if invalid number) or default value
158 int64_t GetArg(const std::string
& strArg
, int64_t nDefault
);
161 * Return boolean argument or default value
163 * @param strArg Argument to get (e.g. "-foo")
164 * @param default (true or false)
165 * @return command-line argument or default value
167 bool GetBoolArg(const std::string
& strArg
, bool fDefault
);
170 * Set an argument if it doesn't already have a value
172 * @param strArg Argument to set (e.g. "-foo")
173 * @param strValue Value (e.g. "1")
174 * @return true if argument gets set, false if it already had a value
176 bool SoftSetArg(const std::string
& strArg
, const std::string
& strValue
);
179 * Set a boolean argument if it doesn't already have a value
181 * @param strArg Argument to set (e.g. "-foo")
182 * @param fValue Value (e.g. false)
183 * @return true if argument gets set, false if it already had a value
185 bool SoftSetBoolArg(const std::string
& strArg
, bool fValue
);
188 * Format a string to be used as group of options in help messages
190 * @param message Group name (e.g. "RPC server options:")
191 * @return the formatted string
193 std::string
HelpMessageGroup(const std::string
& message
);
196 * Format a string to be used as option description in help messages
198 * @param option Option message (e.g. "-rpcuser=<user>")
199 * @param message Option description (e.g. "Username for JSON-RPC connections")
200 * @return the formatted string
202 std::string
HelpMessageOpt(const std::string
& option
, const std::string
& message
);
205 * Return the number of physical cores available on the current system.
206 * @note This does not count virtual cores, such as those provided by HyperThreading
207 * when boost is newer than 1.56.
211 void SetThreadPriority(int nPriority
);
212 void RenameThread(const char* name
);
215 * .. and a wrapper that just calls func once
217 template <typename Callable
> void TraceThread(const char* name
, Callable func
)
219 std::string s
= strprintf("bitcoin-%s", name
);
220 RenameThread(s
.c_str());
223 LogPrintf("%s thread start\n", name
);
225 LogPrintf("%s thread exit\n", name
);
227 catch (const boost::thread_interrupted
&)
229 LogPrintf("%s thread interrupt\n", name
);
232 catch (const std::exception
& e
) {
233 PrintExceptionContinue(&e
, name
);
237 PrintExceptionContinue(NULL
, name
);
242 #endif // BITCOIN_UTIL_H