wjb/evaluation/pin-3.15-98253-gb56e429b1-gcc-linux/source/tools/ImageTests/images_on_attach_app.cpp

/*
 * Copyright 2002-2019 Intel Corporation.
 * 
 * This software is provided to you as Sample Source Code as defined in the accompanying
 * End User License Agreement for the Intel(R) Software Development Products ("Agreement")
 * section 1.L.
 * 
 * This software and the related documents are provided as is, with no express or implied
 * warranties, other than those that are expressly stated in the License.
 */

/*
 *  This application should be used with the images_on_attach_tool tool.
 *  See documentation in the tool for the test details.
 */

#include <iostream>
#include <cstdlib>
#include <dlfcn.h>
#include <unistd.h>

using std::cout;
using std::cerr;
using std::endl;


extern "C"
{
void DoRelease(volatile bool* doRelease)
{
    // Do nothing
}
} // extern "C"


static void WaitForAttach()
{
    const unsigned int timeout = 300;
    unsigned int releaseCounter = 0;
    volatile bool released = false;
    while (!released)
    {
        if (timeout == releaseCounter)
        {
            cerr << "APP ERROR: Timeout reached and the tool did not release the application." << endl;
            exit(1);
        }
        ++releaseCounter;
        DoRelease(&released);
        sleep(1);
    }
}


static void LoadAdditionalLibraries(const char* usrlib)
{
    const void *libutil = dlopen("libutil.so.1", RTLD_LAZY);
    if(NULL == libutil)
    {
        cerr << "APP ERROR: Failed to load libutil.so.1" << endl;
        exit(1);
    }
    const void *usrlibptr = dlopen(usrlib, RTLD_LAZY);
    if(NULL == usrlibptr)
    {
        cerr << "APP ERROR: Failed to load " << usrlib << endl;
        exit(1);
    }
}


/*
 * Expected arguments:
 *
 * [1] - Shared object to load dynamically
 */
int main( int argc, char *argv[] )
{
    // Check the number of parameters.
    if (2 != argc)
    {
        cerr << "Usage: " << argv[0] << " <shared object to load>" << endl;
        return 1;
    }

    // Wait for the tool to attach to the application.
    WaitForAttach();

    // Pin is attached, now load two more shared objects.
    LoadAdditionalLibraries(argv[1]);

    // Done.
    cout << "APP: Application completed successfully." << endl;
    return 0;
}