Subversion Repositories DevTools

/*============================================================================ 
** COPYRIGHT - VIX IP PTY LTD ("VIX"). ALL RIGHTS RESERVED.
**============================================================================
**
**  Project/Product :  JATS ( core)devl)
**  Filename        : gunzip.cpp
**  Author(s)       : dpurdie
**  Created         : 08-Mar-18
**
**  Description     : Implement gunzip by invoking gzip with a -d option
**
**
**  Information     :
**   Compiler       : ANSI C++
**   Target         : 
**
***==========================================================================*/

//
//#include "stdafx.h"
#include <Windows.h>
#include <string>
#include <iostream>
using namespace std;

/*----------------------------------------------------------------------------
** FUNCTION           : ArgvQuote
**
** DESCRIPTION        : Append an argument to a string with correct
**                      quoting for passing as a command line argument 
 
                        This routine appends the given argument to a command line such
                        that CommandLineToArgvW will return the argument string unchanged.
                        Arguments in a command line should be separated by spaces; this
                        function does not add these spaces.
**
** INPUTS             : Argument        - Supplies the argument to encode
**                      CommandLine     - Append to this string
**                      Force           - True: Force quotes
** 
** RETURNS            : Nothing
**
----------------------------------------------------------------------------*/

void ArgvQuote( const std::wstring& Argument,std::wstring& CommandLine, bool Force )
{
    //
    // Unless we're told otherwise, don't quote unless we actually
    // need to do so --- hopefully avoid problems if programs won't
    // parse quotes properly
    //

    if (Force == false &&
        Argument.empty() == false &&
        Argument.find_first_of(L" \t\n\v\"") == Argument.npos)
    {
        CommandLine.append(Argument);
    }
    else {
        CommandLine.append(L"\"");

        for (std::wstring::const_iterator It = Argument.begin(); ; ++It) {
            unsigned NumberBackslashes = 0;

            while (It != Argument.end() && *It == L'\\') {
                ++It;
                ++NumberBackslashes;
            }

            if (It == Argument.end()) {

                //
                // Escape all backslashes, but let the terminating
                // double quotation mark we add below be interpreted
                // as a metacharacter.
                //

                CommandLine.append(NumberBackslashes * 2, L'\\');
                break;
            }
            else if (*It == L'"') {

                //
                // Escape all backslashes and the following
                // double quotation mark.
                //

                CommandLine.append(NumberBackslashes * 2 + 1, L'\\');
                CommandLine.append(1, *It);
            }
            else {

                //
                // Backslashes aren't special here.
                //

                CommandLine.append(NumberBackslashes, L'\\');
                CommandLine.append(1, *It);
            }
        }

        CommandLine.append(L"\"");
    }
}

/*----------------------------------------------------------------------------
** FUNCTION           : s2ws
**
** DESCRIPTION        : String to Wide String
**
**
** INPUTS             : str - std:string
**
** RETURNS            : A widestring
**
----------------------------------------------------------------------------*/
std::wstring s2ws(const std::string& str)
{
    int size_needed = MultiByteToWideChar(CP_UTF8, 0, &str[0], (int)str.size(), NULL, 0);
    std::wstring wstrTo(size_needed, 0);
    MultiByteToWideChar(CP_UTF8, 0, &str[0], (int)str.size(), &wstrTo[0], size_needed);
    return wstrTo;
}

/*----------------------------------------------------------------------------
** FUNCTION           : main
**
** DESCRIPTION        : Invoke gzip with a '-d' option
**
**
** INPUTS             : As per gzip
**
** RETURNS            : Exit code of the gzip utility
**
----------------------------------------------------------------------------*/

int main(int argc, char** argv)
{
    BOOL    rv;
    //
    //  Replace gunzip.exe with gzip.exe -d
    //
    //  Retrieve the command line arguments as an array of pointers to strings
    //  First string will be the program path
    //
    //  Remove program name - keep path
    //  Assume we will find the gzip in the same location us this program or via the same path mechanism
    //  Handle aaaa\bbbb\cccc
    //         aaaa/bbbb/ccccc
    //         cccc
    //
    std::wstring arg = s2ws(argv[0]);
    int index = arg.find_last_of(L"\\", arg.size());
    if (index == std::string::npos) {
        index = arg.find_last_of(L"/", arg.size());
        if (index == std::string::npos) {
            index = -1;
        }
    }
    arg.erase(index + 1);
    arg.append(L"gzip.exe");

    //
    //  Convert the arglist into a fully quoted command line string
    //  Inserting a '-d' option
    //
    std::wstring args = wstring();
    ArgvQuote(arg, args, FALSE);
    args.append(L" ");
    ArgvQuote(L"-d", args, FALSE);
    for (int ii = 1; ii < argc; ii++) {
        args.append(L" ");
        ArgvQuote(s2ws(argv[ii]), args, FALSE);
    }

    //std::wcout << L"Cmd:" << args << std::endl;

    // additional information
    STARTUPINFO si;
    PROCESS_INFORMATION pi;
    DWORD exitStatus = 250;

    // set the size of the structures
    ZeroMemory(&si, sizeof(si));
    ZeroMemory(&pi, sizeof(pi));

    // start the program up
    rv = CreateProcess
    (
        NULL,                   // the path
        (LPWSTR)args.c_str(),   // Command line
        NULL,                   // Process handle not inheritable
        NULL,                   // Thread handle not inheritable
        TRUE,                   // Set handle inheritance to TRUE
        0,                      // Opens file in a separate console
        NULL,                   // Use parent's environment block
        NULL,                   // Use parent's starting directory 
        &si,                    // Pointer to STARTUPINFO structure
        &pi                     // Pointer to PROCESS_INFORMATION structure
    );
    if (!rv) {
        DWORD eCode = GetLastError();
        exitStatus = eCode;
        //std::wcout << L"CreateProcess Error:" << eCode << std::endl;
        return exitStatus;
    }

    // Close process and thread handles.
    WaitForSingleObject(pi.hProcess, INFINITE);
    rv = GetExitCodeProcess(pi.hProcess, &exitStatus);
    if (!rv) {
        DWORD eCode = GetLastError();
        //std::wcout << L"GetExitCodeProcess Error:" << eCode << std::endl;
        exitStatus = 248;
    }
    CloseHandle(pi.hProcess);
    CloseHandle(pi.hThread);
    //std::wcout << L"Exit Status:" << exitStatus << std::endl;
    return exitStatus;
}