96 lines
2.6 KiB
C++
96 lines
2.6 KiB
C++
#ifndef _UTIL_H_
|
|
#define _UTIL_H_ 1
|
|
|
|
#include <pthread.h>
|
|
#include <errno.h>
|
|
#include <openssl/sha.h>
|
|
|
|
#include "uint256.h"
|
|
|
|
#define loop for (;;)
|
|
#define BEGIN(a) ((char*)&(a))
|
|
#define END(a) ((char*)&((&(a))[1]))
|
|
#define UBEGIN(a) ((unsigned char*)&(a))
|
|
#define UEND(a) ((unsigned char*)&((&(a))[1]))
|
|
#define ARRAYLEN(array) (sizeof(array)/sizeof((array)[0]))
|
|
|
|
#define WSAGetLastError() errno
|
|
#define WSAEINVAL EINVAL
|
|
#define WSAEALREADY EALREADY
|
|
#define WSAEWOULDBLOCK EWOULDBLOCK
|
|
#define WSAEMSGSIZE EMSGSIZE
|
|
#define WSAEINTR EINTR
|
|
#define WSAEINPROGRESS EINPROGRESS
|
|
#define WSAEADDRINUSE EADDRINUSE
|
|
#define WSAENOTSOCK EBADF
|
|
#define INVALID_SOCKET (SOCKET)(~0)
|
|
#define SOCKET_ERROR -1
|
|
|
|
inline int myclosesocket(SOCKET& hSocket)
|
|
{
|
|
if (hSocket == INVALID_SOCKET)
|
|
return WSAENOTSOCK;
|
|
#ifdef WIN32
|
|
int ret = closesocket(hSocket);
|
|
#else
|
|
int ret = close(hSocket);
|
|
#endif
|
|
hSocket = INVALID_SOCKET;
|
|
return ret;
|
|
}
|
|
#define closesocket(s) myclosesocket(s)
|
|
|
|
|
|
// Wrapper to automatically initialize mutex
|
|
class CCriticalSection
|
|
{
|
|
protected:
|
|
pthread_rwlock_t mutex;
|
|
public:
|
|
explicit CCriticalSection() { pthread_rwlock_init(&mutex, NULL); }
|
|
~CCriticalSection() { pthread_rwlock_destroy(&mutex); }
|
|
void Enter(bool fShared = false) {
|
|
if (fShared) {
|
|
pthread_rwlock_rdlock(&mutex);
|
|
} else {
|
|
pthread_rwlock_wrlock(&mutex);
|
|
}
|
|
}
|
|
void Leave() { pthread_rwlock_unlock(&mutex); }
|
|
};
|
|
|
|
// Automatically leave critical section when leaving block, needed for exception safety
|
|
class CCriticalBlock
|
|
{
|
|
protected:
|
|
CCriticalSection* pcs;
|
|
public:
|
|
CCriticalBlock(CCriticalSection& cs, bool fShared = false) : pcs(&cs) { pcs->Enter(fShared); }
|
|
operator bool() const { return true; }
|
|
~CCriticalBlock() { pcs->Leave(); }
|
|
};
|
|
|
|
#define CRITICAL_BLOCK(cs) \
|
|
if (CCriticalBlock criticalblock = CCriticalBlock(cs))
|
|
|
|
#define SHARED_CRITICAL_BLOCK(cs) \
|
|
if (CCriticalBlock criticalblock = CCriticalBlock(cs, true))
|
|
|
|
template<typename T1> inline uint256 Hash(const T1 pbegin, const T1 pend)
|
|
{
|
|
static unsigned char pblank[1];
|
|
uint256 hash1;
|
|
SHA256((pbegin == pend ? pblank : (unsigned char*)&pbegin[0]), (pend - pbegin) * sizeof(pbegin[0]), (unsigned char*)&hash1);
|
|
uint256 hash2;
|
|
SHA256((unsigned char*)&hash1, sizeof(hash1), (unsigned char*)&hash2);
|
|
return hash2;
|
|
}
|
|
|
|
void static inline Sleep(int nMilliSec) {
|
|
struct timespec wa;
|
|
wa.tv_sec = nMilliSec/1000;
|
|
wa.tv_nsec = (nMilliSec % 1000) * 1000000;
|
|
nanosleep(&wa, NULL);
|
|
}
|
|
|
|
#endif
|