don't crash if path == base_directory.

[reactos.git] / reactos / tools / rbuild / XML.cpp

diff --git a/reactos/tools/rbuild/XML.cpp b/reactos/tools/rbuild/XML.cpp
index 4e86c8f..af1cb10 100644 (file)
--- a/reactos/tools/rbuild/XML.cpp
+++ b/reactos/tools/rbuild/XML.cpp
@@ -1,5 +1,20 @@
-// XML.cpp
-
+/*
+ * Copyright (C) 2005 Casper S. Hornstrup
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ */
 #include "pch.h"
 
 #ifdef _MSC_VER
@@ -157,21 +172,31 @@ Path::RelativeFromWorkingDirectory ()
 string
 Path::RelativeFromWorkingDirectory ( const string& path )
 {
-       vector<string> vwork, vpath, vout;
-       Path::Split ( vwork, working_directory, true );
+       return Path::RelativeFromDirectory ( path, working_directory );
+}
+
+string
+Path::RelativeFromDirectory (
+       const string& path,
+       const string& base_directory )
+{
+       vector<string> vbase, vpath, vout;
+       Path::Split ( vbase, base_directory, true );
        Path::Split ( vpath, path, true );
 #ifdef WIN32
        // this squirreliness is b/c win32 has drive letters and *nix doesn't...
        // not possible to do relative across different drive letters
-       if ( vwork[0] != vpath[0] )
+       if ( vbase[0] != vpath[0] )
                return path;
 #endif
        size_t i = 0;
-       while ( i < vwork.size() && i < vpath.size() && vwork[i] == vpath[i] )
+       while ( i < vbase.size() && i < vpath.size() && vbase[i] == vpath[i] )
                ++i;
-       if ( i < vwork.size() )
+       if ( vbase.size() == vpath.size() && i == vpath.size() )
+               return ".";
+       if ( i < vbase.size() )
        {
-               // path goes above our working directory, we will need some ..'s
+               // path goes above our base directory, we will need some ..'s
                for ( size_t j = 0; j < i; j++ )
                        vout.push_back ( ".." );
        }