added write support to File

svn-id: r4900

2 files changed, 95 insertions, 14 deletions

diff --git a/common/file.cpp b/common/file.cpp
index 7eace25c82..abc71d1ff3 100644
--- a/common/file.cpp
+++ b/common/file.cpp
@@ -23,7 +23,7 @@
 
 File::File() {
 	_handle = NULL;
-	_readFailed = false;
+	_ioFailed = false;
 	_encbyte = 0;
 }
 
@@ -38,7 +38,7 @@ bool File::open(const char *filename, int mode, byte encbyte) {
 		return false;
 	}
 
-	clearReadFailed();
+	clearIOFailed();
 
 	int32 i = 0, pos = 0;
 
@@ -70,8 +70,29 @@ bool File::open(const char *filename, int mode, byte encbyte) {
 			debug(2, "File %s not found", filename);
 			return false;
 		}
-	} else {
-		warning("Only read mode supported!");
+	}
+	else if (mode == 2) {
+		_handle = fopen(buf, "wb");
+		if (_handle == NULL) {
+			ptr = buf + pos;
+			do
+				*ptr++ = toupper(*ptr);
+			while (*ptr);
+			_handle = fopen(buf, "wb");
+		}
+		if (_handle == NULL) {
+			ptr = buf + pos;
+			do
+				*ptr++ = tolower(*ptr);
+			while (*ptr);
+			_handle = fopen(buf, "wb");
+		}
+		if (_handle == NULL) {
+			debug(2, "File %s not opened", filename);
+			return false;
+		}
+	}	else {
+		warning("Only read/write mode supported!");
 		return false;
 	}
 
@@ -88,12 +109,12 @@ bool File::isOpen() {
 	return _handle != NULL;
 }
 
-bool File::readFailed() {
-	return _readFailed != 0;
+bool File::ioFailed() {
+	return _ioFailed != 0;
 }
 
-void File::clearReadFailed() {
-	_readFailed = false;
+void File::clearIOFailed() {
+	_ioFailed = false;
 }
 
 bool File::eof() {
@@ -137,7 +158,7 @@ uint32 File::read(void *ptr, uint32 size) {
 
 	if ((uint32)fread(ptr2, 1, size, _handle) != size) {
 		clearerr(_handle);
-		_readFailed = true;
+		_ioFailed = true;
 	}
 
 	if (_encbyte != 0) {
@@ -155,7 +176,7 @@ byte File::readByte() {
 
 	if (fread(&b, 1, 1, _handle) != 1) {
 		clearerr(_handle);
-		_readFailed = true;
+		_ioFailed = true;
 	}
 	return b ^ _encbyte;
 }
@@ -183,3 +204,58 @@ uint32 File::readDwordBE() {
 	uint32 a = readWordBE();
 	return (b << 16) | a;
 }
+
+uint32 File::write(void *ptr, uint32 size) {
+	byte *ptr2 = (byte *)ptr;
+
+	if (_handle == NULL) {
+		error("File is not open!");
+		return 0;
+	}
+
+	if (size == 0)
+		return 0;
+
+	if (_encbyte != 0) {
+		uint32 t_size = size;
+		do {
+			*ptr2++ ^= _encbyte;
+		} while (--t_size);
+	}
+
+	if ((uint32)fwrite(ptr2, 1, size, _handle) != size) {
+		clearerr(_handle);
+		_ioFailed = true;
+	}
+
+	return size;
+}
+
+void File::writeByte(byte value) {
+	value ^= _encbyte;
+
+	if (fwrite(&value, 1, 1, _handle) != 1) {
+		clearerr(_handle);
+		_ioFailed = true;
+	}
+}
+
+void File::writeWordLE(uint16 value) {
+	writeByte((byte)(value & 0xff));
+	writeByte((byte)(value >> 8));
+}
+
+void File::writeDwordLE(uint32 value) {
+	writeWordLE((uint16)(value & 0xffff));
+	writeWordLE((uint16)(value >> 16));
+}
+
+void File::writeWordBE(uint16 value) {
+	writeByte((byte)(value >> 8));
+	writeByte((byte)(value & 0xff));
+}
+
+void File::writeDwordBE(uint32 value) {
+	writeWordBE((uint16)(value >> 16));
+	writeWordBE((uint16)(value & 0xffff));
+}
diff --git a/common/file.h b/common/file.h
index 6820e77255..807dcfab0a 100644
--- a/common/file.h
+++ b/common/file.h
@@ -31,7 +31,7 @@ class File {
 private:
 
 	FILE * _handle;
-	bool _readFailed;
+	bool _ioFailed;
 	byte _encbyte;
 
 public:
@@ -41,8 +41,8 @@ public:
 	bool open(const char *filename, int mode = 1, byte encbyte = 0);
 	void close();
 	bool isOpen();
-	bool readFailed();
-	void clearReadFailed();
+	bool ioFailed();
+	void clearIOFailed();
 	bool eof();
 	uint32 pos();
 	void seek(uint32 offs, int whence);
@@ -52,7 +52,12 @@ public:
 	uint32 readDwordLE();
 	uint16 readWordBE();
 	uint32 readDwordBE();
-
+	uint32 write(void *ptr, uint32 size);
+	void writeByte(byte value);
+	void writeWordLE(uint16 value);
+	void writeDwordLE(uint32 value);
+	void writeWordBE(uint16 value);
+	void writeDwordBE(uint32 value);
 };
 
 #endif