Home - sourcemod-central - file revision - core/LibrarySys.cpp@2c7137f077fe Mercurial
summary | shortlog | changelog | graph | tags | branches | files | changeset | file | latest | revisions | annotate | diff | raw | help
core/LibrarySys.cpp
author David Anderson <dvander@alliedmods.net>
Fri Oct 02 03:47:42 2009 -0700 (2009-10-02)
changeset 2815 2c7137f077fe
parent 2685 aacf5c8b3ac2
child 3027 0f90dd9754ca
permissions -rw-r--r--
Re-enabled PDB files. 
     1 /**
     2  * vim: set ts=4 :
     3  * =============================================================================
     4  * SourceMod
     5  * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.
     6  * =============================================================================
     7  *
     8  * This program is free software; you can redistribute it and/or modify it under
     9  * the terms of the GNU General Public License, version 3.0, as published by the
    10  * Free Software Foundation.
    11  * 
    12  * This program is distributed in the hope that it will be useful, but WITHOUT
    13  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
    14  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
    15  * details.
    16  *
    17  * You should have received a copy of the GNU General Public License along with
    18  * this program.  If not, see <http://www.gnu.org/licenses/>.
    19  *
    20  * As a special exception, AlliedModders LLC gives you permission to link the
    21  * code of this program (as well as its derivative works) to "Half-Life 2," the
    22  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
    23  * by the Valve Corporation.  You must obey the GNU General Public License in
    24  * all respects for all other code used.  Additionally, AlliedModders LLC grants
    25  * this exception to all derivative works.  AlliedModders LLC defines further
    26  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
    27  * or <http://www.sourcemod.net/license.php>.
    28  *
    29  * Version: $Id$
    30  */
    31 
    32 #include <sys/stat.h>
    33 #include <stdio.h>
    34 #include <stdarg.h>
    35 #include <string.h>
    36 #include <sm_platform.h>
    37 #include "sm_stringutil.h"
    38 #include "LibrarySys.h"
    39 
    40 LibrarySystem g_LibSys;
    41 
    42 CLibrary::~CLibrary()
    43 {
    44 	if (m_lib)
    45 	{
    46 #if defined PLATFORM_WINDOWS
    47 		FreeLibrary(m_lib);
    48 #elif defined PLATFORM_POSIX
    49 		dlclose(m_lib);
    50 #endif
    51 		m_lib = NULL;
    52 	}
    53 }
    54 
    55 CLibrary::CLibrary(LibraryHandle me)
    56 {
    57 	m_lib = me;
    58 }
    59 
    60 void CLibrary::CloseLibrary()
    61 {
    62 	delete this;
    63 }
    64 
    65 void *CLibrary::GetSymbolAddress(const char *symname)
    66 {
    67 #if defined PLATFORM_WINDOWS
    68 	return GetProcAddress(m_lib, symname);
    69 #elif defined PLATFORM_POSIX
    70 	return dlsym(m_lib, symname);
    71 #endif
    72 }
    73 
    74 
    75 /********************
    76  ** Directory Code **
    77  ********************/
    78 
    79 
    80 CDirectory::CDirectory(const char *path)
    81 {
    82 #if defined PLATFORM_WINDOWS
    83 	char newpath[PLATFORM_MAX_PATH];
    84 	snprintf(newpath, sizeof(newpath), "%s\\*.*", path);
    85 	m_dir = FindFirstFileA(newpath, &m_fd);
    86 	if (!IsValid())
    87 	{
    88 		m_fd.cFileName[0] = '\0';
    89 	}
    90 #elif defined PLATFORM_POSIX
    91 	m_dir = opendir(path);
    92 	if (IsValid())
    93 	{
    94 		/* :TODO: we need to read past "." and ".."! */
    95 		ep = readdir(m_dir);
    96 		snprintf(m_origpath, PLATFORM_MAX_PATH, "%s", path);
    97 	} else {
    98 		ep = NULL;
    99 	}
   100 #endif
   101 }
   102 
   103 CDirectory::~CDirectory()
   104 {
   105 	if (IsValid())
   106 	{
   107 #if defined PLATFORM_WINDOWS
   108 		FindClose(m_dir);
   109 #elif defined PLATFORM_POSIX
   110 		closedir(m_dir);
   111 #endif
   112 	}
   113 }
   114 
   115 void CDirectory::NextEntry()
   116 {
   117 #if defined PLATFORM_WINDOWS
   118 	if (FindNextFile(m_dir, &m_fd) == 0)
   119 	{
   120 		FindClose(m_dir);
   121 		m_dir = INVALID_HANDLE_VALUE;
   122 	}
   123 #elif defined PLATFORM_POSIX
   124 	if ((ep=readdir(m_dir)) == NULL)
   125 	{
   126 		closedir(m_dir);
   127 		m_dir = NULL;
   128 	}
   129 #endif
   130 }
   131 
   132 bool CDirectory::IsEntryValid()
   133 {
   134 	return IsValid();
   135 }
   136 
   137 bool CDirectory::IsEntryDirectory()
   138 {
   139 #if defined PLATFORM_WINDOWS
   140 	return ((m_fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == FILE_ATTRIBUTE_DIRECTORY);
   141 #elif defined PLATFORM_POSIX
   142 	char temppath[PLATFORM_MAX_PATH];
   143 	snprintf(temppath, sizeof(temppath), "%s/%s", m_origpath, GetEntryName());
   144 	return g_LibSys.IsPathDirectory(temppath);
   145 #endif
   146 }
   147 
   148 bool CDirectory::IsEntryFile()
   149 {
   150 #if defined PLATFORM_WINDOWS
   151 	return !(m_fd.dwFileAttributes & (FILE_ATTRIBUTE_DIRECTORY|FILE_ATTRIBUTE_DEVICE));
   152 #elif defined PLATFORM_POSIX
   153 	char temppath[PLATFORM_MAX_PATH];
   154 	snprintf(temppath, sizeof(temppath), "%s/%s", m_origpath, GetEntryName());
   155 	return g_LibSys.IsPathFile(temppath);
   156 #endif
   157 }
   158 
   159 const char *CDirectory::GetEntryName()
   160 {
   161 #if defined PLATFORM_WINDOWS
   162 	return m_fd.cFileName;
   163 #elif defined PLATFORM_POSIX
   164 	return ep ? ep->d_name : "";
   165 #endif
   166 }
   167 
   168 bool CDirectory::MoreFiles()
   169 {
   170 	return IsValid();
   171 }
   172 
   173 bool CDirectory::IsValid()
   174 {
   175 #if defined PLATFORM_WINDOWS
   176 	return (m_dir != INVALID_HANDLE_VALUE);
   177 #elif defined PLATFORM_POSIX
   178 	return (m_dir != NULL);
   179 #endif
   180 }
   181 
   182 
   183 /*************************
   184  ** Library System Code **
   185  *************************/
   186 
   187 
   188 bool LibrarySystem::PathExists(const char *path)
   189 {
   190 #if defined PLATFORM_WINDOWS
   191 	DWORD attr = GetFileAttributesA(path);
   192 
   193 	return (attr != INVALID_FILE_ATTRIBUTES);
   194 #elif defined PLATFORM_POSIX
   195 	struct stat s;
   196 
   197 	return (stat(path, &s) == 0);
   198 #endif
   199 }
   200 
   201 bool LibrarySystem::IsPathFile(const char *path)
   202 {
   203 #if defined PLATFORM_WINDOWS
   204 	DWORD attr = GetFileAttributes(path);
   205 
   206 	if (attr == INVALID_FILE_ATTRIBUTES)
   207 	{
   208 		return false;
   209 	}
   210 
   211 	if (attr & (FILE_ATTRIBUTE_DIRECTORY|FILE_ATTRIBUTE_DEVICE))
   212 	{
   213 		return false;
   214 	}
   215 
   216 	return true;
   217 #elif defined PLATFORM_POSIX
   218 	struct stat s;
   219 
   220 	if (stat(path, &s) != 0)
   221 	{
   222 		return false;
   223 	}
   224 
   225 	return S_ISREG(s.st_mode) ? true : false;
   226 #endif
   227 }
   228 
   229 bool LibrarySystem::IsPathDirectory(const char *path)
   230 {
   231 #if defined PLATFORM_WINDOWS
   232 	DWORD attr = GetFileAttributes(path);
   233 
   234 	if (attr == INVALID_FILE_ATTRIBUTES)
   235 	{
   236 		return false;
   237 	}
   238 
   239 	if (attr & FILE_ATTRIBUTE_DIRECTORY)
   240 	{
   241 		return true;
   242 	}
   243 
   244 #elif defined PLATFORM_LINUX
   245 	struct stat s;
   246 
   247 	if (stat(path, &s) != 0)
   248 	{
   249 		return false;
   250 	}
   251 
   252 	if (S_ISDIR(s.st_mode))
   253 	{
   254 		return true;
   255 	}
   256 #endif
   257 
   258 	return false;
   259 }
   260 
   261 IDirectory *LibrarySystem::OpenDirectory(const char *path)
   262 {
   263 	CDirectory *dir = new CDirectory(path);
   264 
   265 	if (!dir->IsValid())
   266 	{
   267 		delete dir;
   268 		return NULL;
   269 	}
   270 
   271 	return dir;
   272 }
   273 
   274 void LibrarySystem::GetPlatformError(char *error, size_t maxlength)
   275 {
   276 #if defined PLATFORM_WINDOWS
   277 	return GetPlatformErrorEx(GetLastError(), error, maxlength);
   278 #elif defined PLATFORM_POSIX
   279 	return GetPlatformErrorEx(errno, error, maxlength);
   280 #endif
   281 }
   282 
   283 void LibrarySystem::GetPlatformErrorEx(int code, char *error, size_t maxlength)
   284 {
   285 	if (error && maxlength)
   286 	{
   287 #if defined PLATFORM_WINDOWS
   288 		if (FormatMessageA(FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS,
   289 						   NULL,
   290 						   (DWORD)code,
   291 						   MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
   292 						   (LPSTR)error,
   293 						   maxlength,
   294 						   NULL) == 0)
   295 		{
   296 			UTIL_Format(error, maxlength, "error code %08x", code);
   297 		}
   298 #elif defined PLATFORM_LINUX
   299 		const char *ae = strerror_r(code, error, maxlength);
   300 		if (ae != error)
   301 		{
   302 			UTIL_Format(error, maxlength, "%s", ae);
   303 		}
   304 #elif defined PLATFORM_POSIX
   305 		strerror_r(code, error, maxlength);
   306 #endif
   307 	}
   308 }
   309 
   310 void LibrarySystem::GetLoaderError(char *buffer, size_t maxlength)
   311 {
   312 #if defined PLATFORM_WINDOWS
   313 	GetPlatformError(buffer, maxlength);
   314 #elif defined PLATFORM_POSIX
   315 	if (buffer != NULL && maxlength)
   316 	{
   317 		strncopy(buffer, dlerror(), maxlength);
   318 	}
   319 #endif
   320 }
   321 
   322 void LibrarySystem::CloseDirectory(IDirectory *dir)
   323 {
   324 	delete dir;
   325 }
   326 
   327 ILibrary *LibrarySystem::OpenLibrary(const char *path, char *error, size_t maxlength)
   328 {
   329 	LibraryHandle lib;
   330 #if defined PLATFORM_WINDOWS
   331 	lib = LoadLibraryA(path);
   332 #elif defined PLATFORM_POSIX
   333 	lib = dlopen(path, RTLD_NOW);
   334 #endif
   335 
   336 	if (lib == NULL)
   337 	{
   338 		GetLoaderError(error, maxlength);
   339 		return NULL;
   340 	}
   341 
   342 	return new CLibrary(lib);
   343 }
   344 
   345 size_t LibrarySystem::PathFormat(char *buffer, size_t len, const char *fmt, ...)
   346 {
   347 	va_list ap;
   348 	va_start(ap, fmt);
   349 	size_t mylen = vsnprintf(buffer, len, fmt, ap);
   350 	va_end(ap);
   351 
   352 	if (mylen >= len)
   353 	{
   354 		mylen = len - 1;
   355 		buffer[mylen] = '\0';
   356 	}
   357 
   358 	for (size_t i=0; i<mylen; i++)
   359 	{
   360 		if (buffer[i] == PLATFORM_SEP_ALTCHAR)
   361 		{
   362 			buffer[i] = PLATFORM_SEP_CHAR;
   363 		}
   364 	}
   365 
   366 	return mylen;
   367 }
   368 
   369 const char *LibrarySystem::GetFileExtension(const char *filename)
   370 {
   371 	size_t len, end;
   372 
   373 	len = strlen(filename);
   374 
   375 	/* Minimum string length for filename with ext would be 3; example: a.a */
   376 	if (len < 3)
   377 	{
   378 		return NULL;
   379 	}
   380 
   381 	end = len - 1;
   382 
   383 	for (size_t i = end; i <= end; i--)
   384 	{
   385 		if (filename[i] == PLATFORM_SEP_CHAR || filename[i] == PLATFORM_SEP_ALTCHAR)
   386 		{
   387 			break;
   388 		}
   389 
   390 		if (filename[i] == '.' && i != end && i != 0)
   391 		{
   392 			return &filename[++i];
   393 		}
   394 	}
   395 
   396 	return NULL;
   397 }
   398 
   399 bool LibrarySystem::CreateFolder(const char *path)
   400 {
   401 #if defined PLATFORM_WINDOWS
   402 	return (mkdir(path) != -1);
   403 #elif defined PLATFORM_POSIX
   404 	return (mkdir(path, 0775) != -1);
   405 #endif
   406 }
   407 
   408 size_t LibrarySystem::GetFileFromPath(char *buffer, size_t maxlength, const char *path)
   409 {
   410 	size_t length = strlen(path);
   411 
   412 	for (size_t i = length - 1;
   413 		 i >= 0 && i <= length - 1;
   414 		 i--)
   415 	{
   416 		if (path[i] == '/' 
   417 #if defined PLATFORM_WINDOWS
   418 			|| path[i] == '\\'
   419 #endif
   420 			)
   421 		{
   422 			return UTIL_Format(buffer, maxlength, "%s", &path[i+1]);
   423 		}
   424 	}
   425 
   426 	/* We scanned and found no path separator */
   427 	return UTIL_Format(buffer, maxlength, "%s", path);
   428 }
   429 
   430 bool LibrarySystem::FileTime(const char *path, FileTimeType type, time_t *pTime)
   431 {
   432 #ifdef PLATFORM_WINDOWS
   433 	struct _stat s;
   434 	if (_stat(path, &s) != 0)
   435 #elif defined PLATFORM_POSIX
   436 	struct stat s;
   437 	if (stat(path, &s) != 0)
   438 #endif
   439 	{
   440 		return false;
   441 	}
   442 
   443 	if (type == FileTime_LastAccess)
   444 	{
   445 		*pTime = s.st_atime;
   446 	}
   447 	else if (type == FileTime_Created)
   448 	{
   449 		*pTime = s.st_ctime;
   450 	}
   451 	else if (type == FileTime_LastChange)
   452 	{
   453 		*pTime = s.st_mtime;
   454 	}
   455 
   456 	return true;
   457 }
sourcemod-central
Feed