COM And Clipboard

[Computergeek01]

So I found myself making a god awful application today, but in the end I am pretty happy with how it turned out.

This is a program to get the up time on a remote system that will be stared by an ancient inventory management system which is happy to call other applications but provides Zero ability for those other applications to communicate with it. So then how to we get data that it is incapable of gathering on it's own into this system you ask? Well, we hijack to clipboard of course! C&C is welcome as always. Thanks for your time!

MAJOR_EDIT: Code has been revised. See below.

[Computergeek01]

So I didn't get much in the way of a response the first time, but the guys over in the IRC gave me some feed back so I did a revision to include some cleanup and error checking. So feel free to ask questions or to comment on any part of the following code. I'm looking for any errors or oversights especially now since adding the cleanup. I have this linking against the following MS libraries:

msvcrt.lib
msvcprt.lib
Ole32.lib
Wbemuuid.lib
User32.lib
Iphlpapi.lib
Ws2_32.lib
OleAut32.lib

And here is my revised code:

#include <iostream>
#include <string>
#include <cwchar>
#include <cctype>

#define _WIN32_WINNT 0x0600
#define WIN32_LEAN_AND_MEAN ///This Must Be Done So That The 'Winsock.h' Header Is Not Included With 'Windows.h'

#include <windows.h>
#include <winbase.h>
#include <WinCon.h>
#include <Winuser.h>

#include <Objbase.h>
#include <Wbemidl.h>
#include <Wbemcli.h>

#include <Ipexport.h>
#include <Icmpapi.h>
#include <Ws2tcpip.h>

class DATES
{
private:


public:

    DATES(VARIANT& Var)
    {
        std::wstring BaseString;
        std::wstring YearString;
        std::wstring MonthString;
        std::wstring DayString;
        std::wstring HourString;
        std::wstring MinuteString;

        BaseString = Var.bstrVal;

        YearString = BaseString.substr(0, 4);
        Year = wcstol(YearString.c_str(), NULL, 10);

        MonthString = BaseString.substr(4, 2);
        Month = wcstol(MonthString.c_str(), NULL, 10);

        DayString = BaseString.substr(6, 2);
        Day = wcstol(DayString.c_str(), NULL, 10);

        HourString = BaseString.substr(8, 2);
        Hour = wcstol(HourString.c_str(), NULL, 10);

        MinuteString = BaseString.substr(10, 2);
        Minute = wcstol(MinuteString.c_str(), NULL, 10);
    }

    std::wstring DateDifference(DATES& Dates)
    {
        WCHAR DateDifString[MAX_PATH];

        long lYear = Year - Dates.Year;

        long lMonth = Month - Dates.Month;

        long lDay = Day - Dates.Day;

        long lHour = Hour - Dates.Hour;

        long lMinute = Minute - Dates.Minute;


        if(lMinute < 0)
        {
           lMinute = lMinute + 60;
           lHour = lHour - 1;
        }

        if(lHour < 0)
        {
           lHour = lHour + 24;
           lDay = lDay - 1;
        }

        if(lDay < 0)
        {
           lDay = lDay + 30;
           lMonth = lMonth - 1;
        }

        if(lMonth < 0)
        {
           lMonth = lMonth + 12;
        }


        swprintf_s(DateDifString, MAX_PATH, L"Yrs: %i Mts: %i Dys: %i Hrs: %i Min: %i", lYear, lMonth, lDay, lHour, lMinute);

        std::wstring Difference = DateDifString;
        return Difference;
    }

    long Year;
    long Month;
    long Day;

    long Hour;
    long Minute;
};


