Box Backup 0.09 with a few tweeks

1 files changed, 112 insertions, 0 deletions

diff --git a/lib/common/Guards.h b/lib/common/Guards.h
new file mode 100755
index 00000000..6efc5614
--- /dev/null
+++ b/lib/common/Guards.h
@@ -0,0 +1,112 @@
+// --------------------------------------------------------------------------
+//
+// File
+//		Name:    Guards.h
+//		Purpose: Classes which ensure things are closed/deleted properly when
+//				 going out of scope. Easy exception proof code, etc
+//		Created: 2003/07/12
+//
+// --------------------------------------------------------------------------
+
+#ifndef GUARDS__H
+#define GUARDS__H
+
+#include <fcntl.h>
+#include <stdlib.h>
+#include <sys/stat.h>
+#include <unistd.h>
+#include <new>
+
+#include "CommonException.h"
+
+#include "MemLeakFindOn.h"
+
+template <int flags = O_RDONLY, int mode = (S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH)>
+class FileHandleGuard
+{
+public:
+	FileHandleGuard(const char *filename)
+		: mOSFileHandle(::open(filename, flags, mode))
+	{
+		if(mOSFileHandle < 0)
+		{
+			THROW_EXCEPTION(CommonException, OSFileOpenError)
+		}
+	}
+	
+	~FileHandleGuard()
+	{
+		if(mOSFileHandle >= 0)
+		{
+			Close();
+		}
+	}
+	
+	void Close()
+	{
+		if(mOSFileHandle < 0)
+		{
+			THROW_EXCEPTION(CommonException, FileAlreadyClosed)
+		}
+		if(::close(mOSFileHandle) != 0)
+		{
+			THROW_EXCEPTION(CommonException, OSFileCloseError)
+		}
+		mOSFileHandle = -1;
+	}
+	
+	operator int() const
+	{
+		return mOSFileHandle;
+	}
+
+private:
+	int mOSFileHandle;
+};
+
+template<typename type>
+class MemoryBlockGuard
+{
+public:
+	MemoryBlockGuard(int BlockSize)
+		: mpBlock(::malloc(BlockSize))
+	{
+		if(mpBlock == 0)
+		{
+			throw std::bad_alloc();
+		}
+	}
+	
+	~MemoryBlockGuard()
+	{
+		free(mpBlock);
+	}
+	
+	operator type() const
+	{
+		return (type)mpBlock;
+	}
+	
+	type GetPtr() const
+	{
+		return (type)mpBlock;
+	}
+	
+	void Resize(int NewSize)
+	{
+		void *ptrn = ::realloc(mpBlock, NewSize);
+		if(ptrn == 0)
+		{
+			throw std::bad_alloc();
+		}
+		mpBlock = ptrn;
+	}
+	
+private:
+	void *mpBlock;
+};
+
+#include "MemLeakFindOff.h"
+
+#endif // GUARDS__H
+