Exemple #1
0
int DB_init()
{
    apr_pool_t *p = NULL;
    apr_pool_initialize();
    apr_pool_create(&p, NULL);

    if (access(DB_DIR, W_OK | X_OK) == -1) {
        apr_status_t rc = apr_dir_make_recursive(DB_DIR,
            APR_UREAD | APR_UWRITE | APR_UEXECUTE |
            APR_GREAD | APR_GWRITE | APR_GEXECUTE, p);
        check(rc == APR_SUCCESS, "Failed to make database dir: %s", DB_DIR);
    }

    if (access(DB_FILE, W_OK) == -1) {
        FILE *db = DB_open(DB_FILE, "w");
        check(db, "Cannot open database: %s", DB_FILE);
        DB_close(db);
    }

    apr_pool_destroy(p);
    return 0;

error:
    apr_pool_destroy(p);
    return -1;
}
Exemple #2
0
int DB_init()
{

	apr_pool_t *p = NULL;// apr_pool_t被封装了,只能通过API对其进行操作
	apr_pool_initialize();
	apr_pool_create(&p, NULL); // 创建memory pool,返回apr_status_t类型的状态码

	// 确定DB_DIR是否存在、是否有写和执行权限,成功返回0,否则-1
	if(access(DB_DIR, W_OK | X_OK) == -1)
	{
		// 权限测试失败,可能是DB_DIR不存在,尝试创建
		apr_status_t rc = 
			apr_dir_make_recursive(DB_DIR,
								   APR_UREAD | APR_UWRITE | APR_UEXECUTE |
								   APR_GREAD | APR_GWRITE | APR_GEXECUTE, p);
		check(rc == APR_SUCCESS, "Failed to make database dir: %s", DB_DIR);
	}
	// 确定DB_FILE是否存在、是否有写权限
	if(access(DB_FILE, W_OK) == -1)
	{
		// 权限测试失败,可能是DB_FILE不存在,尝试创建
		FILE *db = DB_open(DB_FILE, "w");
		check(db, "Cannot open database: %s", DB_FILE);
		DB_close(db);
	}

	apr_pool_destroy(p);
	return 0;

  error:
	apr_pool_destroy(p);
	return -1;
}
Exemple #3
0
/* 从数据库读取数据,并存入一个字符串中 */
static bstring DB_load()
{
	FILE *db = NULL;
	bstring data = NULL;

	db = DB_open(DB_FILE, "r");
	check(db, "Failed to open database: %s", DB_FILE);

	/*bstring bread (bNread readPtr, void * parm)
	 *
	 *  Use a finite buffer fread-like function readPtr to create a bstring 
	 *  filled with the entire contents of file-like source data in a roughly 
	 *  efficient way.
	 */
	data = bread((bNread)fread, db);
	check(data, "Failed to read from db file: %s", DB_FILE);

	DB_close(db);
	return data;

  error:
	if(db) DB_close(db);
	if(data) bdestroy(data);
	return NULL;
}
Exemple #4
0
int DB_init()
{
	apr_pool_t *p = NULL;
	apr_pool_initialize();
	apr_pool_create(&p, NULL);

	if(access(DB_DIR, W_OK|X_OK) == -1){
		apr_status_t rc = apr_dir_make_recursive(DB_DIR,
				APR_UREAD|APR_UWRITE|APR_UEXECUTE |
				APR_GREAD|APR_GWRITE|APR_GEXECUTE, p);
		check(rc == APR_SUCCESS, "Could not open %s.", DB_DIR);
	} else {
		debug("Access to %s is okay", DB_DIR);
	}

	if(access(DB_FILE, W_OK) == -1){
		FILE *db = DB_open(DB_FILE, "w");
		check(db, "Could not open %s.", DB_FILE);
		DB_close(db);
	} else {
		debug("No need to create db file %s", DB_FILE);
	}

	apr_pool_destroy(p);
	return 0;

error:
	apr_pool_destroy(p);
	return -1;
}
Exemple #5
0
int DB_init()
{
    apr_status_t rv;
    apr_pool_t *mp;

    rv = apr_initialize();
    assert(rv == APR_SUCCESS);

    apr_pool_create(&mp, NULL);

    if (access(DB_DIR, W_OK | X_OK) == -1) {
        apr_status_t rc = apr_dir_make_recursive(
            DB_DIR,
            APR_UREAD | APR_UWRITE | APR_UEXECUTE | APR_GREAD | APR_GWRITE | APR_GEXECUTE, 
            mp
        );

        assert(rc == APR_SUCCESS);
    }
    
    if (access(DB_FILE, W_OK) == -1) {
        FILE *db = DB_open(DB_FILE, "w");
        assert(db != NULL);
        DB_close(db);
    }
    
    apr_pool_destroy(mp);
    apr_terminate();

    return 0;
}
Exemple #6
0
static bstring DB_load() {
    FILE *db = NULL;
    bstring data = NULL;
    db = DB_open(DB_FILE, "r");
    check(db, "Failed to open database: %s", DB_FILE);
    data = bread((bNread)fread, db);
    check(data, "Failed to read from db file: %s", DB_FILE);
    DB_close(db);
    return data;
error:
    if(db) DB_close(db);
    if(data) bdestroy(data);
    return NULL;
}
Exemple #7
0
static bstring DB_load()
{
    FILE *db = NULL;
    bstring data = NULL;
    
    db = DB_open(DB_FILE, "r");
    assert(db != NULL);
    
    data = bread((bNread)fread, db);
    assert(data != NULL);
    
    DB_close(db);

    return data;
}
Exemple #8
0
int DB_update(const char *url) {
    if(DB_find(url)) {
        log_info("Already recorded as installed: %s", url);
    }
    FILE *db = DB_open(DB_FILE, "a+");
    check(db, "Failed to open DB file: %s", DB_FILE);
    bstring line = bfromcstr(url);
    bconchar(line, '\n');
    int rc = fwrite(line->data, blength(line), 1, db);
    check(rc == 1, "Failed to append to the db.");
    return 0;
error:
    if(db) DB_close(db);
    return -1;
}
Exemple #9
0
static bstring DB_load()
{
	FILE *db = NULL;
	bstring data = NULL;

	db = DB_open(DB_FILE, "r");
	check(db, "Failed to open %s.", DB_FILE);

	data = bread((bNread)fread, db);
	check(data, "Couldn't read %s.", DB_FILE);

	DB_close(db);
	return data;

error:
	if(db) DB_close(db);
	if(data) bdestroy(data);
	return NULL;
}
Exemple #10
0
int DB_update(const char *url)
{
    bool found = false;
    assert(DB_find(url, &found) == 0);
    
    if (found) {
        printf("INFO: Already recorded as installed: %s\n", url);
    }
    
    FILE *db = DB_open(DB_FILE, "a+");
    assert(db != NULL);
    
    bstring line = bfromcstr(url);
    bconchar(line, '\n');
    
    int rc = fwrite(line->data, blength(line), 1, db);
    assert(rc == 1);
    
    bdestroy(line);
    
    DB_close(db);
    
    return 0;
}