#include "os_string.hpp"
#include "os_process.hpp"
-#include "trace_resource.hpp"
#include "trace_tools.hpp"
#define TRACE_VARIABLE "DYLD_LIBRARY_PATH"
#define GL_TRACE_WRAPPER "OpenGL"
#elif defined(_WIN32)
-#define TRACE_VARIABLE ""
#define GL_TRACE_WRAPPER "opengl32.dll"
#else
#define TRACE_VARIABLE "LD_PRELOAD"
#endif
+static os::String
+findWrapper(const char *wrapperFilename)
+{
+ os::String wrapperPath;
+
+ os::String processDir = os::getProcessName();
+ processDir.trimFilename();
+
+ // Try relative build directory
+ // XXX: Just make build and install directory layout match
+ wrapperPath = processDir;
+ wrapperPath.join("wrappers");
+ wrapperPath.join(wrapperFilename);
+ if (wrapperPath.exists()) {
+ return wrapperPath;
+ }
+
+ // Try relative install directory
+ wrapperPath = processDir;
+#if defined(_WIN32)
+ wrapperPath.join("..\\lib\\wrappers");
+#elif defined(__APPLE__)
+ wrapperPath.join("../lib/wrappers");
+#else
+ wrapperPath.join("../lib/apitrace/wrappers");
+#endif
+ wrapperPath.join(wrapperFilename);
+ if (wrapperPath.exists()) {
+ return wrapperPath;
+ }
+
+#ifndef _WIN32
+ // Try absolute install directory
+ wrapperPath = APITRACE_WRAPPERS_INSTALL_DIR;
+ wrapperPath.join(wrapperFilename);
+ if (wrapperPath.exists()) {
+ return wrapperPath;
+ }
+#endif
+
+ return "";
+}
+
+
int
traceProgram(API api,
char * const *argv,
const char *output,
bool verbose)
{
- const char *relPath;
- const char *absPath;
+ const char *wrapperFilename;
+
+ /*
+ * TODO: simplify code
+ */
switch (api) {
case API_GL:
- relPath = "wrappers/" GL_TRACE_WRAPPER;
- absPath = APITRACE_WRAPPER_INSTALL_DIR "/" GL_TRACE_WRAPPER;
+ wrapperFilename = GL_TRACE_WRAPPER;
break;
+#ifdef EGL_TRACE_WRAPPER
case API_EGL:
-#ifndef EGL_TRACE_WRAPPER
- std::cerr << "error: unsupported API\n";
- return 1;
-#else
- relPath = "wrappers/" EGL_TRACE_WRAPPER;
- absPath = APITRACE_WRAPPER_INSTALL_DIR "/" EGL_TRACE_WRAPPER;
+ wrapperFilename = EGL_TRACE_WRAPPER;
+ break;
+#endif
+#ifdef _WIN32
+ case API_D3D7:
+ wrapperFilename = "ddraw.dll";
+ break;
+ case API_D3D8:
+ wrapperFilename = "d3d8.dll";
+ break;
+ case API_D3D9:
+ wrapperFilename = "d3d9.dll";
+ break;
+ case API_D3D10:
+ wrapperFilename = "d3d10.dll";
+ break;
+ case API_D3D10_1:
+ wrapperFilename = "d3d10_1.dll";
+ break;
+ case API_D3D11:
+ wrapperFilename = "d3d11.dll";
break;
#endif
default:
- std::cerr << "error: invalid API\n";
+ std::cerr << "error: unsupported API\n";
return 1;
}
- os::String wrapper;
- wrapper = findFile(relPath, absPath, verbose);
+ os::String wrapperPath = findWrapper(wrapperFilename);
- if (!wrapper.length()) {
+ if (!wrapperPath.length()) {
+ std::cerr << "error: failed to find " << wrapperFilename << "\n";
return 1;
}
#if defined(_WIN32)
+ /* On Windows copy the wrapper to the program directory.
+ */
+ os::String tmpWrapper(argv[0]);
+ tmpWrapper.trimFilename();
+ tmpWrapper.join(wrapperFilename);
- std::cerr <<
- "The 'apitrace trace' command is not supported for this operating system.\n"
- "Instead, you will need to copy opengl32.dll, d3d8.dll, or d3d9.dll from\n"
- APITRACE_WRAPPER_INSTALL_DIR "\n"
- "to the directory with the application to trace, then run the application.\n";
+ if (verbose) {
+ std::cerr << wrapperPath << " -> " << tmpWrapper << "\n";
+ }
- return 1;
+ if (tmpWrapper.exists()) {
+ std::cerr << "error: not overwriting " << tmpWrapper << "\n";
+ return 1;
+ }
-#else
+ if (!os::copyFile(wrapperPath, tmpWrapper, false)) {
+ std::cerr << "error: failed to copy " << wrapperPath << " into " << tmpWrapper << "\n";
+ return 1;
+ }
+#endif /* _WIN32 */
#if defined(__APPLE__)
/* On Mac OS X, using DYLD_LIBRARY_PATH, we actually set the
* directory, not the file. */
- wrapper.trimFilename();
+ wrapperPath.trimFilename();
#endif
+#if defined(TRACE_VARIABLE)
if (verbose) {
- std::cerr << TRACE_VARIABLE << "=" << wrapper.str() << "\n";
+ std::cerr << TRACE_VARIABLE << "=" << wrapperPath.str() << "\n";
}
-
/* FIXME: Don't modify the current environment */
- setenv(TRACE_VARIABLE, wrapper.str(), 1);
+ os::setEnvironment(TRACE_VARIABLE, wrapperPath.str());
+#endif /* TRACE_VARIABLE */
if (output) {
- setenv("TRACE_FILE", output, 1);
+ os::setEnvironment("TRACE_FILE", output);
}
if (verbose) {
int status = os::execute(argv);
- unsetenv(TRACE_VARIABLE);
+#if defined(TRACE_VARIABLE)
+ os::unsetEnvironment(TRACE_VARIABLE);
+#endif
+#if defined(_WIN32)
+ os::removeFile(tmpWrapper);
+#endif
+
if (output) {
- unsetenv("TRACE_FILE");
+ os::unsetEnvironment("TRACE_FILE");
}
return status;
-#endif
}