| /* |
| * Copyright (c) 2010 Massachusetts Institute of Technology |
| * All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions are |
| * met: redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer; |
| * redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in the |
| * documentation and/or other materials provided with the distribution; |
| * neither the name of the copyright holders nor the names of its |
| * contributors may be used to endorse or promote products derived from |
| * this software without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| * |
| * Authors: Chia-Hsin Owen Chen |
| */ |
| |
| #include "mem/ruby/network/orion/ConfigFile.hh" |
| |
| using std::string; |
| |
| ConfigFile::ConfigFile( string filename, string delimiter, |
| string comment, string sentry ) |
| : myDelimiter(delimiter), myComment(comment), mySentry(sentry) |
| { |
| // Construct a ConfigFile, getting keys and values from given file |
| |
| std::ifstream in( filename.c_str() ); |
| |
| if( !in ) throw file_not_found( filename ); |
| |
| in >> (*this); |
| } |
| |
| |
| ConfigFile::ConfigFile() |
| : myDelimiter( string(1,'=') ), myComment( string(1,'#') ) |
| { |
| // Construct a ConfigFile without a file; empty |
| } |
| |
| |
| void ConfigFile::remove( const string& key ) |
| { |
| // Remove key and its value |
| myContents.erase( myContents.find( key ) ); |
| return; |
| } |
| |
| |
| bool ConfigFile::keyExists( const string& key ) const |
| { |
| // Indicate whether key is found |
| mapci p = myContents.find( key ); |
| return ( p != myContents.end() ); |
| } |
| |
| |
| /* static */ |
| void ConfigFile::trim( string& s ) |
| { |
| // Remove leading and trailing whitespace |
| static const char whitespace[] = " \n\t\v\r\f"; |
| s.erase( 0, s.find_first_not_of(whitespace) ); |
| s.erase( s.find_last_not_of(whitespace) + 1U ); |
| } |
| |
| |
| std::ostream& operator<<( std::ostream& os, const ConfigFile& cf ) |
| { |
| // Save a ConfigFile to os |
| for( ConfigFile::mapci p = cf.myContents.begin(); |
| p != cf.myContents.end(); |
| ++p ) |
| { |
| os << p->first << " " << cf.myDelimiter << " "; |
| os << p->second << std::endl; |
| } |
| return os; |
| } |
| |
| |
| std::istream& operator>>( std::istream& is, ConfigFile& cf ) |
| { |
| // Load a ConfigFile from is |
| // Read in keys and values, keeping internal whitespace |
| typedef string::size_type pos; |
| const string& delim = cf.myDelimiter; // separator |
| const string& comm = cf.myComment; // comment |
| const string& sentry = cf.mySentry; // end of file sentry |
| const pos skip = delim.length(); // length of separator |
| |
| string nextline = ""; // might need to read ahead to see where value ends |
| |
| while( is || nextline.length() > 0 ) |
| { |
| // Read an entire line at a time |
| string line; |
| if( nextline.length() > 0 ) |
| { |
| line = nextline; // we read ahead; use it now |
| nextline = ""; |
| } |
| else |
| { |
| std::getline( is, line ); |
| } |
| |
| // Ignore comments |
| line = line.substr( 0, line.find(comm) ); |
| |
| // Check for end of file sentry |
| if( sentry != "" && line.find(sentry) != string::npos ) return is; |
| |
| // Parse the line if it contains a delimiter |
| pos delimPos = line.find( delim ); |
| if( delimPos < string::npos ) |
| { |
| // Extract the key |
| string key = line.substr( 0, delimPos ); |
| line.replace( 0, delimPos+skip, "" ); |
| |
| // See if value continues on the next line |
| // Stop at blank line, next line with a key, end of stream, |
| // or end of file sentry |
| bool terminate = false; |
| while( !terminate && is ) |
| { |
| std::getline( is, nextline ); |
| terminate = true; |
| |
| string nlcopy = nextline; |
| ConfigFile::trim(nlcopy); |
| if( nlcopy == "" ) continue; |
| |
| nextline = nextline.substr( 0, nextline.find(comm) ); |
| if( nextline.find(delim) != string::npos ) |
| continue; |
| if( sentry != "" && nextline.find(sentry) != string::npos ) |
| continue; |
| |
| nlcopy = nextline; |
| ConfigFile::trim(nlcopy); |
| if( nlcopy != "" ) line += "\n"; |
| line += nextline; |
| terminate = false; |
| } |
| |
| // Store key and value |
| ConfigFile::trim(key); |
| ConfigFile::trim(line); |
| cf.myContents[key] = line; // overwrites if key is repeated |
| } |
| } |
| |
| return is; |
| } |