Edit

kc3-lang/angle/src/common/system_utils_win.cpp

Branch :

  • Show log

    Commit

  • Author : Jamie Madill
    Date : 2019-01-16 14:12:28
    Hash : bab0302b
    Message : Make path utils return std::string. This avoids potential issues with returning local variables. It also solves potential threading conflicts by returning the string value instead of a pointer. The code should be optimized with RVO with a modern c++ compiler. Bug: angleproject:2601 Change-Id: I8a01702d2675a17dd060f27920105efab0c49454 Reviewed-on: https://chromium-review.googlesource.com/c/1415910 Reviewed-by: Kai Ninomiya <kainino@chromium.org> Reviewed-by: Geoff Lang <geofflang@chromium.org> Commit-Queue: Jamie Madill <jmadill@chromium.org>

  • src/common/system_utils_win.cpp
  • //
    // Copyright (c) 2014 The ANGLE Project Authors. All rights reserved.
    // Use of this source code is governed by a BSD-style license that can be
    // found in the LICENSE file.
    //
    
    // system_utils_win.cpp: Implementation of OS-specific functions for Windows
    
    #include "system_utils.h"
    
    #include <stdarg.h>
    #include <windows.h>
    #include <array>
    #include <vector>
    
    namespace angle
    {
    std::string GetExecutablePath()
    {
        std::array<char, MAX_PATH> executableFileBuf;
        DWORD executablePathLen = GetModuleFileNameA(nullptr, executableFileBuf.data(),
                                                     static_cast<DWORD>(executableFileBuf.size()));
        return (executablePathLen > 0 ? std::string(executableFileBuf.data()) : "");
    }
    
    std::string GetExecutableDirectory()
    {
        std::string executablePath = GetExecutablePath();
        size_t lastPathSepLoc      = executablePath.find_last_of("\\/");
        return (lastPathSepLoc != std::string::npos) ? executablePath.substr(0, lastPathSepLoc) : "";
    }
    
    const char *GetSharedLibraryExtension()
    {
        return "dll";
    }
    
    Optional<std::string> GetCWD()
    {
        std::array<char, MAX_PATH> pathBuf;
        DWORD result = GetCurrentDirectoryA(static_cast<DWORD>(pathBuf.size()), pathBuf.data());
        if (result == 0)
        {
            return Optional<std::string>::Invalid();
        }
        return std::string(pathBuf.data());
    }
    
    bool SetCWD(const char *dirName)
    {
        return (SetCurrentDirectoryA(dirName) == TRUE);
    }
    
    bool UnsetEnvironmentVar(const char *variableName)
    {
        return (SetEnvironmentVariableA(variableName, nullptr) == TRUE);
    }
    
    bool SetEnvironmentVar(const char *variableName, const char *value)
    {
        return (SetEnvironmentVariableA(variableName, value) == TRUE);
    }
    
    std::string GetEnvironmentVar(const char *variableName)
    {
        std::array<char, MAX_PATH> oldValue;
        DWORD result =
            GetEnvironmentVariableA(variableName, oldValue.data(), static_cast<DWORD>(oldValue.size()));
        if (result == 0)
        {
            return std::string();
        }
        else
        {
            return std::string(oldValue.data());
        }
    }
    
    const char *GetPathSeparator()
    {
        return ";";
    }
    
    class Win32Library : public Library
    {
      public:
        Win32Library(const char *libraryName)
        {
            char buffer[MAX_PATH];
            int ret = snprintf(buffer, MAX_PATH, "%s.%s", libraryName, GetSharedLibraryExtension());
            if (ret > 0 && ret < MAX_PATH)
            {
                mModule = LoadLibraryA(buffer);
            }
        }
    
        ~Win32Library() override
        {
            if (mModule)
            {
                FreeLibrary(mModule);
            }
        }
    
        void *getSymbol(const char *symbolName) override
        {
            if (!mModule)
            {
                return nullptr;
            }
    
            return reinterpret_cast<void *>(GetProcAddress(mModule, symbolName));
        }
    
        void *getNative() const override { return reinterpret_cast<void *>(mModule); }
    
      private:
        HMODULE mModule = nullptr;
    };
    
    Library *OpenSharedLibrary(const char *libraryName)
    {
        return new Win32Library(libraryName);
    }
    }  // namespace angle