int main(int NumArgs, char* Argvc[])
{
    WSADATA WsaData;
    WSAStartup(MAKEWORD(2,2), &WsaData);

    addrinfo* Hints = new addrinfo;
        Hints->ai_flags = AI_ADDRCONFIG;
        Hints->ai_family = AF_UNSPEC;
        Hints->ai_addrlen = 0;
        Hints->ai_canonname = NULL;
        Hints->ai_next = NULL;
        Hints->ai_addr = NULL;
        Hints->ai_socktype = SOCK_STREAM;

    addrinfo* IP_Address = NULL;

    HANDLE ICMP = IcmpCreateFile();

    WCHAR strComputer[MAX_COMPUTERNAME_LENGTH];
    WCHAR ComConnection[MAX_PATH];

    IWbemLocator* pLocator = NULL;
    IWbemServices* pWMI = NULL;

    ULONG ClassEnumReturned = 0;
    IEnumWbemClassObject* ClassEnum = NULL;
    IWbemClassObject*     ClassInstance = NULL;

    VARIANT LastBootTime;
    VARIANT LocalDateTime;

    BOOL Clipboard = OpenClipboard(GetConsoleWindow());
    EmptyClipboard();

    char Err[MAX_PATH];
    char SendBuffer[MAX_PATH];

    char* cStrData = NULL;;

    const int rBufferSize = 100;
    ICMP_ECHO_REPLY rBuffer[rBufferSize];
        ZeroMemory(rBuffer, sizeof(rBuffer));

    std::string Data = "NO_DATA";
    size_t DataSize = Data.size() + 1;

    HGLOBAL gHandle = NULL;
    LPVOID hMemory = NULL;

    std::string HostName;
    std::string dotAddress;

    std::wstring wStrData;

    TCHAR* IP_String = NULL;
    sockaddr_in* AddrIn = NULL;
    IPAddr IP;

    size_t cStrDataSize = 0;

    try
    {
        if(!Clipboard)
        {
            throw "NO CLIPBOARD";
        }

        if(NumArgs >= 2)
        {
            HostName = Argvc[1];

            if(HostName.length() >= MAX_COMPUTERNAME_LENGTH)
            {
                throw "Host Name Too Long";
            }
            else
            {
                mbstowcs_s(NULL, strComputer, MAX_COMPUTERNAME_LENGTH, Argvc[1], MAX_COMPUTERNAME_LENGTH);
            }
        }
        else
        {
            throw "Too Few Arguments";
        }

        swprintf_s(ComConnection, MAX_PATH, L"\\\\%s\\ROOT\\CIMV2", strComputer);

        CoInitializeEx(NULL, COINIT_MULTITHREADED);
        CoInitializeSecurity(NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_DEFAULT, RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE, NULL);

        CoCreateInstance(CLSID_WbemLocator, 0, CLSCTX_INPROC_SERVER, IID_IWbemLocator, (LPVOID*) &pLocator);

        if(getaddrinfo(HostName.c_str(), NULL, Hints, &IP_Address) != 0)
        {
            sprintf(Err, "COULD NOT RESOLVE ADDRESS %i", GetLastError());

            throw Err;
        }

        IP_String = new TCHAR[IP_Address->ai_addrlen];
        AddrIn = reinterpret_cast<sockaddr_in*>(IP_Address->ai_addr);

        dotAddress = inet_ntoa(AddrIn->sin_addr);
        IP = inet_addr(dotAddress.c_str());

        if(IcmpSendEcho(ICMP, IP, SendBuffer, MAX_PATH, NULL, rBuffer, sizeof(rBuffer), 150) == 0)
        {
            if(IcmpSendEcho(ICMP, IP, SendBuffer, MAX_PATH, NULL, rBuffer, sizeof(rBuffer), 150) == 0)
            {
                throw "NO PING";
            }
        }

        if(pLocator->ConnectServer(BSTR(ComConnection), NULL, NULL, 0, NULL, 0, 0, &pWMI) != WBEM_NO_ERROR)
        {
            sprintf(Err, "COM ERROR %i", GetLastError());

            throw Err;
        }

        CoSetProxyBlanket(pWMI, RPC_C_AUTHN_WINNT, RPC_C_AUTHZ_NONE, NULL, RPC_C_AUTHN_LEVEL_CALL, RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE);
        pWMI->ExecQuery(BSTR(L"WQL"), BSTR(L"Select * From Win32_OperatingSystem"), WBEM_FLAG_BIDIRECTIONAL, NULL, &ClassEnum);

        ClassEnum->Next(WBEM_INFINITE, 1, &ClassInstance, &ClassEnumReturned);

        ClassInstance->Get(BSTR(L"LastBootUpTime"), 0, &LastBootTime, NULL, NULL);
        ClassInstance->Get(BSTR(L"LocalDateTime"), 0, &LocalDateTime, NULL, NULL);

        DATES LastBoot(LastBootTime);
        DATES LocalDate(LocalDateTime);

        wStrData = LocalDate.DateDifference(LastBoot);

        cStrDataSize = wStrData.size() + 1;
        cStrData = new char[cStrDataSize];

        wcstombs_s(NULL, cStrData, cStrDataSize, wStrData.c_str(), cStrDataSize);

        Data = cStrData;

        delete cStrData;
        delete IP_String;

        VariantClear(&LastBootTime);
        VariantClear(&LocalDateTime);

        ClassEnum->Release();
        ClassInstance->Release();

        pWMI->Release();
        pLocator->Release();
    }
    catch(const char* Msg)
    {
        Data = Msg;
    }

    DataSize = Data.size() + 1;

    gHandle = GlobalAlloc(GMEM_ZEROINIT, DataSize);
    hMemory = GlobalLock(gHandle);

    memcpy(hMemory, Data.data(), DataSize);

    SetClipboardData(CF_TEXT, gHandle);
    CloseClipboard();

    GlobalUnlock(gHandle);
    GlobalFree(gHandle);

    delete Hints;

    WSACleanup();

    CoUninitialize();

    return 0;
}

[ResidentBiscuit]

What was the point of this application? To get the uptime of the system?

[Computergeek01]

We use this decrepit old system in my office to manage where our PC's are and what software licensing they have applied to them. This "Database" does not allow external applications to interact with it but it does provide the ability to launch other apps and also the ability to paste from the clipboard into itself. This was something I wrote so that we could get the up time for a set of remote systems, one at a time, and feed that data into the clipboard so that this DB could then paste it into one of it's fields. It's a horrible workaround conceptually but I'm happy with the way it turned out on this end.