mirror of https://github.com/schoebel/mars
152 lines
3.7 KiB
C
152 lines
3.7 KiB
C
/*
|
|
* MARS Long Distance Replication Software
|
|
*
|
|
* This file is part of MARS project: http://schoebel.github.io/mars/
|
|
*
|
|
* Copyright (C) 2010-2014 Thomas Schoebel-Theuer
|
|
* Copyright (C) 2011-2014 1&1 Internet AG
|
|
*
|
|
* 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
|
|
* (at your option) 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.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
|
|
//#define BRICK_DEBUGGING
|
|
//#define MARS_DEBUGGING
|
|
//#define IO_DEBUGGING
|
|
|
|
#include <linux/kernel.h>
|
|
#include <linux/module.h>
|
|
#include <linux/string.h>
|
|
|
|
#include "strategy.h"
|
|
#include "../mars_net.h"
|
|
|
|
static
|
|
char *_mars_translate_hostname(const char *name)
|
|
{
|
|
char *res = brick_strdup(name);
|
|
char *tmp;
|
|
char *trans;
|
|
|
|
for (tmp = res; *tmp; tmp++) {
|
|
if (*tmp == ':') {
|
|
*tmp = '\0';
|
|
break;
|
|
}
|
|
}
|
|
|
|
tmp = path_make("/mars/ips/ip-%s", res);
|
|
if (unlikely(!tmp)) {
|
|
goto done;
|
|
}
|
|
|
|
trans = ordered_readlink(tmp, NULL);
|
|
if (trans && trans[0]) {
|
|
MARS_DBG("'%s' => '%s'\n", tmp, trans);
|
|
brick_string_free(res);
|
|
res = trans;
|
|
} else {
|
|
MARS_DBG("no translation for '%s'\n", tmp);
|
|
brick_string_free(trans);
|
|
}
|
|
brick_string_free(tmp);
|
|
done:
|
|
return res;
|
|
}
|
|
|
|
int mars_send_dent_list(struct mars_global *global, struct mars_socket *sock)
|
|
{
|
|
struct list_head *anchor;
|
|
struct list_head *tmp;
|
|
struct mars_dent *dent;
|
|
int status = 0;
|
|
|
|
down_read(&global->dent_mutex);
|
|
anchor = &global->dent_anchor;
|
|
for (tmp = anchor->next; tmp != anchor; tmp = tmp->next) {
|
|
dent = container_of(tmp, struct mars_dent, dent_link);
|
|
dent->d_proto = MARS_PROTO_LEVEL;
|
|
status = mars_send_struct(sock, dent, mars_dent_meta, true);
|
|
if (status < 0)
|
|
break;
|
|
}
|
|
if (status >= 0) { // send EOR
|
|
status = mars_send_struct(sock, NULL, mars_dent_meta, false);
|
|
}
|
|
up_read(&global->dent_mutex);
|
|
return status;
|
|
}
|
|
EXPORT_SYMBOL_GPL(mars_send_dent_list);
|
|
|
|
int mars_recv_dent_list(struct mars_global *global, struct mars_socket *sock)
|
|
{
|
|
struct list_head *anchor;
|
|
int status;
|
|
|
|
down_write(&global->dent_mutex);
|
|
anchor = &global->dent_anchor;
|
|
for (;;) {
|
|
struct mars_dent *dent = brick_zmem_alloc(sizeof(struct mars_dent));
|
|
struct list_head *hash_anchor;
|
|
unsigned int hash;
|
|
|
|
if (!dent)
|
|
return -ENOMEM;
|
|
|
|
//MARS_IO("\n");
|
|
|
|
INIT_LIST_HEAD(&dent->dent_link);
|
|
INIT_LIST_HEAD(&dent->dent_hash_link);
|
|
INIT_LIST_HEAD(&dent->brick_list);
|
|
|
|
status = mars_recv_struct(sock, dent, mars_dent_meta);
|
|
if (status <= 0) {
|
|
mars_free_dent(NULL, dent);
|
|
goto done;
|
|
}
|
|
if (dent->d_proto >= 2) {
|
|
sock->s_remote_proto_level = dent->d_proto;
|
|
sock->s_common_proto_level = min(dent->d_proto, MARS_PROTO_LEVEL);
|
|
}
|
|
list_add_tail(&dent->dent_link, anchor);
|
|
if (!dent->d_path || dent->d_unordered)
|
|
continue;
|
|
hash = dent_hash(dent->d_path, strlen(dent->d_path));
|
|
dent->d_hash = hash;
|
|
hash_anchor = DENT_HASH_ANCHOR(global, hash);
|
|
list_add_tail(&dent->dent_hash_link, hash_anchor);
|
|
}
|
|
done:
|
|
up_write(&global->dent_mutex);
|
|
return status;
|
|
}
|
|
EXPORT_SYMBOL_GPL(mars_recv_dent_list);
|
|
|
|
|
|
////////////////// module init stuff /////////////////////////
|
|
|
|
|
|
int __init init_sy_net(void)
|
|
{
|
|
MARS_INF("init_sy_net()\n");
|
|
mars_translate_hostname = _mars_translate_hostname;
|
|
return 0;
|
|
}
|
|
|
|
void exit_sy_net(void)
|
|
{
|
|
MARS_INF("exit_sy_net()\n");
|
|
}
|