common.cpp

linux 下 源代码测试系统 用 tar 打开

/*
 * $File: common.cpp
 * $Author: Jiakai -- [e-mail]
 * $Date: Thu Mar 26 20:21:07 2009
 */
/*
Copyright (C) (2008) (Jiakai) <[e-mail]>

This program is free software; you can redistribute it and/or 
modify it under the terms of the GNU General Public License 
as published by the Free Software Foundation; either 
version 2 of the License, or any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

*/

#define DEBUG_MODE
#define NEED_LIST_DIR
#include "common.h"

#include <sys/types.h>
#include <dirent.h>

#include <cstring>
#include <cstdio>
using namespace std;

bool list_dir(const char *dirpath, vector<string> &ans)
{
	ans.clear();
	DIR *dir = opendir(dirpath);
	if (dir == NULL)
		return false;
	dirent *pdirent;
	while ((pdirent = readdir(dir)) != NULL)
	{
		if (pdirent->d_type != DT_DIR || pdirent->d_name[0] == '.')
			continue;
		ans.push_back(string(pdirent->d_name));
	}
	closedir(dir);
	return true;
}

void write_debug_msg(const char *msg)
{
	static int count = 0;
	FILE *file = fopen("debug.log", "a");
	fprintf(file, "[string %d]%s \n", count ++, msg);
	fclose(file);
}

void write_debug_msg(int msg)
{
	static int count = 0;
	FILE *file = fopen("debug.log", "a");
	fprintf(file, "[string %d]%d \n", count ++, msg);
	fclose(file);
}