2 * $Id: cnid_dbd.c,v 1.5 2009-02-04 20:28:01 didg Exp $
4 * Copyright (C) Joerg Lenneis 2003
5 * All Rights Reserved. See COPYING.
10 #endif /* HAVE_CONFIG_H */
12 #ifdef CNID_BACKEND_DBD
15 #ifdef HAVE_SYS_STAT_H
17 #endif /* HAVE_SYS_STAT_H */
20 #endif /* HAVE_SYS_UIO_H */
26 #include <sys/socket.h>
27 #include <sys/param.h>
29 #include <netinet/in.h>
31 #include <netinet/tcp.h>
32 #include <netinet/in.h>
33 #include <arpa/inet.h>
38 #include <netatalk/endian.h>
39 #include <atalk/logger.h>
40 #include <atalk/adouble.h>
41 #include <atalk/cnid.h>
43 #include <atalk/cnid_dbd_private.h>
46 #define SOL_TCP IPPROTO_TCP
47 #endif /* ! SOL_TCP */
49 static void RQST_RESET(struct cnid_dbd_rqst *r)
51 memset(r, 0, sizeof(struct cnid_dbd_rqst ));
55 extern char Cnid_srv[MAXHOSTNAMELEN + 1];
58 static int tsock_getfd(char *host, int port)
61 struct sockaddr_in server;
66 server.sin_family=AF_INET;
67 server.sin_port=htons((unsigned short)port);
69 LOG(log_error, logtype_cnid, "getfd: -cnidserver not defined");
73 hp=gethostbyname(host);
75 unsigned long int addr=inet_addr(host);
76 LOG(log_warning, logtype_cnid, "getfd: Could not resolve host %s, trying numeric address instead", host);
77 if (addr!= (unsigned)-1)
78 hp=gethostbyaddr((char*)addr,sizeof(addr),AF_INET);
81 LOG(log_error, logtype_cnid, "getfd: Could not resolve host %s", host);
85 memcpy((char*)&server.sin_addr,(char*)hp->h_addr,sizeof(server.sin_addr));
86 sock=socket(PF_INET,SOCK_STREAM,0);
88 LOG(log_error, logtype_cnid, "getfd: socket %s: %s", host, strerror(errno));
92 if (setsockopt(sock, SOL_TCP, TCP_NODELAY, &attr, sizeof(attr)) == -1) {
93 LOG(log_error, logtype_cnid, "getfd: set TCP_NODELAY %s: %s", host, strerror(errno));
97 if(connect(sock ,(struct sockaddr*)&server,sizeof(server))==-1) {
102 LOG(log_error, logtype_cnid, "getfd: connect %s: %s", host, strerror(err));
109 select(0, NULL, NULL, NULL, &tv);
116 /* --------------------- */
117 static int write_vec(int fd, struct iovec *iov, size_t towrite)
122 len1 = iov[1].iov_len;
123 while (towrite > 0) {
124 if (((len = writev(fd, iov, 2)) == -1 && errno == EINTR) || !len)
127 if ((size_t)len == towrite) /* wrote everything out */
129 else if (len < 0) { /* error */
134 if (towrite > len1) { /* skip part of header */
135 iov[0].iov_base = (char *) iov[0].iov_base + len;
136 iov[0].iov_len -= len;
137 } else { /* skip to data */
138 if (iov[0].iov_len) {
139 len -= iov[0].iov_len;
142 iov[1].iov_base = (char *) iov[1].iov_base + len;
143 iov[1].iov_len -= len;
149 /* --------------------- */
150 static int init_tsock(CNID_private *db)
156 if ((fd = tsock_getfd(Cnid_srv, Cnid_port)) < 0)
159 len = strlen(db->db_dir);
161 iov[0].iov_base = &len;
162 iov[0].iov_len = sizeof(int);
164 iov[1].iov_base = db->db_dir;
165 iov[1].iov_len = len;
167 if (write_vec(fd, iov, len + sizeof(int)) < 0) {
168 LOG(log_error, logtype_cnid, "init_tsock: Error/short write: %s", strerror(errno));
175 /* --------------------- */
176 static int send_packet(CNID_private *db, struct cnid_dbd_rqst *rqst, int silent)
182 if (!rqst->namelen) {
183 if (write(db->fd, rqst, sizeof(struct cnid_dbd_rqst)) != sizeof(struct cnid_dbd_rqst)) {
185 LOG(log_warning, logtype_cnid, "send_packet: Error/short write rqst (db_dir %s): %s",
186 db->db_dir, strerror(errno));
192 iov[0].iov_base = rqst;
193 iov[0].iov_len = sizeof(struct cnid_dbd_rqst);
195 iov[1].iov_base = rqst->name;
196 iov[1].iov_len = rqst->namelen;
198 towrite = sizeof(struct cnid_dbd_rqst) +rqst->namelen;
200 if (write_vec(db->fd, iov, towrite) < 0) {
202 LOG(log_warning, logtype_cnid, "send_packet: Error writev rqst (db_dir %s): %s",
203 db->db_dir, strerror(errno));
210 /* ------------------- */
211 static void dbd_initstamp(struct cnid_dbd_rqst *rqst)
214 rqst->op = CNID_DBD_OP_GETSTAMP;
217 /* ------------------- */
218 static int dbd_reply_stamp(struct cnid_dbd_rply *rply)
220 switch (rply->result) {
221 case CNID_DBD_RES_OK:
223 case CNID_DBD_RES_NOTFOUND:
225 case CNID_DBD_RES_ERR_DB:
233 /* ---------------------
234 * send a request and get reply
235 * assume send is non blocking
236 * if no answer after sometime (at least MAX_DELAY secondes) return an error
239 static int dbd_rpc(CNID_private *db, struct cnid_dbd_rqst *rqst, struct cnid_dbd_rply *rply, int silent)
248 if (send_packet(db, rqst, silent) < 0) {
252 FD_SET(db->fd, &readfds);
256 tv.tv_sec = MAX_DELAY;
257 while ((ret = select(maxfd + 1, &readfds, NULL, NULL, &tv)) < 0 && errno == EINTR);
261 LOG(log_error, logtype_cnid, "dbd_rpc: Error in select (db_dir %s): %s",
262 db->db_dir, strerror(errno));
269 LOG(log_error, logtype_cnid, "dbd_rpc: select timed out (db_dir %s)",
275 nametmp = rply->name;
276 /* assume that if we have something then everything is there (doesn't sleep) */
277 if ((ret = read(db->fd, rply, sizeof(struct cnid_dbd_rply))) != sizeof(struct cnid_dbd_rply)) {
279 LOG(log_error, logtype_cnid, "dbd_rpc: Error reading header from fd (db_dir %s): %s",
280 db->db_dir, ret == -1?strerror(errno):"closed");
281 rply->name = nametmp;
284 rply->name = nametmp;
285 if (rply->namelen && rply->namelen > len) {
287 LOG(log_error, logtype_cnid,
288 "dbd_rpc: Error reading name (db_dir %s): %s name too long wanted %d only %d, garbage?",
289 db->db_dir, rply->namelen, len);
292 if (rply->namelen && (ret = read(db->fd, rply->name, rply->namelen)) != (ssize_t)rply->namelen) {
294 LOG(log_error, logtype_cnid, "dbd_rpc: Error reading name from fd (db_dir %s): %s",
295 db->db_dir, ret == -1?strerror(errno):"closed");
301 /* -------------------- */
302 static int transmit(CNID_private *db, struct cnid_dbd_rqst *rqst, struct cnid_dbd_rply *rply)
307 int clean = 1; /* no errors so far - to prevent sleep on first try */
310 /* volume and db don't have the same timestamp
318 struct cnid_dbd_rqst rqst_stamp;
319 struct cnid_dbd_rply rply_stamp;
320 char stamp[ADEDLEN_PRIVSYN];
322 if ((db->fd = init_tsock(db)) < 0) {
324 if (t - orig > MAX_DELAY)
328 dbd_initstamp(&rqst_stamp);
329 memset(stamp, 0, ADEDLEN_PRIVSYN);
330 rply_stamp.name = stamp;
331 rply_stamp.namelen = ADEDLEN_PRIVSYN;
333 if (dbd_rpc(db, &rqst_stamp, &rply_stamp, silent) < 0)
335 if (dbd_reply_stamp(&rply_stamp ) < 0)
339 if (memcmp(stamp, db->stamp, ADEDLEN_PRIVSYN)) {
340 LOG(log_error, logtype_cnid, "transmit: not the same db!");
347 if (db->client_stamp) {
348 memcpy(db->client_stamp, stamp, ADEDLEN_PRIVSYN);
350 memcpy(db->stamp, stamp, ADEDLEN_PRIVSYN);
353 if (!dbd_rpc(db, rqst, rply, silent)) {
357 silent = 0; /* From now on dbd_rpc and subroutines called from there
358 will log messages if something goes wrong again */
361 db->fd = -1; /* FD not valid... will need to reconnect */
364 if (t - orig > MAX_DELAY) {
365 LOG(log_error, logtype_cnid, "transmit: Request to dbd daemon (db_dir %s) timed out.", db->db_dir);
369 if (!clean) { /* don't sleep if just got disconnected by cnid server */
370 /* sleep a little before retry */
373 select(0, NULL, NULL, NULL, &tv); /* sleep for 5 seconds */
375 clean = 0; /* false... next time sleep */
381 /* ---------------------- */
382 static struct _cnid_db *cnid_dbd_new(const char *volpath)
384 struct _cnid_db *cdb;
386 if ((cdb = (struct _cnid_db *)calloc(1, sizeof(struct _cnid_db))) == NULL)
389 if ((cdb->volpath = strdup(volpath)) == NULL) {
394 cdb->flags = CNID_FLAG_PERSISTENT | CNID_FLAG_LAZY_INIT;
396 cdb->cnid_add = cnid_dbd_add;
397 cdb->cnid_delete = cnid_dbd_delete;
398 cdb->cnid_get = cnid_dbd_get;
399 cdb->cnid_lookup = cnid_dbd_lookup;
400 cdb->cnid_nextid = NULL;
401 cdb->cnid_resolve = cnid_dbd_resolve;
402 cdb->cnid_getstamp = cnid_dbd_getstamp;
403 cdb->cnid_update = cnid_dbd_update;
404 cdb->cnid_rebuild_add = cnid_dbd_rebuild_add;
405 cdb->cnid_close = cnid_dbd_close;
410 /* ---------------------- */
411 struct _cnid_db *cnid_dbd_open(const char *dir, mode_t mask _U_)
413 CNID_private *db = NULL;
414 struct _cnid_db *cdb = NULL;
420 if ((cdb = cnid_dbd_new(dir)) == NULL) {
421 LOG(log_error, logtype_default, "cnid_open: Unable to allocate memory for database");
425 if ((db = (CNID_private *)calloc(1, sizeof(CNID_private))) == NULL) {
426 LOG(log_error, logtype_cnid, "cnid_open: Unable to allocate memory for database");
427 goto cnid_dbd_open_fail;
432 /* We keep a copy of the directory in the db structure so that we can
433 transparently reconnect later. */
434 strcpy(db->db_dir, dir);
435 db->magic = CNID_DB_MAGIC;
438 LOG(log_info, logtype_cnid, "opening database connection to %s", db->db_dir);
444 if (cdb->volpath != NULL) {
455 /* ---------------------- */
456 void cnid_dbd_close(struct _cnid_db *cdb)
461 LOG(log_error, logtype_afpd, "cnid_close called with NULL argument !");
465 if ((db = cdb->_private) != NULL) {
467 LOG(log_info, logtype_cnid, "closing database connection to %s", db->db_dir);
480 /* ---------------------- */
481 cnid_t cnid_dbd_add(struct _cnid_db *cdb, const struct stat *st,
482 const cnid_t did, char *name, const size_t len,
486 struct cnid_dbd_rqst rqst;
487 struct cnid_dbd_rply rply;
490 if (!cdb || !(db = cdb->_private) || !st || !name) {
491 LOG(log_error, logtype_cnid, "cnid_add: Parameter error");
492 errno = CNID_ERR_PARAM;
496 if (len > MAXPATHLEN) {
497 LOG(log_error, logtype_cnid, "cnid_add: Path name is too long");
498 errno = CNID_ERR_PATH;
503 rqst.op = CNID_DBD_OP_ADD;
505 if (!(cdb->flags & CNID_FLAG_NODEV)) {
506 rqst.dev = st->st_dev;
509 rqst.ino = st->st_ino;
510 rqst.type = S_ISDIR(st->st_mode)?1:0;
516 if (transmit(db, &rqst, &rply) < 0) {
521 switch(rply.result) {
522 case CNID_DBD_RES_OK:
525 case CNID_DBD_RES_ERR_MAX:
526 errno = CNID_ERR_MAX;
529 case CNID_DBD_RES_ERR_DB:
530 case CNID_DBD_RES_ERR_DUPLCNID:
540 /* ---------------------- */
541 cnid_t cnid_dbd_get(struct _cnid_db *cdb, const cnid_t did, char *name, const size_t len)
544 struct cnid_dbd_rqst rqst;
545 struct cnid_dbd_rply rply;
549 if (!cdb || !(db = cdb->_private) || !name) {
550 LOG(log_error, logtype_cnid, "cnid_get: Parameter error");
551 errno = CNID_ERR_PARAM;
555 if (len > MAXPATHLEN) {
556 LOG(log_error, logtype_cnid, "cnid_add: Path name is too long");
557 errno = CNID_ERR_PATH;
562 rqst.op = CNID_DBD_OP_GET;
568 if (transmit(db, &rqst, &rply) < 0) {
573 switch(rply.result) {
574 case CNID_DBD_RES_OK:
577 case CNID_DBD_RES_NOTFOUND:
580 case CNID_DBD_RES_ERR_DB:
591 /* ---------------------- */
592 char *cnid_dbd_resolve(struct _cnid_db *cdb, cnid_t *id, void *buffer, size_t len)
595 struct cnid_dbd_rqst rqst;
596 struct cnid_dbd_rply rply;
599 if (!cdb || !(db = cdb->_private) || !id || !(*id)) {
600 LOG(log_error, logtype_cnid, "cnid_resolve: Parameter error");
601 errno = CNID_ERR_PARAM;
605 /* TODO: We should maybe also check len. At the moment we rely on the caller
606 to provide a buffer that is large enough for MAXPATHLEN plus
607 CNID_HEADER_LEN plus 1 byte, which is large enough for the maximum that
608 can come from the database. */
611 rqst.op = CNID_DBD_OP_RESOLVE;
614 /* This mimicks the behaviour of the "regular" cnid_resolve. So far,
615 nobody uses the content of buffer. It only provides space for the
616 name in the caller. */
617 rply.name = (char *)buffer + CNID_HEADER_LEN;
618 rply.namelen = len - CNID_HEADER_LEN;
620 if (transmit(db, &rqst, &rply) < 0) {
626 switch (rply.result) {
627 case CNID_DBD_RES_OK:
631 case CNID_DBD_RES_NOTFOUND:
635 case CNID_DBD_RES_ERR_DB:
647 /* ---------------------- */
648 int cnid_dbd_getstamp(struct _cnid_db *cdb, void *buffer, const size_t len)
652 if (!cdb || !(db = cdb->_private) || len != ADEDLEN_PRIVSYN) {
653 LOG(log_error, logtype_cnid, "cnid_getstamp: Parameter error");
654 errno = CNID_ERR_PARAM;
657 db->client_stamp = buffer;
658 db->stamp_size = len;
659 memset(buffer,0, len);
663 /* ---------------------- */
664 cnid_t cnid_dbd_lookup(struct _cnid_db *cdb, const struct stat *st, const cnid_t did,
665 char *name, const size_t len)
668 struct cnid_dbd_rqst rqst;
669 struct cnid_dbd_rply rply;
672 if (!cdb || !(db = cdb->_private) || !st || !name) {
673 LOG(log_error, logtype_cnid, "cnid_lookup: Parameter error");
674 errno = CNID_ERR_PARAM;
678 if (len > MAXPATHLEN) {
679 LOG(log_error, logtype_cnid, "cnid_lookup: Path name is too long");
680 errno = CNID_ERR_PATH;
685 rqst.op = CNID_DBD_OP_LOOKUP;
687 if (!(cdb->flags & CNID_FLAG_NODEV)) {
688 rqst.dev = st->st_dev;
691 rqst.ino = st->st_ino;
692 rqst.type = S_ISDIR(st->st_mode)?1:0;
698 if (transmit(db, &rqst, &rply) < 0) {
703 switch (rply.result) {
704 case CNID_DBD_RES_OK:
707 case CNID_DBD_RES_NOTFOUND:
710 case CNID_DBD_RES_ERR_DB:
721 /* ---------------------- */
722 int cnid_dbd_update(struct _cnid_db *cdb, const cnid_t id, const struct stat *st,
723 const cnid_t did, char *name, const size_t len)
726 struct cnid_dbd_rqst rqst;
727 struct cnid_dbd_rply rply;
730 if (!cdb || !(db = cdb->_private) || !id || !st || !name) {
731 LOG(log_error, logtype_cnid, "cnid_update: Parameter error");
732 errno = CNID_ERR_PARAM;
736 if (len > MAXPATHLEN) {
737 LOG(log_error, logtype_cnid, "cnid_update: Path name is too long");
738 errno = CNID_ERR_PATH;
743 rqst.op = CNID_DBD_OP_UPDATE;
745 if (!(cdb->flags & CNID_FLAG_NODEV)) {
746 rqst.dev = st->st_dev;
748 rqst.ino = st->st_ino;
749 rqst.type = S_ISDIR(st->st_mode)?1:0;
755 if (transmit(db, &rqst, &rply) < 0) {
760 switch (rply.result) {
761 case CNID_DBD_RES_OK:
762 case CNID_DBD_RES_NOTFOUND:
764 case CNID_DBD_RES_ERR_DB:
772 /* ---------------------- */
773 cnid_t cnid_dbd_rebuild_add(struct _cnid_db *cdb, const struct stat *st,
774 const cnid_t did, char *name, const size_t len,
778 struct cnid_dbd_rqst rqst;
779 struct cnid_dbd_rply rply;
782 if (!cdb || !(db = cdb->_private) || !st || !name || hint == CNID_INVALID) {
783 LOG(log_error, logtype_cnid, "cnid_rebuild_add: Parameter error");
784 errno = CNID_ERR_PARAM;
788 if (len > MAXPATHLEN) {
789 LOG(log_error, logtype_cnid, "cnid_rebuild_add: Path name is too long");
790 errno = CNID_ERR_PATH;
795 rqst.op = CNID_DBD_OP_REBUILD_ADD;
797 if (!(cdb->flags & CNID_FLAG_NODEV)) {
798 rqst.dev = st->st_dev;
801 rqst.ino = st->st_ino;
802 rqst.type = S_ISDIR(st->st_mode)?1:0;
808 if (transmit(db, &rqst, &rply) < 0) {
813 switch(rply.result) {
814 case CNID_DBD_RES_OK:
817 case CNID_DBD_RES_ERR_MAX:
818 errno = CNID_ERR_MAX;
821 case CNID_DBD_RES_ERR_DB:
822 case CNID_DBD_RES_ERR_DUPLCNID:
832 /* ---------------------- */
833 int cnid_dbd_delete(struct _cnid_db *cdb, const cnid_t id)
836 struct cnid_dbd_rqst rqst;
837 struct cnid_dbd_rply rply;
840 if (!cdb || !(db = cdb->_private) || !id) {
841 LOG(log_error, logtype_cnid, "cnid_delete: Parameter error");
842 errno = CNID_ERR_PARAM;
847 rqst.op = CNID_DBD_OP_DELETE;
851 if (transmit(db, &rqst, &rply) < 0) {
856 switch (rply.result) {
857 case CNID_DBD_RES_OK:
858 case CNID_DBD_RES_NOTFOUND:
860 case CNID_DBD_RES_ERR_DB:
869 struct _cnid_module cnid_dbd_module = {
876 #endif /* CNID_DBD */