make dabOutput more object-oriented

1 files changed, 111 insertions, 0 deletions

diff --git a/src/dabOutput/dabOutputFile.cpp b/src/dabOutput/dabOutputFile.cpp
new file mode 100644
index 0000000..349a308
--- /dev/null
+++ b/src/dabOutput/dabOutputFile.cpp
@@ -0,0 +1,111 @@
+#include <cstdio>
+#include <cstring>
+#include <fcntl.h>
+#include <limits.h>
+#include "dabOutput.h"
+
+int DabOutputFile::Open(const char* filename)
+{
+    char* token = strchr((char*)filename, '?');
+    if (token != NULL) {
+        *(token++) = 0;
+        char* nextPair;
+        char* key;
+        char* value;
+        // Go through all the &stuff=foo pairs
+        // Only the key "type" is supported
+        do {
+            nextPair = strchr(token, '&');
+            if (nextPair != NULL) {
+                *nextPair = 0;
+            }
+            key = token;
+            value = strchr(token, '=');
+            if (value != NULL) {
+                *(value++) = 0;
+                if (strcmp(key, "type") == 0) {
+                    if (strcmp(value, "raw") == 0) {
+                        this->type_ = ETI_FILE_TYPE_RAW;
+                        break;
+                    } else if (strcmp(value, "framed") == 0) {
+                        this->type_ = ETI_FILE_TYPE_FRAMED;
+                        break;
+                    } else if (strcmp(value, "streamed") == 0) {
+                        this->type_ = ETI_FILE_TYPE_STREAMED;
+                        break;
+                    } else {
+                        etiLog.printHeader(TcpLog::ERR,
+                                "File type '%s' is not supported.\n", value);
+                        return -1;
+                    }
+                }
+                else {
+                    etiLog.printHeader(TcpLog::WARNING, "Parameter '%s' unknown\n", key);
+                }
+            }
+        } while (nextPair != NULL);
+    }
+
+    this->file_ = open(filename, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY, 0666);
+    if (this->file_ == -1) {
+        perror(filename);
+        return -1;
+    }
+    return 0;
+}
+
+int DabOutputFile::Write(void* buffer, int size)
+{
+    uint8_t padding[6144];
+    ++nbFrames_;
+
+    switch (this->type_) {
+    case ETI_FILE_TYPE_FRAMED:
+        // Writting nb of frames at beginning of file
+        if (lseek(this->file_, 0, SEEK_SET) == -1) goto FILE_WRITE_ERROR;
+        if (write(this->file_, &this->nbFrames_, 4) == -1) goto FILE_WRITE_ERROR;
+
+        // Writting nb frame length at end of file
+        if (lseek(this->file_, 0, SEEK_END) == -1) goto FILE_WRITE_ERROR;
+        if (write(this->file_, &size, 2) == -1) goto FILE_WRITE_ERROR;
+
+        // Appending data
+        if (write(this->file_, buffer, size) == -1) goto FILE_WRITE_ERROR;
+        break;
+    case ETI_FILE_TYPE_STREAMED:
+        // Writting nb frame length at end of file
+        if (write(this->file_, &size, 2) == -1) goto FILE_WRITE_ERROR;
+
+        // Appending data
+        if (write(this->file_, buffer, size) == -1) goto FILE_WRITE_ERROR;
+        break;
+    case ETI_FILE_TYPE_RAW:
+        // Appending data
+        if (write(this->file_, buffer, size) == -1) goto FILE_WRITE_ERROR;
+
+        // Appending padding
+        memset(padding, 0x55, 6144 - size);
+        if (write(this->file_, padding, 6144 - size) == -1) goto FILE_WRITE_ERROR;
+        break;
+    default:
+        etiLog.printHeader(TcpLog::ERR, "File type is not supported.\n");
+        return -1;
+    }
+
+    return size;
+
+FILE_WRITE_ERROR:
+    perror("Error while writting to file");
+    return -1;
+}
+
+int DabOutputFile::Close()
+{
+    if (close(this->file_) == 0) {
+        this->file_ = -1;
+        return 0;
+    }
+    perror("Can't close file");
+    return -1;
+}
+