lldir_linux.cpp
上传用户:king477883
上传日期:2021-03-01
资源大小:9553k
文件大小:9k
源码类别:

游戏引擎

开发平台:

C++ Builder

  1. /** 
  2.  * @file lldir_linux.cpp
  3.  * @brief Implementation of directory utilities for linux
  4.  *
  5.  * $LicenseInfo:firstyear=2002&license=viewergpl$
  6.  * 
  7.  * Copyright (c) 2002-2010, Linden Research, Inc.
  8.  * 
  9.  * Second Life Viewer Source Code
  10.  * The source code in this file ("Source Code") is provided by Linden Lab
  11.  * to you under the terms of the GNU General Public License, version 2.0
  12.  * ("GPL"), unless you have obtained a separate licensing agreement
  13.  * ("Other License"), formally executed by you and Linden Lab.  Terms of
  14.  * the GPL can be found in doc/GPL-license.txt in this distribution, or
  15.  * online at http://secondlifegrid.net/programs/open_source/licensing/gplv2
  16.  * 
  17.  * There are special exceptions to the terms and conditions of the GPL as
  18.  * it is applied to this Source Code. View the full text of the exception
  19.  * in the file doc/FLOSS-exception.txt in this software distribution, or
  20.  * online at
  21.  * http://secondlifegrid.net/programs/open_source/licensing/flossexception
  22.  * 
  23.  * By copying, modifying or distributing this software, you acknowledge
  24.  * that you have read and understood your obligations described above,
  25.  * and agree to abide by those obligations.
  26.  * 
  27.  * ALL LINDEN LAB SOURCE CODE IS PROVIDED "AS IS." LINDEN LAB MAKES NO
  28.  * WARRANTIES, EXPRESS, IMPLIED OR OTHERWISE, REGARDING ITS ACCURACY,
  29.  * COMPLETENESS OR PERFORMANCE.
  30.  * $/LicenseInfo$
  31.  */
  32. #include "linden_common.h"
  33. #include "lldir_linux.h"
  34. #include "llerror.h"
  35. #include "llrand.h"
  36. #include <sys/types.h>
  37. #include <sys/stat.h>
  38. #include <unistd.h>
  39. #include <glob.h>
  40. #include <pwd.h>
  41. static std::string getCurrentUserHome(char* fallback)
  42. {
  43. const uid_t uid = getuid();
  44. struct passwd *pw;
  45. char *result_cstr = fallback;
  46. pw = getpwuid(uid);
  47. if ((pw != NULL) && (pw->pw_dir != NULL))
  48. {
  49. result_cstr = (char*) pw->pw_dir;
  50. }
  51. else
  52. {
  53. llinfos << "Couldn't detect home directory from passwd - trying $HOME" << llendl;
  54. const char *const home_env = getenv("HOME"); /* Flawfinder: ignore */ 
  55. if (home_env)
  56. {
  57. result_cstr = (char*) home_env;
  58. }
  59. else
  60. {
  61. llwarns << "Couldn't detect home directory!  Falling back to " << fallback << llendl;
  62. }
  63. }
  64. return std::string(result_cstr);
  65. }
  66. LLDir_Linux::LLDir_Linux()
  67. {
  68. mDirDelimiter = "/";
  69. mCurrentDirIndex = -1;
  70. mCurrentDirCount = -1;
  71. mDirp = NULL;
  72. char tmp_str[LL_MAX_PATH]; /* Flawfinder: ignore */ 
  73. if (getcwd(tmp_str, LL_MAX_PATH) == NULL)
  74. {
  75. strcpy(tmp_str, "/tmp");
  76. llwarns << "Could not get current directory; changing to "
  77. << tmp_str << llendl;
  78. if (chdir(tmp_str) == -1)
  79. {
  80. llerrs << "Could not change directory to " << tmp_str << llendl;
  81. }
  82. }
  83. mExecutableFilename = "";
  84. mExecutablePathAndName = "";
  85. mExecutableDir = tmp_str;
  86. mWorkingDir = tmp_str;
  87. #ifdef APP_RO_DATA_DIR
  88. mAppRODataDir = APP_RO_DATA_DIR;
  89. #else
  90. mAppRODataDir = tmp_str;
  91. #endif
  92.     U32 indra_pos = mExecutableDir.find("/indra");
  93.     if (indra_pos != std::string::npos)
  94.     {
  95. // ...we're in a dev checkout
  96. mSkinBaseDir = mExecutableDir.substr(0, indra_pos) + "/indra/newview/skins";
  97. llinfos << "Running in dev checkout with mSkinBaseDir "
  98.  << mSkinBaseDir << llendl;
  99.     }
  100.     else
  101.     {
  102. // ...normal installation running
  103. mSkinBaseDir = mAppRODataDir + mDirDelimiter + "skins";
  104.     }
  105. mOSUserDir = getCurrentUserHome(tmp_str);
  106. mOSUserAppDir = "";
  107. mLindenUserDir = "";
  108. char path [32]; /* Flawfinder: ignore */ 
  109. // *NOTE: /proc/%d/exe doesn't work on FreeBSD. But that's ok,
  110. // because this is the linux implementation.
  111. snprintf (path, sizeof(path), "/proc/%d/exe", (int) getpid ()); 
  112. int rc = readlink (path, tmp_str, sizeof (tmp_str)-1); /* Flawfinder: ignore */ 
  113. if ( (rc != -1) && (rc <= ((int) sizeof (tmp_str)-1)) )
  114. {
  115. tmp_str[rc] = ''; //readlink() doesn't 0-terminate the buffer
  116. mExecutablePathAndName = tmp_str;
  117. char *path_end;
  118. if ((path_end = strrchr(tmp_str,'/')))
  119. {
  120. *path_end = '';
  121. mExecutableDir = tmp_str;
  122. mWorkingDir = tmp_str;
  123. mExecutableFilename = path_end+1;
  124. }
  125. else
  126. {
  127. mExecutableFilename = tmp_str;
  128. }
  129. }
  130. mLLPluginDir = mExecutableDir + mDirDelimiter + "llplugin";
  131. // *TODO: don't use /tmp, use $HOME/.secondlife/tmp or something.
  132. mTempDir = "/tmp";
  133. }
  134. LLDir_Linux::~LLDir_Linux()
  135. {
  136. }
  137. // Implementation
  138. void LLDir_Linux::initAppDirs(const std::string &app_name,
  139.   const std::string& app_read_only_data_dir)
  140. {
  141. // Allow override so test apps can read newview directory
  142. if (!app_read_only_data_dir.empty())
  143. {
  144. mAppRODataDir = app_read_only_data_dir;
  145. mSkinBaseDir = mAppRODataDir + mDirDelimiter + "skins";
  146. }
  147. mAppName = app_name;
  148. std::string upper_app_name(app_name);
  149. LLStringUtil::toUpper(upper_app_name);
  150. char* app_home_env = getenv((upper_app_name + "_USER_DIR").c_str()); /* Flawfinder: ignore */ 
  151. if (app_home_env)
  152. {
  153. // user has specified own userappdir i.e. $SECONDLIFE_USER_DIR
  154. mOSUserAppDir = app_home_env;
  155. }
  156. else
  157. {
  158. // traditionally on unixoids, MyApp gets ~/.myapp dir for data
  159. mOSUserAppDir = mOSUserDir;
  160. mOSUserAppDir += "/";
  161. mOSUserAppDir += ".";
  162. std::string lower_app_name(app_name);
  163. LLStringUtil::toLower(lower_app_name);
  164. mOSUserAppDir += lower_app_name;
  165. }
  166. // create any directories we expect to write to.
  167. int res = LLFile::mkdir(mOSUserAppDir);
  168. if (res == -1)
  169. {
  170. if (errno != EEXIST)
  171. {
  172. llwarns << "Couldn't create app user dir " << mOSUserAppDir << llendl;
  173. llwarns << "Default to base dir" << mOSUserDir << llendl;
  174. mOSUserAppDir = mOSUserDir;
  175. }
  176. }
  177. res = LLFile::mkdir(getExpandedFilename(LL_PATH_LOGS,""));
  178. if (res == -1)
  179. {
  180. if (errno != EEXIST)
  181. {
  182. llwarns << "Couldn't create LL_PATH_LOGS dir " << getExpandedFilename(LL_PATH_LOGS,"") << llendl;
  183. }
  184. }
  185. res = LLFile::mkdir(getExpandedFilename(LL_PATH_USER_SETTINGS,""));
  186. if (res == -1)
  187. {
  188. if (errno != EEXIST)
  189. {
  190. llwarns << "Couldn't create LL_PATH_USER_SETTINGS dir " << getExpandedFilename(LL_PATH_USER_SETTINGS,"") << llendl;
  191. }
  192. }
  193. res = LLFile::mkdir(getExpandedFilename(LL_PATH_CACHE,""));
  194. if (res == -1)
  195. {
  196. if (errno != EEXIST)
  197. {
  198. llwarns << "Couldn't create LL_PATH_CACHE dir " << getExpandedFilename(LL_PATH_CACHE,"") << llendl;
  199. }
  200. }
  201. mCAFile = getExpandedFilename(LL_PATH_APP_SETTINGS, "CA.pem");
  202. }
  203. U32 LLDir_Linux::countFilesInDir(const std::string &dirname, const std::string &mask)
  204. {
  205. U32 file_count = 0;
  206. glob_t g;
  207. std::string tmp_str;
  208. tmp_str = dirname;
  209. tmp_str += mask;
  210. if(glob(tmp_str.c_str(), GLOB_NOSORT, NULL, &g) == 0)
  211. {
  212. file_count = g.gl_pathc;
  213. globfree(&g);
  214. }
  215. return (file_count);
  216. }
  217. // get the next file in the directory
  218. // automatically wrap if we've hit the end
  219. BOOL LLDir_Linux::getNextFileInDir(const std::string &dirname, const std::string &mask, std::string &fname, BOOL wrap)
  220. {
  221. glob_t g;
  222. BOOL result = FALSE;
  223. fname = "";
  224. if(!(dirname == mCurrentDir))
  225. {
  226. // different dir specified, close old search
  227. mCurrentDirIndex = -1;
  228. mCurrentDirCount = -1;
  229. mCurrentDir = dirname;
  230. }
  231. std::string tmp_str;
  232. tmp_str = dirname;
  233. tmp_str += mask;
  234. if(glob(tmp_str.c_str(), GLOB_NOSORT, NULL, &g) == 0)
  235. {
  236. if(g.gl_pathc > 0)
  237. {
  238. if((int)g.gl_pathc != mCurrentDirCount)
  239. {
  240. // Number of matches has changed since the last search, meaning a file has been added or deleted.
  241. // Reset the index.
  242. mCurrentDirIndex = -1;
  243. mCurrentDirCount = g.gl_pathc;
  244. }
  245. mCurrentDirIndex++;
  246. if((mCurrentDirIndex >= (int)g.gl_pathc) && wrap)
  247. {
  248. mCurrentDirIndex = 0;
  249. }
  250. if(mCurrentDirIndex < (int)g.gl_pathc)
  251. {
  252. // llinfos << "getNextFileInDir: returning number " << mCurrentDirIndex << ", path is " << g.gl_pathv[mCurrentDirIndex] << llendl;
  253. // The API wants just the filename, not the full path.
  254. //fname = g.gl_pathv[mCurrentDirIndex];
  255. char *s = strrchr(g.gl_pathv[mCurrentDirIndex], '/');
  256. if(s == NULL)
  257. s = g.gl_pathv[mCurrentDirIndex];
  258. else if(s[0] == '/')
  259. s++;
  260. fname = s;
  261. result = TRUE;
  262. }
  263. }
  264. globfree(&g);
  265. }
  266. return(result);
  267. }
  268. // get a random file in the directory
  269. // automatically wrap if we've hit the end
  270. void LLDir_Linux::getRandomFileInDir(const std::string &dirname, const std::string &mask, std::string &fname)
  271. {
  272. S32 num_files;
  273. S32 which_file;
  274. DIR *dirp;
  275. dirent *entryp = NULL;
  276. fname = "";
  277. num_files = countFilesInDir(dirname,mask);
  278. if (!num_files)
  279. {
  280. return;
  281. }
  282. which_file = ll_rand(num_files);
  283. // llinfos << "Random select file #" << which_file << llendl;
  284.     // which_file now indicates the (zero-based) index to which file to play
  285. if (!((dirp = opendir(dirname.c_str()))))
  286. {
  287. while (which_file--)
  288. {
  289. if (!((entryp = readdir(dirp))))
  290. {
  291. return;
  292. }
  293. }    
  294. if ((!which_file) && entryp)
  295. {
  296. fname = entryp->d_name;
  297. }
  298. closedir(dirp);
  299. }
  300. }
  301. std::string LLDir_Linux::getCurPath()
  302. {
  303. char tmp_str[LL_MAX_PATH]; /* Flawfinder: ignore */ 
  304. if (getcwd(tmp_str, LL_MAX_PATH) == NULL)
  305. {
  306. llwarns << "Could not get current directory" << llendl;
  307. tmp_str[0] = '';
  308. }
  309. return tmp_str;
  310. }
  311. BOOL LLDir_Linux::fileExists(const std::string &filename) const
  312. {
  313. struct stat stat_data;
  314. // Check the age of the file
  315. // Now, we see if the files we've gathered are recent...
  316. int res = stat(filename.c_str(), &stat_data);
  317. if (!res)
  318. {
  319. return TRUE;
  320. }
  321. else
  322. {
  323. return FALSE;
  324. }
  325. }
  326. /*virtual*/ std::string LLDir_Linux::getLLPluginLauncher()
  327. {
  328. return gDirUtilp->getExecutableDir() + gDirUtilp->getDirDelimiter() +
  329. "SLPlugin";
  330. }
  331. /*virtual*/ std::string LLDir_Linux::getLLPluginFilename(std::string base_name)
  332. {
  333. return gDirUtilp->getLLPluginDir() + gDirUtilp->getDirDelimiter() +
  334. "lib" + base_name + ".so";
  335. }