2 * $Id: cnid_dbd.c,v 1.11 2009-11-05 14:38:08 franklahm 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;
56 extern char *Cnid_port;
60 /* *MUST* be < afp tickle or it's never triggered (got EINTR first) */
63 static void delay(int sec)
69 select(0, NULL, NULL, NULL, &tv);
72 static int tsock_getfd(const char *host, const char *port)
78 struct addrinfo hints, *servinfo, *p;
80 /* Prepare hint for getaddrinfo */
81 memset(&hints, 0, sizeof hints);
82 hints.ai_family = AF_UNSPEC;
83 hints.ai_socktype = SOCK_STREAM;
84 hints.ai_flags = AI_NUMERICSERV;
86 if ((err = getaddrinfo(host, port, &hints, &servinfo)) != 0) {
87 LOG(log_error, logtype_default, "tsock_getfd: getaddrinfo: %s:%s : %s\n", host, port, gai_strerror(err));
91 /* loop through all the results and bind to the first we can */
92 for (p = servinfo; p != NULL; p = p->ai_next) {
93 if ((sock = socket(p->ai_family, p->ai_socktype, p->ai_protocol)) == -1) {
94 LOG(log_info, logtype_default, "tsock_getfd: socket %s:: %s", host, strerror(errno));
99 if (setsockopt(sock, SOL_TCP, TCP_NODELAY, &attr, sizeof(attr)) == -1) {
100 LOG(log_error, logtype_cnid, "getfd: set TCP_NODELAY %s: %s", host, strerror(errno));
105 tv.tv_sec = SOCK_DELAY;
107 if (setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv)) < 0) {
108 LOG(log_error, logtype_cnid, "getfd: set SO_RCVTIMEO %s: %s", host, strerror(errno));
113 tv.tv_sec = SOCK_DELAY;
115 if (setsockopt(sock, SOL_SOCKET, SO_SNDTIMEO, &tv, sizeof(tv)) < 0) {
116 LOG(log_error, logtype_cnid, "getfd: set SO_SNDTIMEO %s: %s", host, strerror(errno));
121 if (connect(sock, p->ai_addr, p->ai_addrlen) == -1) {
125 LOG(log_error, logtype_cnid, "getfd: connect %s: %s", host, strerror(err));
129 /* We've got a socket */
134 freeaddrinfo(servinfo);
143 LOG(log_error, logtype_cnid, "tsock_getfd: no suitable network config from %s:%s", host, port);
150 /* --------------------- */
151 static int write_vec(int fd, struct iovec *iov, size_t towrite)
156 LOG(log_maxdebug, logtype_cnid, "write_vec: request to write %d bytes", towrite);
158 len1 = iov[1].iov_len;
159 while (towrite > 0) {
160 if (((len = writev(fd, iov, 2)) == -1 && errno == EINTR) || !len)
163 if ((size_t)len == towrite) /* wrote everything out */
165 else if (len < 0) { /* error */
170 if (towrite > len1) { /* skip part of header */
171 iov[0].iov_base = (char *) iov[0].iov_base + len;
172 iov[0].iov_len -= len;
173 } else { /* skip to data */
174 if (iov[0].iov_len) {
175 len -= iov[0].iov_len;
178 iov[1].iov_base = (char *) iov[1].iov_base + len;
179 iov[1].iov_len -= len;
183 LOG(log_maxdebug, logtype_cnid, "write_vec: wrote %d bytes", towrite);
188 /* --------------------- */
189 static int init_tsock(CNID_private *db)
195 LOG(log_debug, logtype_cnid, "init_tsock: BEGIN. Opening volume '%s', CNID Server: %s/%s", db->db_dir, Cnid_srv, Cnid_port);
197 if ((fd = tsock_getfd(Cnid_srv, Cnid_port)) < 0)
200 len = strlen(db->db_dir);
202 iov[0].iov_base = &len;
203 iov[0].iov_len = sizeof(int);
205 iov[1].iov_base = db->db_dir;
206 iov[1].iov_len = len;
208 if (write_vec(fd, iov, len + sizeof(int)) < 0) {
209 LOG(log_error, logtype_cnid, "init_tsock: Error/short write: %s", strerror(errno));
214 LOG(log_debug, logtype_cnid, "init_tsock: ok");
219 /* --------------------- */
220 static int send_packet(CNID_private *db, struct cnid_dbd_rqst *rqst)
225 LOG(log_maxdebug, logtype_cnid, "send_packet: BEGIN");
227 if (!rqst->namelen) {
228 if (write(db->fd, rqst, sizeof(struct cnid_dbd_rqst)) != sizeof(struct cnid_dbd_rqst)) {
229 LOG(log_warning, logtype_cnid, "send_packet: Error/short write rqst (db_dir %s): %s",
230 db->db_dir, strerror(errno));
233 LOG(log_maxdebug, logtype_cnid, "send_packet: OK");
237 iov[0].iov_base = rqst;
238 iov[0].iov_len = sizeof(struct cnid_dbd_rqst);
240 iov[1].iov_base = rqst->name;
241 iov[1].iov_len = rqst->namelen;
243 towrite = sizeof(struct cnid_dbd_rqst) +rqst->namelen;
245 if (write_vec(db->fd, iov, towrite) < 0) {
246 LOG(log_warning, logtype_cnid, "send_packet: Error writev rqst (db_dir %s): %s",
247 db->db_dir, strerror(errno));
251 LOG(log_maxdebug, logtype_cnid, "send_packet: OK");
255 /* ------------------- */
256 static void dbd_initstamp(struct cnid_dbd_rqst *rqst)
259 rqst->op = CNID_DBD_OP_GETSTAMP;
262 /* ------------------- */
263 static int dbd_reply_stamp(struct cnid_dbd_rply *rply)
265 switch (rply->result) {
266 case CNID_DBD_RES_OK:
268 case CNID_DBD_RES_NOTFOUND:
270 case CNID_DBD_RES_ERR_DB:
278 /* ------------------- */
279 static ssize_t dbd_read(int socket, void *data, const size_t length)
285 while (stored < length) {
286 len = read(socket, (u_int8_t *) data + stored, length - stored);
300 /* ---------------------
301 * send a request and get reply
302 * assume send is non blocking
303 * if no answer after sometime (at least MAX_DELAY secondes) return an error
305 static int dbd_rpc(CNID_private *db, struct cnid_dbd_rqst *rqst, struct cnid_dbd_rply *rply)
311 LOG(log_maxdebug, logtype_cnid, "dbd_rpc: BEGIN");
313 if (send_packet(db, rqst) < 0) {
317 nametmp = rply->name;
319 ret = dbd_read(db->fd, rply, sizeof(struct cnid_dbd_rply));
321 if (ret != sizeof(struct cnid_dbd_rply)) {
322 LOG(log_error, logtype_cnid, "dbd_rpc: Error reading header from fd (db_dir %s): %s",
323 db->db_dir, ret == -1?strerror(errno):"closed");
324 rply->name = nametmp;
327 rply->name = nametmp;
328 if (rply->namelen && rply->namelen > len) {
329 LOG(log_error, logtype_cnid,
330 "dbd_rpc: Error reading name (db_dir %s): %s name too long: %d. only wanted %d, garbage?",
331 db->db_dir, rply->name, rply->namelen, len);
334 if (rply->namelen && (ret = dbd_read(db->fd, rply->name, rply->namelen)) != (ssize_t)rply->namelen) {
335 LOG(log_error, logtype_cnid, "dbd_rpc: Error reading name from fd (db_dir %s): %s",
336 db->db_dir, ret == -1?strerror(errno):"closed");
340 LOG(log_maxdebug, logtype_cnid, "dbd_rpc: END");
345 /* -------------------- */
346 static int transmit(CNID_private *db, struct cnid_dbd_rqst *rqst, struct cnid_dbd_rply *rply)
349 int clean = 1; /* no errors so far - to prevent sleep on first try */
351 LOG(log_debug7, logtype_cnid, "transmit: BEGIN");
354 /* volume and db don't have the same timestamp
360 struct cnid_dbd_rqst rqst_stamp;
361 struct cnid_dbd_rply rply_stamp;
362 char stamp[ADEDLEN_PRIVSYN];
367 LOG(log_debug, logtype_cnid, "transmit: connecting to cnid_dbd ...");
368 if ((db->fd = init_tsock(db)) < 0) {
370 if (t - orig > MAX_DELAY)
374 dbd_initstamp(&rqst_stamp);
375 memset(stamp, 0, ADEDLEN_PRIVSYN);
376 rply_stamp.name = stamp;
377 rply_stamp.namelen = ADEDLEN_PRIVSYN;
379 if (dbd_rpc(db, &rqst_stamp, &rply_stamp) < 0)
381 if (dbd_reply_stamp(&rply_stamp ) < 0)
385 LOG(log_debug7, logtype_cnid, "transmit: reconnected to cnid_dbd, comparing database stamps...");
386 if (memcmp(stamp, db->stamp, ADEDLEN_PRIVSYN)) {
387 LOG(log_error, logtype_cnid, "transmit: ... not the same db!");
391 LOG(log_debug7, logtype_cnid, "transmit: ... OK.");
395 if (db->client_stamp)
396 memcpy(db->client_stamp, stamp, ADEDLEN_PRIVSYN);
397 memcpy(db->stamp, stamp, ADEDLEN_PRIVSYN);
399 LOG(log_debug, logtype_cnid, "transmit: succesfully attached to cnid_dbd for volume '%s' with stamp '%08lx'.",
400 db->db_dir, *(uint64_t *)stamp);
402 if (!dbd_rpc(db, rqst, rply)) {
403 LOG(log_debug7, logtype_cnid, "transmit: END OK");
409 db->fd = -1; /* FD not valid... will need to reconnect */
412 if (!clean) { /* don't sleep if just got disconnected by cnid server */
414 if (t - orig > MAX_DELAY) {
415 LOG(log_error, logtype_cnid, "transmit: Request to dbd daemon (db_dir %s) timed out.", db->db_dir);
418 /* sleep a little before retry */
421 clean = 0; /* false... next time sleep */
428 /* ---------------------- */
429 static struct _cnid_db *cnid_dbd_new(const char *volpath)
431 struct _cnid_db *cdb;
433 if ((cdb = (struct _cnid_db *)calloc(1, sizeof(struct _cnid_db))) == NULL)
436 if ((cdb->volpath = strdup(volpath)) == NULL) {
441 cdb->flags = CNID_FLAG_PERSISTENT | CNID_FLAG_LAZY_INIT;
443 cdb->cnid_add = cnid_dbd_add;
444 cdb->cnid_delete = cnid_dbd_delete;
445 cdb->cnid_get = cnid_dbd_get;
446 cdb->cnid_lookup = cnid_dbd_lookup;
447 cdb->cnid_nextid = NULL;
448 cdb->cnid_resolve = cnid_dbd_resolve;
449 cdb->cnid_getstamp = cnid_dbd_getstamp;
450 cdb->cnid_update = cnid_dbd_update;
451 cdb->cnid_rebuild_add = cnid_dbd_rebuild_add;
452 cdb->cnid_close = cnid_dbd_close;
457 /* ---------------------- */
458 struct _cnid_db *cnid_dbd_open(const char *dir, mode_t mask _U_)
460 CNID_private *db = NULL;
461 struct _cnid_db *cdb = NULL;
467 if ((cdb = cnid_dbd_new(dir)) == NULL) {
468 LOG(log_error, logtype_cnid, "cnid_open: Unable to allocate memory for database");
472 if ((db = (CNID_private *)calloc(1, sizeof(CNID_private))) == NULL) {
473 LOG(log_error, logtype_cnid, "cnid_open: Unable to allocate memory for database");
474 goto cnid_dbd_open_fail;
479 /* We keep a copy of the directory in the db structure so that we can
480 transparently reconnect later. */
481 strcpy(db->db_dir, dir);
482 db->magic = CNID_DB_MAGIC;
485 LOG(log_debug, logtype_cnid, "cnid_dbd_open: Finished initializing cnid dbd module for volume '%s'", db->db_dir);
491 if (cdb->volpath != NULL) {
502 /* ---------------------- */
503 void cnid_dbd_close(struct _cnid_db *cdb)
508 LOG(log_error, logtype_cnid, "cnid_close called with NULL argument !");
512 if ((db = cdb->_private) != NULL) {
513 LOG(log_info, logtype_cnid, "closing database connection for volume '%s'", db->db_dir);
526 /* ---------------------- */
527 cnid_t cnid_dbd_add(struct _cnid_db *cdb, const struct stat *st,
528 const cnid_t did, char *name, const size_t len,
532 struct cnid_dbd_rqst rqst;
533 struct cnid_dbd_rply rply;
536 if (!cdb || !(db = cdb->_private) || !st || !name) {
537 LOG(log_error, logtype_cnid, "cnid_add: Parameter error");
538 errno = CNID_ERR_PARAM;
542 if (len > MAXPATHLEN) {
543 LOG(log_error, logtype_cnid, "cnid_add: Path name is too long");
544 errno = CNID_ERR_PATH;
549 rqst.op = CNID_DBD_OP_ADD;
551 if (!(cdb->flags & CNID_FLAG_NODEV)) {
552 rqst.dev = st->st_dev;
555 rqst.ino = st->st_ino;
556 rqst.type = S_ISDIR(st->st_mode)?1:0;
561 LOG(log_debug, logtype_cnid, "cnid_dbd_add: CNID: %u, name: '%s', inode: 0x%llx, type: %d (0=file, 1=dir)",
562 ntohl(did), name, (long long)st->st_ino, rqst.type);
565 if (transmit(db, &rqst, &rply) < 0) {
570 switch(rply.result) {
571 case CNID_DBD_RES_OK:
573 LOG(log_debug, logtype_cnid, "cnid_dbd_add: got CNID: %u", ntohl(id));
575 case CNID_DBD_RES_ERR_MAX:
576 errno = CNID_ERR_MAX;
579 case CNID_DBD_RES_ERR_DB:
580 case CNID_DBD_RES_ERR_DUPLCNID:
591 /* ---------------------- */
592 cnid_t cnid_dbd_get(struct _cnid_db *cdb, const cnid_t did, char *name, const size_t len)
595 struct cnid_dbd_rqst rqst;
596 struct cnid_dbd_rply rply;
599 if (!cdb || !(db = cdb->_private) || !name) {
600 LOG(log_error, logtype_cnid, "cnid_dbd_get: Parameter error");
601 errno = CNID_ERR_PARAM;
605 if (len > MAXPATHLEN) {
606 LOG(log_error, logtype_cnid, "cnid_dbd_get: Path name is too long");
607 errno = CNID_ERR_PATH;
611 LOG(log_debug, logtype_cnid, "cnid_dbd_get: DID: %u, name: '%s'", ntohl(did), name);
614 rqst.op = CNID_DBD_OP_GET;
620 if (transmit(db, &rqst, &rply) < 0) {
625 switch(rply.result) {
626 case CNID_DBD_RES_OK:
628 LOG(log_debug, logtype_cnid, "cnid_dbd_get: got CNID: %u", ntohl(id));
630 case CNID_DBD_RES_NOTFOUND:
633 case CNID_DBD_RES_ERR_DB:
644 /* ---------------------- */
645 char *cnid_dbd_resolve(struct _cnid_db *cdb, cnid_t *id, void *buffer, size_t len)
648 struct cnid_dbd_rqst rqst;
649 struct cnid_dbd_rply rply;
652 if (!cdb || !(db = cdb->_private) || !id || !(*id)) {
653 LOG(log_error, logtype_cnid, "cnid_resolve: Parameter error");
654 errno = CNID_ERR_PARAM;
658 LOG(log_debug, logtype_cnid, "cnid_dbd_resolve: resolving CNID: %u", ntohl(*id));
660 /* TODO: We should maybe also check len. At the moment we rely on the caller
661 to provide a buffer that is large enough for MAXPATHLEN plus
662 CNID_HEADER_LEN plus 1 byte, which is large enough for the maximum that
663 can come from the database. */
666 rqst.op = CNID_DBD_OP_RESOLVE;
669 /* This mimicks the behaviour of the "regular" cnid_resolve. So far,
670 nobody uses the content of buffer. It only provides space for the
671 name in the caller. */
672 rply.name = (char *)buffer + CNID_HEADER_LEN;
673 rply.namelen = len - CNID_HEADER_LEN;
675 if (transmit(db, &rqst, &rply) < 0) {
681 switch (rply.result) {
682 case CNID_DBD_RES_OK:
685 LOG(log_debug, logtype_cnid, "cnid_dbd_resolve: resolved did: %u, name: '%s'", ntohl(*id), name);
687 case CNID_DBD_RES_NOTFOUND:
691 case CNID_DBD_RES_ERR_DB:
703 /* ---------------------- */
704 int cnid_dbd_getstamp(struct _cnid_db *cdb, void *buffer, const size_t len)
708 if (!cdb || !(db = cdb->_private) || len != ADEDLEN_PRIVSYN) {
709 LOG(log_error, logtype_cnid, "cnid_getstamp: Parameter error");
710 errno = CNID_ERR_PARAM;
713 db->client_stamp = buffer;
714 db->stamp_size = len;
715 memset(buffer,0, len);
719 /* ---------------------- */
720 cnid_t cnid_dbd_lookup(struct _cnid_db *cdb, const struct stat *st, const cnid_t did,
721 char *name, const size_t len)
724 struct cnid_dbd_rqst rqst;
725 struct cnid_dbd_rply rply;
728 if (!cdb || !(db = cdb->_private) || !st || !name) {
729 LOG(log_error, logtype_cnid, "cnid_lookup: Parameter error");
730 errno = CNID_ERR_PARAM;
734 if (len > MAXPATHLEN) {
735 LOG(log_error, logtype_cnid, "cnid_lookup: Path name is too long");
736 errno = CNID_ERR_PATH;
741 rqst.op = CNID_DBD_OP_LOOKUP;
743 if (!(cdb->flags & CNID_FLAG_NODEV)) {
744 rqst.dev = st->st_dev;
747 rqst.ino = st->st_ino;
748 rqst.type = S_ISDIR(st->st_mode)?1:0;
753 LOG(log_debug, logtype_cnid, "cnid_dbd_lookup: CNID: %u, name: '%s', inode: 0x%llx, type: %d (0=file, 1=dir)",
754 ntohl(did), name, (long long)st->st_ino, rqst.type);
757 if (transmit(db, &rqst, &rply) < 0) {
762 switch (rply.result) {
763 case CNID_DBD_RES_OK:
765 LOG(log_debug, logtype_cnid, "cnid_dbd_lookup: got CNID: %u", ntohl(id));
767 case CNID_DBD_RES_NOTFOUND:
770 case CNID_DBD_RES_ERR_DB:
781 /* ---------------------- */
782 int cnid_dbd_update(struct _cnid_db *cdb, const cnid_t id, const struct stat *st,
783 const cnid_t did, char *name, const size_t len)
786 struct cnid_dbd_rqst rqst;
787 struct cnid_dbd_rply rply;
789 if (!cdb || !(db = cdb->_private) || !id || !st || !name) {
790 LOG(log_error, logtype_cnid, "cnid_update: Parameter error");
791 errno = CNID_ERR_PARAM;
795 if (len > MAXPATHLEN) {
796 LOG(log_error, logtype_cnid, "cnid_update: Path name is too long");
797 errno = CNID_ERR_PATH;
802 rqst.op = CNID_DBD_OP_UPDATE;
804 if (!(cdb->flags & CNID_FLAG_NODEV)) {
805 rqst.dev = st->st_dev;
807 rqst.ino = st->st_ino;
808 rqst.type = S_ISDIR(st->st_mode)?1:0;
813 LOG(log_debug, logtype_cnid, "cnid_dbd_update: CNID: %u, name: '%s', inode: 0x%llx, type: %d (0=file, 1=dir)",
814 ntohl(id), name, (long long)st->st_ino, rqst.type);
817 if (transmit(db, &rqst, &rply) < 0) {
822 switch (rply.result) {
823 case CNID_DBD_RES_OK:
824 LOG(log_debug, logtype_cnid, "cnid_dbd_update: updated");
825 case CNID_DBD_RES_NOTFOUND:
827 case CNID_DBD_RES_ERR_DB:
835 /* ---------------------- */
836 cnid_t cnid_dbd_rebuild_add(struct _cnid_db *cdb, const struct stat *st,
837 const cnid_t did, char *name, const size_t len,
841 struct cnid_dbd_rqst rqst;
842 struct cnid_dbd_rply rply;
845 if (!cdb || !(db = cdb->_private) || !st || !name || hint == CNID_INVALID) {
846 LOG(log_error, logtype_cnid, "cnid_rebuild_add: Parameter error");
847 errno = CNID_ERR_PARAM;
851 if (len > MAXPATHLEN) {
852 LOG(log_error, logtype_cnid, "cnid_rebuild_add: Path name is too long");
853 errno = CNID_ERR_PATH;
858 rqst.op = CNID_DBD_OP_REBUILD_ADD;
860 if (!(cdb->flags & CNID_FLAG_NODEV)) {
861 rqst.dev = st->st_dev;
864 rqst.ino = st->st_ino;
865 rqst.type = S_ISDIR(st->st_mode)?1:0;
871 LOG(log_debug, logtype_cnid, "cnid_dbd_rebuild_add: CNID: %u, name: '%s', inode: 0x%llx, type: %d (0=file, 1=dir), hint: %u",
872 ntohl(did), name, (long long)st->st_ino, rqst.type, hint);
874 if (transmit(db, &rqst, &rply) < 0) {
879 switch(rply.result) {
880 case CNID_DBD_RES_OK:
882 LOG(log_debug, logtype_cnid, "cnid_dbd_rebuild_add: got CNID: %u", ntohl(id));
884 case CNID_DBD_RES_ERR_MAX:
885 errno = CNID_ERR_MAX;
888 case CNID_DBD_RES_ERR_DB:
889 case CNID_DBD_RES_ERR_DUPLCNID:
899 /* ---------------------- */
900 int cnid_dbd_delete(struct _cnid_db *cdb, const cnid_t id)
903 struct cnid_dbd_rqst rqst;
904 struct cnid_dbd_rply rply;
906 if (!cdb || !(db = cdb->_private) || !id) {
907 LOG(log_error, logtype_cnid, "cnid_delete: Parameter error");
908 errno = CNID_ERR_PARAM;
912 LOG(log_debug, logtype_cnid, "cnid_dbd_delete: delete CNID: %u", ntohl(id));
915 rqst.op = CNID_DBD_OP_DELETE;
919 if (transmit(db, &rqst, &rply) < 0) {
924 switch (rply.result) {
925 case CNID_DBD_RES_OK:
926 LOG(log_debug, logtype_cnid, "cnid_dbd_delete: deleted CNID: %u", ntohl(id));
927 case CNID_DBD_RES_NOTFOUND:
929 case CNID_DBD_RES_ERR_DB:
938 struct _cnid_module cnid_dbd_module = {
945 #endif /* CNID_DBD */