git.cworth.org Git - apitrace/blob - common/trace_file_read.cpp

   1 /**************************************************************************
   2  *
   3  * Copyright 2011 Jose Fonseca
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a copy
   7  * of this software and associated documentation files (the "Software"), to deal
   8  * in the Software without restriction, including without limitation the rights
   9  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10  * copies of the Software, and to permit persons to whom the Software is
  11  * furnished to do so, subject to the following conditions:
  12  *
  13  * The above copyright notice and this permission notice shall be included in
  14  * all copies or substantial portions of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  22  * THE SOFTWARE.
  23  *
  24  **************************************************************************/
  25
  26
  27 #include <fstream>
  28
  29 #include "os.hpp"
  30 #include "trace_file.hpp"
  31
  32
  33 using namespace trace;
  34
  35
  36 File *
  37 File::createForRead(const char *filename)
  38 {
  39     std::ifstream stream(filename, std::ifstream::binary | std::ifstream::in);
  40     if (!stream.is_open()) {
  41         os::log("error: failed to open %s\n", filename);
  42         return NULL;
  43     }
  44     unsigned char byte1, byte2;
  45     stream >> byte1;
  46     stream >> byte2;
  47     stream.close();
  48
  49     File *file;
  50     if (byte1 == SNAPPY_BYTE1 && byte2 == SNAPPY_BYTE2) {
  51         file = File::createSnappy();
  52     } else if (byte1 == 0x1f && byte2 == 0x8b) {
  53         file = File::createZLib();
  54     } else  {
  55         os::log("error: %s: unkwnown compression\n", filename);
  56         return NULL;
  57     }
  58     if (!file) {
  59         return NULL;
  60     }
  61
  62     if (!file->open(filename, File::Read)) {
  63         os::log("error: could not open %s for reading\n", filename);
  64         delete file;
  65         return NULL;
  66     }
  67
  68     return file;
  69 }