Files
renderdoc/renderdoc/os/linux/linux_stringio.cpp
T
2016-03-12 06:29:19 +08:00

473 lines
11 KiB
C++

/******************************************************************************
* The MIT License (MIT)
*
* Copyright (c) 2015-2016 Baldur Karlsson
* Copyright (c) 2014 Crytek
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
******************************************************************************/
#include "os/os_specific.h"
#include "api/app/renderdoc_app.h"
#include <time.h>
#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <pwd.h>
#include <X11/Xlib.h>
#include <X11/keysym.h>
#include <xcb/xcb_keysyms.h>
#include <iconv.h>
// for dladdr
#include <dlfcn.h>
#include "serialise/string_utils.h"
#include "common/threading.h"
using std::string;
// gives us an address to identify this so with
static int soLocator=0;
namespace Keyboard
{
void Init()
{
}
xcb_connection_t *connection;
xcb_key_symbols_t *symbols;
void CloneDisplay(Display *dpy) {}
void UseConnection(xcb_connection_t *conn)
{
connection = conn;
symbols = xcb_key_symbols_alloc(conn);
}
void AddInputWindow(void *wnd)
{
// TODO check against this drawable & parent window being focused in GetKeyState
}
void RemoveInputWindow(void *wnd)
{
}
bool GetKeyState(int key)
{
KeySym ks = 0;
if(symbols == NULL) return false;
if(key >= eRENDERDOC_Key_A && key <= eRENDERDOC_Key_Z) ks = key;
if(key >= eRENDERDOC_Key_0 && key <= eRENDERDOC_Key_9) ks = key;
switch(key)
{
case eRENDERDOC_Key_Divide: ks = XK_KP_Divide; break;
case eRENDERDOC_Key_Multiply: ks = XK_KP_Multiply; break;
case eRENDERDOC_Key_Subtract: ks = XK_KP_Subtract; break;
case eRENDERDOC_Key_Plus: ks = XK_KP_Add; break;
case eRENDERDOC_Key_F1: ks = XK_F1; break;
case eRENDERDOC_Key_F2: ks = XK_F2; break;
case eRENDERDOC_Key_F3: ks = XK_F3; break;
case eRENDERDOC_Key_F4: ks = XK_F4; break;
case eRENDERDOC_Key_F5: ks = XK_F5; break;
case eRENDERDOC_Key_F6: ks = XK_F6; break;
case eRENDERDOC_Key_F7: ks = XK_F7; break;
case eRENDERDOC_Key_F8: ks = XK_F8; break;
case eRENDERDOC_Key_F9: ks = XK_F9; break;
case eRENDERDOC_Key_F10: ks = XK_F10; break;
case eRENDERDOC_Key_F11: ks = XK_F11; break;
case eRENDERDOC_Key_F12: ks = XK_F12; break;
case eRENDERDOC_Key_Home: ks = XK_Home; break;
case eRENDERDOC_Key_End: ks = XK_End; break;
case eRENDERDOC_Key_Insert: ks = XK_Insert; break;
case eRENDERDOC_Key_Delete: ks = XK_Delete; break;
case eRENDERDOC_Key_PageUp: ks = XK_Prior; break;
case eRENDERDOC_Key_PageDn: ks = XK_Next; break;
case eRENDERDOC_Key_Backspace: ks = XK_BackSpace; break;
case eRENDERDOC_Key_Tab: ks = XK_Tab; break;
case eRENDERDOC_Key_PrtScrn: ks = XK_Print; break;
case eRENDERDOC_Key_Pause: ks = XK_Pause; break;
default:
break;
}
if(ks == 0)
return false;
xcb_keycode_t *keyCodes = xcb_key_symbols_get_keycode(symbols, ks);
if(!keyCodes)
return false;
xcb_query_keymap_cookie_t keymapcookie = xcb_query_keymap(connection);
xcb_query_keymap_reply_t *keys = xcb_query_keymap_reply(connection, keymapcookie, NULL);
bool ret = false;
if(keys && keyCodes[0] != XCB_NO_SYMBOL)
{
int byteIdx = (keyCodes[0]/8);
int bitMask = 1 << (keyCodes[0]%8);
ret = (keys->keys[byteIdx] & bitMask) != 0;
}
free(keyCodes);
free(keys);
return ret;
}
}
namespace FileIO
{
string GetAppFolderFilename(const string &filename)
{
passwd *pw = getpwuid(getuid());
const char *homedir = pw->pw_dir;
string ret = string(homedir) + "/.renderdoc/";
mkdir(ret.c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
return ret + filename;
}
void CreateParentDirectory(const string &filename)
{
string fn = dirname(filename);
// want trailing slash so that we create all directories
fn.push_back('/');
if(fn[0] != '/')
return;
size_t offs = fn.find('/', 1);
while(offs != string::npos)
{
// create directory path from 0 to offs by NULLing the
// / at offs, mkdir, then set it back
fn[offs] = 0;
mkdir(fn.c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
fn[offs] = '/';
offs = fn.find_first_of('/', offs+1);
}
}
string GetFullPathname(const string &filename)
{
char path[512] = {0};
readlink(filename.c_str(), path, 511);
return string(path);
}
void GetExecutableFilename(string &selfName)
{
char path[512] = {0};
readlink("/proc/self/exe", path, 511);
selfName = string(path);
}
string GetReplayAppFilename()
{
// look up the shared object's path via dladdr
Dl_info info;
dladdr((void *)&soLocator, &info);
string path = info.dli_fname ? info.dli_fname : "";
path = dirname(path);
string replay = path + "/qrenderdoc";
FILE *f = FileIO::fopen(replay.c_str(), "r");
if(f)
{
FileIO::fclose(f);
return replay;
}
// if it's not in the same directory, try in a sibling /bin
// e.g. /foo/bar/lib/librenderdoc.so -> /foo/bar/bin/qrenderdoc
replay = path + "/../bin/qrenderdoc";
f = FileIO::fopen(replay.c_str(), "r");
if(f)
{
FileIO::fclose(f);
return replay;
}
// random guesses!
const char *guess[] = {
"/opt/renderdoc/qrenderdoc",
"/opt/renderdoc/bin/qrenderdoc",
"/usr/local/bin/qrenderdoc",
"/usr/bin/qrenderdoc"
};
for(size_t i=0; i < ARRAY_COUNT(guess); i++)
{
f = FileIO::fopen(guess[i], "r");
if(f)
{
FileIO::fclose(f);
return guess[i];
}
}
// out of ideas
return "";
}
void GetDefaultFiles(const char *logBaseName, string &capture_filename, string &logging_filename, string &target)
{
char path[2048] = {0};
readlink("/proc/self/exe", path, 511);
const char *mod = strrchr(path, '/');
if(mod == NULL)
mod = "unknown";
else
mod++;
target = string(mod);
time_t t = time(NULL);
tm now = *localtime(&t);
char temp_folder[2048] = { "/tmp" };
char *temp_override = getenv("RENDERDOC_TEMP");
if(temp_override && temp_override[0] == '/')
{
strncpy(temp_folder, temp_override, sizeof(temp_folder)-1);
size_t len = strlen(temp_folder);
while(temp_folder[len-1] == '/') temp_folder[--len] = 0;
}
char temp_filename[2048] = {0};
snprintf(temp_filename, sizeof(temp_filename)-1, "/tmp/%s_%04d.%02d.%02d_%02d.%02d.rdc", mod, 1900+now.tm_year, now.tm_mon+1, now.tm_mday, now.tm_hour, now.tm_min);
capture_filename = string(temp_filename);
snprintf(temp_filename, sizeof(temp_filename)-1, "/tmp/%s_%04d.%02d.%02d_%02d.%02d.%02d.log", logBaseName, 1900+now.tm_year, now.tm_mon+1, now.tm_mday, now.tm_hour, now.tm_min, now.tm_sec);
logging_filename = string(temp_filename);
}
uint64_t GetModifiedTimestamp(const string &filename)
{
struct ::stat st;
int res = stat(filename.c_str(), &st);
if(res == 0)
{
return (uint64_t)st.st_mtime;
}
return 0;
}
void Copy(const char *from, const char *to, bool allowOverwrite)
{
if(from[0] == 0 || to[0] == 0)
return;
FILE *ff = ::fopen(from, "r");
if(!ff)
{
RDCERR("Can't open source file for copy '%s'", from);
return;
}
FILE *tf = ::fopen(to, "r");
if(tf && !allowOverwrite)
{
RDCERR("Destination file for non-overwriting copy '%s' already exists", from);
::fclose(ff);
::fclose(tf);
return;
}
if(tf)
::fclose(tf);
tf = ::fopen(to, "w");
if(!tf)
{
::fclose(ff);
RDCERR("Can't open destination file for copy '%s'", to);
}
char buffer[BUFSIZ];
while(!::feof(ff))
{
size_t nread = ::fread(buffer, 1, BUFSIZ, ff);
::fwrite(buffer, 1, nread, tf);
}
::fclose(ff);
::fclose(tf);
}
void Delete(const char *path)
{
unlink(path);
}
FILE *fopen(const char *filename, const char *mode)
{
return ::fopen(filename, mode);
}
size_t fread(void *buf, size_t elementSize, size_t count, FILE *f) { return ::fread(buf, elementSize, count, f); }
size_t fwrite(const void *buf, size_t elementSize, size_t count, FILE *f) { return ::fwrite(buf, elementSize, count, f); }
uint64_t ftell64(FILE *f) { return (uint64_t)::ftell(f); }
void fseek64(FILE *f, uint64_t offset, int origin) { ::fseek(f, (long)offset, origin); }
bool feof(FILE *f) { return ::feof(f) != 0; }
int fclose(FILE *f) { return ::fclose(f); }
};
namespace StringFormat
{
void sntimef(char *str, size_t bufSize, const char *format)
{
time_t tim;
time(&tim);
tm *tmv = localtime(&tim);
strftime(str, bufSize, format, tmv);
}
string Fmt(const char *format, ...)
{
va_list args;
va_start(args, format);
va_list args2;
va_copy(args2, args);
int size = StringFormat::vsnprintf(NULL, 0, format, args2);
char *buf = new char[size+1];
StringFormat::vsnprintf(buf, size+1, format, args);
buf[size] = 0;
va_end(args);
va_end(args2);
string ret = buf;
delete[] buf;
return ret;
}
// cache iconv_t descriptor to save on iconv_open/iconv_close each time
iconv_t iconvWide2UTF8 = (iconv_t)-1;
// iconv is not thread safe when sharing an iconv_t descriptor
// I don't expect much contention but if it happens we could TryLock
// before creating a temporary iconv_t, or hold two iconv_ts, or something.
Threading::CriticalSection lockWide2UTF8;
string Wide2UTF8(const std::wstring &s)
{
// include room for null terminator, assuming unicode input (not ucs)
// utf-8 characters can be max 4 bytes.
size_t len = (s.length()+1)*4;
vector<char> charBuffer;
if(charBuffer.size() < len)
charBuffer.resize(len);
size_t ret;
{
SCOPED_LOCK(lockWide2UTF8);
if(iconvWide2UTF8 == (iconv_t)-1)
iconvWide2UTF8 = iconv_open("UTF-8", "WCHAR_T");
if(iconvWide2UTF8 == (iconv_t)-1)
{
RDCERR("Couldn't open iconv for WCHAR_T to UTF-8: %d", errno);
return "";
}
char *inbuf = (char *)s.c_str();
size_t insize = (s.length()+1)*sizeof(wchar_t); // include null terminator
char *outbuf = &charBuffer[0];
size_t outsize = len;
ret = iconv(iconvWide2UTF8, &inbuf, &insize, &outbuf, &outsize);
}
if(ret == (size_t)-1)
{
#if !defined(_RELEASE)
RDCWARN("Failed to convert wstring");
#endif
return "";
}
// convert to string from null-terminated string - utf-8 never contains
// 0 bytes before the null terminator, and this way we don't care if
// charBuffer is larger than the string
return string(&charBuffer[0]);
}
};
namespace OSUtility
{
void WriteOutput(int channel, const char *str)
{
if(channel == OSUtility::Output_StdOut)
fprintf(stdout, "%s", str);
else if(channel == OSUtility::Output_StdErr)
fprintf(stderr, "%s", str);
}
};