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., 675 Mass Ave, Cambridge, MA 02139, USA.
  20 //
  21 // $Id$
  22
  23
  24 #include <simgear/compiler.h>
  25
  26 #include STL_STRING
  27
  28 #if defined(_MSC_VER) || defined(__MINGW32__)
  29 #  include <io.h>
  30 #endif
  31
  32 #include <simgear/debug/logstream.hxx>
  33
  34 #include "sg_file.hxx"
  35
  36 SG_USING_STD(string);
  37
  38
  39 SGFile::SGFile( const string &file) {
  40     set_type( sgFileType );
  41     file_name = file;
  42     eof_flag = true;
  43 }
  44
  45
  46 SGFile::~SGFile() {
  47 }
  48
  49
  50 // open the file based on specified direction
  51 bool SGFile::open( const SGProtocolDir d ) {
  52     set_dir( d );
  53
  54     if ( get_dir() == SG_IO_OUT ) {
  55 #if defined(_MSC_VER) || defined(__MINGW32__)
  56         int mode = 00666;
  57 #else
  58         mode_t mode = S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH;
  59 #endif
  60         fp = ::open( file_name.c_str(), O_WRONLY | O_CREAT | O_TRUNC, mode );
  61     } else if ( get_dir() == SG_IO_IN ) {
  62         fp = ::open( file_name.c_str(), O_RDONLY );
  63     } else {
  64         SG_LOG( SG_IO, SG_ALERT,
  65                 "Error:  bidirection mode not available for files." );
  66         return false;
  67     }
  68
  69     if ( fp == -1 ) {
  70         SG_LOG( SG_IO, SG_ALERT, "Error opening file: " << file_name );
  71         return false;
  72     }
  73
  74     eof_flag = false;
  75     return true;
  76 }
  77
  78
  79 // read a block of data of specified size
  80 int SGFile::read( char *buf, int length ) {
  81     // read a chunk
  82     ssize_t result = ::read( fp, buf, length );
  83     if ( length > 0 && result == 0 ) {
  84         eof_flag = true;
  85     }
  86     return result;
  87 }
  88
  89
  90 // read a line of data, length is max size of input buffer
  91 int SGFile::readline( char *buf, int length ) {
  92     // save our current position
  93     int pos = lseek( fp, 0, SEEK_CUR );
  94
  95     // read a chunk
  96     ssize_t result = ::read( fp, buf, length );
  97     if ( length > 0 && result == 0 ) {
  98         eof_flag = true;
  99     }
 100
 101     // find the end of line and reset position
 102     int i;
 103     for ( i = 0; i < result && buf[i] != '\n'; ++i );
 104     if ( buf[i] == '\n' ) {
 105         result = i + 1;
 106     } else {
 107         result = i;
 108     }
 109     lseek( fp, pos + result, SEEK_SET );
 110
 111     // just in case ...
 112     buf[ result ] = '\0';
 113
 114     return result;
 115 }
 116
 117
 118 // write data to a file
 119 int SGFile::write( const char *buf, const int length ) {
 120     int result = ::write( fp, buf, length );
 121     if ( result != length ) {
 122         SG_LOG( SG_IO, SG_ALERT, "Error writing data: " << file_name );
 123     }
 124
 125     return result;
 126 }
 127
 128
 129 // write null terminated string to a file
 130 int SGFile::writestring( const char *str ) {
 131     int length = strlen( str );
 132     return write( str, length );
 133 }
 134
 135
 136 // close the port
 137 bool SGFile::close() {
 138     if ( ::close( fp ) == -1 ) {
 139         return false;
 140     }
 141
 142     eof_flag = true;
 143     return true;
 144 }