source: trunk/synergy/lib/arch/CArchFileWindows.cpp@ 2828

/*
 * synergy -- mouse and keyboard sharing utility
 * Copyright (C) 2002 Chris Schoeneman
 * 
 * This package is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * found in the file COPYING that should have accompanied this file.
 * 
 * This package 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.
 */

#include "CArchFileWindows.h"
#include <windows.h>
#include <shlobj.h>
#include <tchar.h>
#include <string.h>

//
// CArchFileWindows
//

CArchFileWindows::CArchFileWindows()
{
        // do nothing
}

CArchFileWindows::~CArchFileWindows()
{
        // do nothing
}

const char*
CArchFileWindows::getBasename(const char* pathname)
{
        if (pathname == NULL) {
                return NULL;
        }

        // check for last slash
        const char* basename = strrchr(pathname, '/');
        if (basename != NULL) {
                ++basename;
        }
        else {
                basename = pathname;
        }

        // check for last backslash
        const char* basename2 = strrchr(pathname, '\\');
        if (basename2 != NULL && basename2 > basename) {
                basename = basename2 + 1;
        }

        return basename;
}

std::string
CArchFileWindows::getUserDirectory()
{
        // try %HOMEPATH%
        TCHAR dir[MAX_PATH];
        DWORD size   = sizeof(dir) / sizeof(TCHAR);
        DWORD result = GetEnvironmentVariable(_T("HOMEPATH"), dir, size);
        if (result != 0 && result <= size) {
                // sanity check -- if dir doesn't appear to start with a
                // drive letter and isn't a UNC name then don't use it
                // FIXME -- allow UNC names
                if (dir[0] != '\0' && (dir[1] == ':' ||
                        ((dir[0] == '\\' || dir[0] == '/') &&
                        (dir[1] == '\\' || dir[1] == '/')))) {
                        return dir;
                }
        }

        // get the location of the personal files.  that's as close to
        // a home directory as we're likely to find.
        ITEMIDLIST* idl;
        if (SUCCEEDED(SHGetSpecialFolderLocation(NULL, CSIDL_PERSONAL, &idl))) {
                TCHAR* path = NULL;
                if (SHGetPathFromIDList(idl, dir)) {
                        DWORD attr = GetFileAttributes(dir);
                        if (attr != 0xffffffff && (attr & FILE_ATTRIBUTE_DIRECTORY) != 0)
                                path = dir;
                }

                IMalloc* shalloc;
                if (SUCCEEDED(SHGetMalloc(&shalloc))) {
                        shalloc->Free(idl);
                        shalloc->Release();
                }

                if (path != NULL) {
                        return path;
                }
        }

        // use root of C drive as a default
        return "C:";
}

std::string
CArchFileWindows::getSystemDirectory()
{
        // get windows directory
        char dir[MAX_PATH];
        if (GetWindowsDirectory(dir, sizeof(dir)) != 0) {
                return dir;
        }
        else {
                // can't get it.  use C:\ as a default.
                return "C:";
        }
}

std::string
CArchFileWindows::concatPath(const std::string& prefix,
                                const std::string& suffix)
{
        std::string path;
        path.reserve(prefix.size() + 1 + suffix.size());
        path += prefix;
        if (path.size() == 0 ||
                (path[path.size() - 1] != '\\' &&
                path[path.size() - 1] != '/')) {
                path += '\\';
        }
        path += suffix;
        return path;
}