Macintosh-SW
/
CAP
mirror of https://github.com/mabam/CAP.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
CAP/applications/aufs/afpidsrvr.c

831 lines
15 KiB
C
Raw Blame History

/*
* $Author: djh $ $Date: 1996/06/19 10:32:13 $
* $Header: /mac/src/cap60/applications/aufs/RCS/afpidsrvr.c,v 2.5 1996/06/19 10:32:13 djh Rel djh $
* $Revision: 2.5 $
*
*/
/*
* Server to provide AUFS fixed directory ID database write facilities
*
* John Forrest <jf@ap.co.umist.ac.uk>
*
*/
#ifdef FIXED_DIRIDS
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/param.h>
#include <sys/time.h>
#include <sys/ioctl.h>
#include <netinet/in.h>
#include <netdb.h>
#include <errno.h>
#include <stdio.h>
#include <signal.h>
#include <fcntl.h>
#ifdef USE_GDBM
#include <gdbm.h>
#else /* USE_GDBM */
#include <ndbm.h>
#endif /* USE_GDBM */
#include <assert.h>
#ifdef USESTRINGDOTH
# include <string.h>
#else USESTRINGDOTH
# include <strings.h>
#endif USESTRINGDOTH
#include "../../lib/afp/afpidaufs.h"
int verbose = 0;
int disconnect = 1;
char *log = NULL;
int continue_clean;
extern char *optarg;
extern int optind, opterr;
int queries; /* request socket */
int bound_queries = 0; /* whether to delete socket or not */
void create_queries();
void session();
void set_signal();
void do_disconnect();
void clean_entries();
void open_database();
void
fatal(message)
char *message;
{
if (message != NULL)
fprintf(stderr, "%s:", message);
if (errno > 0) {
fprintf(stderr, " %d: ", errno);
perror("");
} else
putc('\n', stderr);
#ifdef USE_GDBM
if (db != NULL)
gdbm_close(db);
#else /* USE_GDBM */
if (db != NULL)
dbm_close(db);
#endif /* USE_GDBM */
if (bound_queries)
unlink(aufsSockname);
exit(-1);
}
void
fatal2(message, a1)
char *message, *a1;
{
fprintf(stderr, "afpidsrvr: fatal error: %s%s :", message, a1);
fatal(NULL);
}
void
clean_exit(n)
int n;
{
#ifdef USE_GDBM
if (db != NULL)
gdbm_close(db);
#else /* USE_GDBM */
if (db != NULL)
dbm_close(db);
#endif /* USE_GDBM */
if (bound_queries)
unlink(aufsSockname);
exit(n);
}
main(argc, argv)
char *argv[];
{
int c;
extern char *optarg;
int doClean = 0;
while ((c = getopt(argc, argv, "vtcl:")) != -1) {
switch (c) {
case 'v':
verbose = 1;
break;
case 't':
disconnect = 0;
break;
case 'l':
log = optarg;
break;
case 'c':
doClean = 1;
break;
case '?':
fprintf(stderr,
"usage: afpidsrvr [-c] [-v] [-t] [-l logfile]\n");
exit(-1);
}
}
set_signal();
if (disconnect)
do_disconnect();
if (log) {
int fd;
if ((fd = open(log, O_WRONLY|O_APPEND|O_CREAT)) < 0)
fd = open("/dev/null", O_WRONLY);
if (fd >= 0) {
#ifndef NODUP2
dup2(fd, 2);
#else NODUP2
close(2);
dup(fd);
#endif NODUP2
close(fd);
}
}
init_aufsExt();
open_database();
create_queries();
if (doClean)
do
clean_entries();
while (continue_clean);
#ifdef USE_GDBM
gdbm_close(db);
db = NULL;
#endif /* USE_GDBM */
{
time_t now;
time(&now);
fprintf(stderr, "Starting afpidsrvr with %s at %s",
aufsDbName, ctime(&now));
}
session();
/* NOTREACHED */
}
/*
* create queries "pipe" entry
*
*/
void
create_queries()
{
struct sockaddr *addr;
int addrlen;
if ((queries = query_socket()) < 0)
fatal("Problem creating socket");
query_addr(&addr, &addrlen);
if (bind(queries, addr, addrlen) < 0)
fatal2("Problem binding socket ", aufsSockname);
bound_queries = 1;
#ifdef linux
chmod(aufsSockname, 0666);
#endif /* linux */
if (listen(queries, 5) < 0)
fatal("Listen");
return;
}
void
do_exit()
{
clean_exit(-1);
}
/*
* force clean exit
*
*/
void
set_signal()
{
if (signal(SIGHUP, SIG_IGN)!=SIG_IGN)
signal(SIGHUP, do_exit);
if (signal(SIGINT, SIG_IGN)!=SIG_IGN)
signal(SIGINT, do_exit);
if (signal(SIGTERM, SIG_IGN)!=SIG_IGN)
signal(SIGTERM, do_exit);
}
void
create_database()
{
datum initial_key, initial;
datum root, root_datum;
#ifdef USE_GDBM
if ((db = gdbm_open(aufsDbName, 2048, GDBM_WRCREAT, 0644, 0L)) == NULL)
fatal("Creating db");
#else /* USE_GDBM */
if ((db = dbm_open(aufsDbName, O_RDWR|O_CREAT, 0644)) == NULL)
fatal("Creating db");
#endif /* USE_GDBM */
rootEid = valid_id(); /* any will do */
root_datum = new_entry(aufsRootName, (sdword)0);
root = num_datum(rootEid);
#ifdef USE_GDBM
if (gdbm_store(db, root, root_datum, GDBM_REPLACE) < 0)
fatal("Writing root record");
#else /* USE_GDBM */
if (dbm_store(db, root, root_datum, DBM_REPLACE) < 0)
fatal("Writing root record");
#endif /* USE_GDBM */
initial_key.dptr = "I";
initial_key.dsize = strlen(initial_key.dptr);
initial = create_init(aufsDbVersion, rootEid);
#ifdef USE_GDBM
if (gdbm_store(db, initial_key, initial, GDBM_REPLACE) < 0)
fatal("Writing root record");
#else /* USE_GDBM */
if (dbm_store(db, initial_key, initial, DBM_REPLACE) < 0)
fatal("Writing root record");
#endif /* USE_GDBM */
flush_database();
}
void
open_database()
{
datum initial_key, initial;
char *version;
initial_key.dptr = "I";
initial_key.dsize = strlen(initial_key.dptr);
#ifdef USE_GDBM
db = gdbm_open(aufsDbName, 2048, GDBM_READER, 0644, 0L);
#else /* USE_GDBM */
db = dbm_open(aufsDbName, O_RDWR, 0644);
#endif /* USE_GDBM */
if (db == NULL && errno == ENOENT)
create_database();
if (db == NULL)
fatal("Opening db");
#ifdef USE_GDBM
initial = gdbm_fetch(db, initial_key);
#else /* USE_GDBM */
initial = dbm_fetch(db, initial_key);
#endif /* USE_GDBM */
if (initial.dptr == NULL)
fatal("Suspect Database");
if (extract_init(initial, &version, &rootEid) < 0)
fatal("Problem with initial record");
if (strcmp(version, aufsDbVersion) != 0)
fatal2("Incompatible d/b, can't deal with version ", version);
#ifdef USE_GDBM
free(initial.dptr);
#endif /* USE_GDBM */
return;
}
void
add_entry(entry)
char *entry;
{
int ret;
if (!is_directory(entry))
return;
if ((ret = lookup_path(entry, NULL, NULL, 1)) <= 0)
fprintf(stderr, "Failed to create '%s' (%d, %d)\n",
entry, ret, errno);
return;
}
void
delete_entry_id();
void
add_entry_id(parent, rest)
sdword parent;
char *rest;
{
int ret;
sdword id;
char *entry;
if ((ret = lookup_path_id(parent, rest, &id, NULL, 1)) <= 0) {
fprintf(stderr, "Failed to create %d/'%s' (%d, %d)\n",
parent, rest, ret, errno);
return;
}
if (verbose)
fprintf(stderr, "Created entry %d/'%s' (%d, %d) id = %d\n",
parent, rest, ret, errno, id);
entry = equiv_path(id);
if (!is_directory(entry)) /* check existance in restrospect! */
delete_entry_id(id);
flush_database();
return;
}
void
add_entry_fid(parent, rest)
sdword parent;
char *rest;
{
int ret;
sdword id;
char *entry;
if ((ret = lookup_path_id(parent, rest, &id, NULL, 1)) <= 0) {
fprintf(stderr, "Failed to create %d/'%s' (%d, %d)\n",
parent, rest, ret, errno);
return;
}
entry = equiv_path(id);
if (!is_file(entry)) /* check existance in restrospect! */
delete_entry_id(id);
flush_database();
return;
}
void
delete_entry(entry)
char *entry;
{
sdword id, parent;
datum data;
if (is_directory(entry) || is_file(entry))
return;
if (lookup_path(entry, &id, &data, 0) > 0){
(void) extract_entry(data, NULL, &parent, NULL, NULL);
do_delete_entry(id);
delete_child(parent, id);
flush_database();
}
return;
}
void
delete_entry_id(id)
sdword id;
{
sdword parent;
datum data;
char *entry = equiv_path(id);
if (is_directory(entry) || is_file(entry))
return;
data = get_datum(id);
if (data.dptr == NULL)
return;
if (extract_entry(data, NULL, &parent, NULL, NULL) < 0)
return;
do_delete_entry(id);
delete_child(parent, id);
flush_database();
return;
}
void
delete_entry_fid(id)
sdword id;
{
sdword parent;
datum data;
char *entry = equiv_path(id);
if (is_file(entry))
return;
data = get_datum(id);
if (data.dptr == NULL)
return;
if (extract_entry(data, NULL, &parent, NULL, NULL) < 0)
return;
do_delete_entry(id);
delete_child(parent, id);
flush_database();
return;
}
void
move_entry(from, to)
char *from, *to;
{
datum data;
sdword id, from_parent, to_parent;
char to_directory[MAXPATHLEN];
char *new_name;
char *name;
char *ptr;
assert(to[0] == '/');
if (is_directory(from) || !is_directory(to))
if (is_file(from) || !is_file(to))
return;
if (lookup_path(from, &id, &data, 0) <= 0) {
/* did not know previously! */
add_entry(to);
return;
}
if (lookup_path(to, NULL, NULL, 0) > 0) /* exists already! */
return;
extract_entry(data, &name, &from_parent, NULL, NULL);
name = string_copy(name); /* just in case it gets clobbered */
ptr = rindex(to, '/');
strcpy(to_directory, "");
strncat(to_directory, to, ptr-to);
new_name = ptr+1;
assert(new_name[0] != '\0');
/* will happen for trailing /, so be careful */
lookup_path(to_directory, &to_parent, NULL, 1);
fprintf(stderr, "Moving %d/%s to %d/%s\n",
from_parent, name, to_parent, new_name);
if (from_parent != to_parent) {
/* NB it may be quicker to try compare the two strings */
delete_child(from_parent, id);
add_child(to_parent, id);
}
data = get_datum(id); /* get again - in case overwritten */
data = modify_parent(data, to_parent);
if (strcmp(name, new_name) != 0) /* name change too */
data = modify_name(data, new_name);
store_datum(id, data, DBM_REPLACE);
flush_database();
return;
}
void
move_entry_id(from_parent, name, to_parent, new_name)
sdword from_parent, to_parent;
char *name, *new_name;
{
datum data;
sdword id;
char *ptr;
if (lookup_path_id(from_parent, name, &id, NULL, 0) <= 0) {
/* did not know previously! */
add_entry_id(from_parent, name);
return;
}
if (lookup_path_id(to_parent, new_name, NULL, NULL, 0) > 0)
/* exists already! */
return;
if (is_directory(equiv_path(id))||is_file(equiv_path(id)))
return;
if (from_parent == to_parent) {
data = get_datum(id); /* get here in case corrupted earlier */
data = modify_name(data, new_name);
store_datum(id, data, DBM_REPLACE);
} else {
delete_child(from_parent, id);
add_child(to_parent, id);
data = get_datum(id); /* get here in case corrupted earlier */
data = modify_parent(data, to_parent);
if (strcmp(name, new_name) != 0) /* name change too */
data = modify_name(data, new_name);
store_datum(id, data, DBM_REPLACE);
}
if (! is_directory(equiv_path(id)) && ! is_file(equiv_path(id)))
delete_entry_id(id); /* does not actually exist */
else
flush_database();
return;
}
void
rename_entry_id(id, new_name)
sdword id;
char *new_name;
{
datum data;
int ret;
if (verbose)
fprintf(stderr, "rename_entry_id: %d -> %s\n",
id, new_name);
data = get_datum(id);
if (data.dptr == NULL) {
if (verbose)
fprintf(stderr, "Unknown id: %d\n", id);
/* did not know previously! */
return;
}
data = modify_name(data, new_name);
if (ret = store_datum(id, data, DBM_REPLACE) < 0)
if (verbose)
fprintf(stderr, "rename_entry_id: store returned %d\n",
ret);
if (! is_directory(equiv_path(id)) && ! is_file(equiv_path(id))) {
if (verbose)
fprintf(stderr, "rename_entry_id: Ooops! %s doesn't exist!\n",
equiv_path(id));
delete_entry_id(id); /* does not actually exist */
} else
flush_database();
return;
}
void
rename_entry(path, new_name)
char *path, *new_name;
{
datum data;
sdword id;
char *ptr;
if (lookup_path(path, &id, &data, 0) <= 0) {
/* did not know previously! */
return;
}
data = modify_name(data, new_name);
store_datum(id, data, DBM_REPLACE);
if (! is_directory(equiv_path(id)) && ! is_file(equiv_path(id)))
delete_entry_id(id); /* does not actually exist */
else
flush_database();
return;
}
void
clean_entries()
{
/* currently no-op */
continue_clean = 0;
return;
}
void
session()
{
struct sockaddr addr;
int addrlen;
int sock;
char buff[2*MAXPATHLEN+3];
char command, arg1[MAXPATHLEN], arg2[MAXPATHLEN];
sdword id1, id2;
int count;
int prob, args, fileID;
for (;;) {
fileID = 0;
addrlen = sizeof(struct sockaddr);
if ((sock = accept(queries, &addr, &addrlen)) < 0)
fatal("Accept");
if ((count = recv(sock, buff, 2*MAXPATHLEN+3, 0)) < 0)
fatal("Recv");
buff[count] = '\0';
if (verbose && count > 0)
fprintf(stderr, "Received: '%s'\n", buff);
prob = 0;
if (buff[0] != '\0') {
command = buff[0];
switch (command) { /* first decode arguments */
case 'A': case 'D':
args = sscanf(buff+1, "%[^\277]\277", arg1);
break;
case 'M':
args = sscanf(buff+1, "%[^\277]\277%[^\277]\277", arg1, arg2);
break;
case 'R':
args = sscanf(buff+1, "%[^\277]\277%[^\277]\277", arg1, arg2);
break;
case 'a':
args = sscanf(buff+1, "%d\277%[^\277]\277", &id1, arg1);
break;
case 'f':
args = sscanf(buff+1, "%d\277%[^\277]\277", &id1, arg1);
break;
case 'd':
args = sscanf(buff+1, "%d\277", &id1);
break;
case 'm':
args = sscanf(buff+1, "%d\277%[^\277]\277%d\277%[^\277]\277",
&id1, arg1, &id2, arg2);
break;
case 'r':
args = sscanf(buff+1, "%d\277%[^\277]\277", &id1, arg1);
break;
}
switch (command) {
case 'A':
if (args < 1 || arg1[0] != '/')
prob = 1;
else
add_entry(arg1);
break;
case 'a':
if (args < 2)
prob = 1;
else
add_entry_id(id1, arg1);
break;
case 'f':
if (args < 2)
prob = 1;
else
add_entry_fid(id1, arg1);
break;
case 'D':
if (args < 1 || arg1[0] != '/')
prob = 1;
else
delete_entry(arg1);
break;
case 'd':
if (args < 1)
prob = 1;
else
delete_entry_id(id1);
break;
case 'M':
if (args < 2 || arg1[0] != '/' || arg2[0] != '/')
prob = 1;
else
move_entry(arg1, arg2);
break;
case 'm':
if (args < 4)
prob = 1;
else
move_entry_id(id1, arg1, id2, arg2);
break;
case 'R':
if (args < 2 || arg1[0] != '/')
prob = 1;
else
rename_entry(arg1, arg2);
break;
case 'r':
if (args < 2)
prob = 1;
else
rename_entry_id(id1, arg1);
break;
case 'C':
clean_entries();
break;
default:
prob = 1;
break;
}
}
if (prob)
fprintf(stderr, "Bad command '%s'\n", buff);
close(sock);
}
return;
}
/*
* disassociate
*
*/
void
do_disconnect()
{
if (fork())
_exit(0);
{
int f;
for (f = 0; f < 10; f++)
(void) close(f);
}
(void) open("/", 0);
#ifndef NODUP2
(void) dup2(0, 1);
(void) dup2(0, 2);
#else NODUP2
(void)dup(0); /* for slot 1 */
(void)dup(0); /* for slot 2 */
#endif NODUP2
#ifndef POSIX
#ifdef TIOCNOTTY
{
int t;
if ((t = open("/dev/tty", 2)) >= 0) {
ioctl(t, TIOCNOTTY, (char *)0);
(void) close(t);
}
}
#endif TIOCNOTTY
#ifdef xenix5
/*
* USG process groups:
* The fork guarantees that the child is not a process group leader.
* Then setpgrp() can work, whick loses the controllong tty.
* Note that we must be careful not to be the first to open any tty,
* or it will become our controlling tty. C'est la vie.
*
*/
setpgrp();
#endif xenix5
#else POSIX
(void) setsid();
#endif POSIX
}
/*
* These are here to ensure we pick up these
* versions for the server, and not those in lib_client.c
*
*/
int
amAufsExt()
{
return(1);
}
/*
* flush_database: force any outstanding writes
*
*/
void
flush_database()
{
#ifndef USE_GDBM
extern void open_database( /* void */ );
dbm_close(db);
open_database();
#endif /* USE_GDBM */
return;
}
#else FIXED_DIRIDS
#include <stdio.h>
main()
{
printf("afpidsrvr: not compiled with -DFIXED_DIRIDS\n");
}
#endif FIXED_DIRIDS
Reference in New Issue View Git Blame Copy Permalink