Fetching contributors…
Cannot retrieve contributors at this time
113 lines (93 sloc) 2.76 KB
/* -*- Mode: C; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
/*
* CDDL HEADER START
*
* The contents of this file are subject to the terms of the
* Common Development and Distribution License (the "License").
* You may not use this file except in compliance with the License.
*
* See LICENSE.txt included in this distribution for the specific
* language governing permissions and limitations under the License.
*
* When distributing Covered Code, include this CDDL HEADER in each
* file and include the License file at LICENSE.txt.
* If applicable, add the following below this CDDL HEADER, with the
* fields enclosed by brackets "[]" replaced with your own identifying
* information: Portions Copyright [yyyy] [name of copyright owner]
*
* CDDL HEADER END
*/
/*
* Copyright 2010 Trond Norbye
*/
#include "config.h"
#include <unistd.h>
#include <sys/fcntl.h>
#include <ctype.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "vbucket.h"
#ifdef HAVE_LIBVBUCKET
VBUCKET_CONFIG_HANDLE vbucket_handle;
bool initialize_vbuckets(const char *location)
{
if (access(location, F_OK) == 0) {
vbucket_handle = vbucket_config_parse_file(location);
if (vbucket_handle == NULL) {
fprintf(stderr, "Failed to parse vbucket config: %s\n",
vbucket_get_error());
}
return vbucket_handle != NULL;
}
const char *p = location;
while (isalpha(*p) || isdigit(*p) || *p == '.' || *p == ':') {
++p;
}
if (*p) {
fprintf(stderr, "%s is not a file, and doesn't look like a URL to me\n",
location);
return false;
}
char cmd[2048];
snprintf(cmd, sizeof(cmd),
"wget -O - http://%s/pools/default/buckets/default", location);
fprintf(stdout, "Downloading vbucket config from %s\n",
location);
fflush(stdout);
FILE *in = popen(cmd, "r");
char *config = malloc(8196);
*config = '\0';
while (fgets(cmd, sizeof(cmd), in) != NULL) {
strcat(config, cmd);
}
pclose(in);
if (strlen(config) == 0) {
return false;
}
vbucket_handle = vbucket_config_parse_string(config);
if (vbucket_handle == NULL) {
fprintf(stderr, "Failed to parse vbucket config: %s\n",
vbucket_get_error());
fprintf(stderr, "Config: [%s]", config);
}
return vbucket_handle != NULL;
}
uint16_t get_vbucket(const char *key, size_t nkey) {
if (vbucket_handle) {
return vbucket_get_vbucket_by_key(vbucket_handle, key, nkey);
}
return 0;
}
#else
bool initialize_vbuckets(const char *location)
{
(void)location;
return false;
}
uint16_t get_vbucket(const char *key, size_t nkey) {
(void)key;
(void)nkey;
return 0;
}
#endif