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