simgear/io/sg_file.cxx

   1 // sg_file.cxx -- File I/O routines
   2 //
   3 // Written by Curtis Olson, started November 1999.
   4 //
   5 // Copyright (C) 1999  Curtis L. Olson - http://www.flightgear.org/~curt
   6 //
   7 // This program is free software; you can redistribute it and/or
   8 // modify it under the terms of the GNU General Public License as
   9 // published by the Free Software Foundation; either version 2 of the
  10 // License, or (at your option) any later version.
  11 //
  12 // This program is distributed in the hope that it will be useful, but
  13 // WITHOUT ANY WARRANTY; without even the implied warranty of
  14 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15 // General Public License for more details.
  16 //
  17 // You should have received a copy of the GNU General Public License
  18 // along with this program; if not, write to the Free Software
  19 // Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  20 //
  21 // $Id$
  22
  23
  24 #include <simgear/compiler.h>
  25
  26 #include <string>
  27
  28 #if defined(_MSC_VER) || defined(__MINGW32__)
  29 #  include <io.h>
  30 #endif
  31
  32 #include <cstring>
  33
  34 #include <simgear/misc/stdint.hxx>
  35 #include <simgear/debug/logstream.hxx>
  36
  37 #include "sg_file.hxx"
  38
  39 SG_USING_STD(string);
  40
  41
  42 SGFile::SGFile(const string &file, bool repeat_)
  43     : file_name(file), fp(-1), eof_flag(true), repeat(repeat_)
  44 {
  45     set_type( sgFileType );
  46 }
  47
  48
  49 SGFile::~SGFile() {
  50 }
  51
  52
  53 // open the file based on specified direction
  54 bool SGFile::open( const SGProtocolDir d ) {
  55     set_dir( d );
  56
  57     if ( get_dir() == SG_IO_OUT ) {
  58 #if defined(_MSC_VER) || defined(__MINGW32__)
  59         int mode = 00666;
  60 #else
  61         mode_t mode = S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH;
  62 #endif
  63         fp = ::open( file_name.c_str(), O_WRONLY | O_CREAT | O_TRUNC, mode );
  64     } else if ( get_dir() == SG_IO_IN ) {
  65         fp = ::open( file_name.c_str(), O_RDONLY );
  66     } else {
  67         SG_LOG( SG_IO, SG_ALERT,
  68                 "Error:  bidirection mode not available for files." );
  69         return false;
  70     }
  71
  72     if ( fp == -1 ) {
  73         SG_LOG( SG_IO, SG_ALERT, "Error opening file: " << file_name );
  74         return false;
  75     }
  76
  77     eof_flag = false;
  78     return true;
  79 }
  80
  81
  82 // read a block of data of specified size
  83 int SGFile::read( char *buf, int length ) {
  84     // read a chunk
  85     ssize_t result = ::read( fp, buf, length );
  86     if ( length > 0 && result == 0 ) {
  87         if (repeat) {
  88             // loop reading the file, unless it is empty
  89             off_t fileLen = ::lseek(fp, 0, SEEK_CUR);
  90             if (fileLen == 0) {
  91                 eof_flag = true;
  92                 return 0;
  93             } else {
  94                 ::lseek(fp, 0, SEEK_SET);
  95                 return ::read(fp, buf, length);
  96             }
  97         } else {
  98             eof_flag = true;
  99         }
 100     }
 101     return result;
 102 }
 103
 104
 105 // read a line of data, length is max size of input buffer
 106 int SGFile::readline( char *buf, int length ) {
 107     // save our current position
 108     int pos = lseek( fp, 0, SEEK_CUR );
 109
 110     // read a chunk
 111     ssize_t result = ::read( fp, buf, length );
 112     if ( length > 0 && result == 0 ) {
 113         if (repeat && pos != 0) {
 114             pos = ::lseek(fp, 0, SEEK_SET);
 115             result = ::read(fp, buf, length);
 116         } else {
 117             eof_flag = true;
 118         }
 119     }
 120
 121     // find the end of line and reset position
 122     int i;
 123     for ( i = 0; i < result && buf[i] != '\n'; ++i );
 124     if ( buf[i] == '\n' ) {
 125         result = i + 1;
 126     } else {
 127         result = i;
 128     }
 129     lseek( fp, pos + result, SEEK_SET );
 130
 131     // just in case ...
 132     buf[ result ] = '\0';
 133
 134     return result;
 135 }
 136
 137
 138 // write data to a file
 139 int SGFile::write( const char *buf, const int length ) {
 140     int result = ::write( fp, buf, length );
 141     if ( result != length ) {
 142         SG_LOG( SG_IO, SG_ALERT, "Error writing data: " << file_name );
 143     }
 144
 145     return result;
 146 }
 147
 148
 149 // write null terminated string to a file
 150 int SGFile::writestring( const char *str ) {
 151     int length = std::strlen( str );
 152     return write( str, length );
 153 }
 154
 155
 156 // close the port
 157 bool SGFile::close() {
 158     if ( ::close( fp ) == -1 ) {
 159         return false;
 160     }
 161
 162     eof_flag = true;
 163     return true;
 164 }