#define _MOOF_LOG_H_
/**
- * @file log.h
+ * \file Log.h
* Functions related to logging the process.
* The logging functions are logError(), logWarning(), and logInfo(),
* listed from most critical to least critical.
/**
- * Macro which tests an assertion and issues an logError() and exits if
- * false.
- * @param X test to perform
+ * Macro which tests an assertion and issues a logError() and exits if the
+ * assertion is false.
+ * \param X test to perform
*/
#undef ASSERT
namespace Mf {
+/**
+ * A class for handling a log priority. There are two ways to log
+ * messages: by treating a log object as a function whose parameters are
+ * printed with default spacing, or by treating a log object as an output
+ * stream. Either way, any object can be printed to the log as long as
+ * there is an override for the ostream insertion operator.
+ */
class Log
{
public:
+ /**
+ * A type for the level or priority of a log message.
+ */
enum Level
{
NONE = 0, ///< Disable all logging.
INFO = 3, ///< Log everything.
};
+
+ /**
+ * Set the lowest-priority log message that will be outputted to the
+ * log. Any logging with a lower priority will be ignored.
+ * \param level The log level.
+ */
static void setLevel(Level level);
- static Level getLevel();
+ /**
+ * Get the current lowest-priority log level. If unchanged, the
+ * default level is INFO.
+ * \return The log level.
+ */
+ static Level level();
- Log(Level level, const char* type) :
+
+ /**
+ * Construct a log with a certain priority and prefix string.
+ * \param level The log level.
+ * \param prefix The string printed before each log message.
+ */
+ Log(Level level, const char* prefix) :
mLevel(level),
- mType(type) /* only pass literal strings */ {}
+ mPrefix(prefix) /* only pass literal strings */ {}
- template <class T>
- void operator () (const T& item)
+ template <class A>
+ void operator () (const A& a)
{
- *this << item << std::endl;
+ *this << a << std::endl;
}
+ template <class A, class B>
+ void operator () (const A& a, const B& b)
+ {
+ *this << a << " " << b << std::endl;
+ }
+
+ template <class A, class B, class C>
+ void operator () (const A& a, const B& b, const C& c)
+ {
+ *this << a << " " << b << " " << c << std::endl;
+ }
+
+ template <class A, class B, class C, class D>
+ void operator () (const A& a, const B& b, const C& c, const D& d)
+ {
+ *this << a << " " << b << " " << c << " " << d << std::endl;
+ }
+
+ template <class A, class B, class C, class D, class E>
+ void operator () (const A& a,
+ const B& b,
+ const C& c,
+ const D& d,
+ const E& e)
+ {
+ *this << a << " "
+ << b << " "
+ << c << " "
+ << d << " "
+ << e << std::endl;
+ }
+
+
private:
template <class T> friend std::ostream& operator << (Log&, const T&);
static Level gLevel;
Level mLevel;
- const char* mType;
+ const char* mPrefix;
};
inline std::ostream& operator << (Log& logObj, const T& item)
{
if (Log::gLevel < logObj.mLevel) return nullLog;
- return log << logObj.mType << item;
+ return log << logObj.mPrefix << item;
}
**************************************************************************/
/**
- * \file Network.hh
+ * \file Socket.hh
* Network-related classes, including a reinterpreted sockets API.
*/
#include <string>
#include <vector>
+#if HAVE_FCNTL_H
+#include <fcntl.h>
+#else
+#error No alternative to fcntl implemented yet.
+#endif
+
#if defined(_WIN32)
#include <winsock2.h>
#include <ws2tcpip.h>
#include <wspiapi.h>
+#define SHUT_RD SD_RECEIVE
+#define SHUT_WR SD_SEND
+#define SHUT_RDWR SD_BOTH
#else
#include <arpa/inet.h>
#include <netdb.h>
#include <sys/socket.h>
#include <sys/types.h>
-#if HAVE_FCNTL_H
-#include <fcntl.h>
-#else
-#include <sys/ioctl.h>
-#endif
#endif
#include <Moof/Log.hh>
#include <Moof/Thread.hh>
-#ifndef SO_NONBLOCK
-#define SO_NONBLOCK 1024
+#ifndef AI_ADDRCONFIG
+#define AI_ADDRCONFIG 0
+#endif
+
+#ifndef AI_V4MAPPED
+#define AI_V4MAPPED 0
#endif
template <class T>
int set(int option, const T& value, int level = SOL_SOCKET)
{
+#if defined(_WIN32)
+ return setsockopt(mImpl.fd,
+ level,
+ option,
+ reinterpret_cast<const char*>(&value),
+ sizeof(value));
+#else
return setsockopt(mImpl.fd, level, option, &value, sizeof(value));
+#endif
}
/**
char str[256] = {'\0'};
socklen_t size = sizeof(str);
+#if defined(_WIN32)
+ int result = getsockopt(mImpl.fd,
+ level,
+ option,
+ reinterpret_cast<char*>(&str),
+ &size);
+#else
int result = getsockopt(mImpl.fd, level, option, &str, &size);
+#endif
value.assign(str, size);
return result;
}
+ /**
+ * Set the socket IO mode to either blocking or non-blocking.
+ * \param isBlocking True if the socket blocks, false otherwise.
+ */
void setBlocking(bool isBlocking)
{
- int value = isBlocking;
#ifdef HAVE_FCNTL
int flags = fcntl(mImpl.fd, F_GETFL);
- fcntl(mImpl.fd, F_SETFL, flags | (value ? O_NONBLOCK : 0));
-#else
- ioctl(mImpl.fd, FIONBIO, value);
+ flags = isBlocking ? (flags & ~O_NONBLOCK) : (flags | O_NONBLOCK);
+ fcntl(mImpl.fd, F_SETFL, flags);
+#elif defined(_WIN32)
+ u_long value = isBlocking;
+ ioctlsocket(mImpl.fd, FIONBIO, &value);
#endif
}
+ /**
+ * Get whether or not the socket is blocking or non-blocking. If the
+ * IO mode can't be determined, this method will assume the socket is
+ * a blocking socket.
+ * \return True if the socket blocks, false otherwise.
+ */
bool isBlocking() const
{
#ifdef HAVE_FCNTL
int flags = fcntl(mImpl.fd, F_GETFL);
- return flags & O_NONBLOCK;
-#else
- int value;
- ioctl(mImpl.fd, FIONBIO, &value);
- return value;
+ return !(flags & O_NONBLOCK);
#endif
+ return true;
}
*/
ssize_t write(const void* bytes, size_t size, int flags = 0)
{
+#if defined(_WIN32)
+ return send(mImpl.fd,
+ reinterpret_cast<const char *>(bytes), size,
+ flags);
+#else
return send(mImpl.fd, bytes, size, flags);
+#endif
}
/**
const SocketAddress& address,
int flags = 0)
{
+#if defined(_WIN32)
+ return sendto(mImpl.fd,
+ reinterpret_cast<const char*>(bytes), size,
+ flags,
+ address.address(), address.size());
+#else
return sendto(mImpl.fd, bytes, size, flags,
address.address(), address.size());
+#endif
}
/**
*/
ssize_t read(void* bytes, size_t size, int flags = 0)
{
+#if defined(_WIN32)
+ ssize_t result = recv(mImpl.fd,
+ reinterpret_cast<char*>(bytes), size,
+ flags);
+#else
ssize_t result = recv(mImpl.fd, bytes, size, flags);
+#endif
if (result == 0) mImpl.isConnected = false;
return result;
}
} addr;
socklen_t length = sizeof(addr);
+#if defined(_WIN32)
+ ssize_t result = recvfrom(mImpl.fd,
+ reinterpret_cast<char*>(bytes), size,
+ flags,
+ &addr.sa, &length);
+#else
ssize_t result = recvfrom(mImpl.fd, bytes, size, flags,
- &addr.sa, &length);
+ &addr.sa, &length);
+#endif
if (result != -1)
{
address = SocketAddress(&addr.sa, length, mImpl.address.type());
projects / chaz / yoink / commitdiff
