2 * $Id: cnid_dbd.c,v 1.10 2009-10-19 11:00:28 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];
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(char *host, int port)
75 struct sockaddr_in server;
81 server.sin_family=AF_INET;
82 server.sin_port=htons((unsigned short)port);
84 LOG(log_error, logtype_cnid, "getfd: -cnidserver not defined");
88 hp=gethostbyname(host);
90 unsigned long int addr=inet_addr(host);
91 LOG(log_warning, logtype_cnid, "getfd: Could not resolve host %s, trying numeric address instead", host);
92 if (addr!= (unsigned)-1)
93 hp=gethostbyaddr((char*)addr,sizeof(addr),AF_INET);
96 LOG(log_error, logtype_cnid, "getfd: Could not resolve host %s", host);
100 memcpy((char*)&server.sin_addr,(char*)hp->h_addr,sizeof(server.sin_addr));
101 sock=socket(PF_INET,SOCK_STREAM,0);
103 LOG(log_error, logtype_cnid, "getfd: socket %s: %s", host, strerror(errno));
107 if (setsockopt(sock, SOL_TCP, TCP_NODELAY, &attr, sizeof(attr)) == -1) {
108 LOG(log_error, logtype_cnid, "getfd: set TCP_NODELAY %s: %s", host, strerror(errno));
113 tv.tv_sec = SOCK_DELAY;
115 if (setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv)) < 0) {
116 LOG(log_error, logtype_cnid, "getfd: set SO_RCVTIMEO %s: %s", host, strerror(errno));
121 tv.tv_sec = SOCK_DELAY;
123 if (setsockopt(sock, SOL_SOCKET, SO_SNDTIMEO, &tv, sizeof(tv)) < 0) {
124 LOG(log_error, logtype_cnid, "getfd: set SO_SNDTIMEO %s: %s", host, strerror(errno));
129 if(connect(sock ,(struct sockaddr*)&server,sizeof(server))==-1) {
133 LOG(log_error, logtype_cnid, "getfd: connect %s: %s", host, strerror(err));
142 LOG(log_debug7, logtype_cnid, "tsock_getfd: using sockfd %d for cnid server '%s:%d'", sock, host, port);
147 /* --------------------- */
148 static int write_vec(int fd, struct iovec *iov, size_t towrite)
153 LOG(log_maxdebug, logtype_cnid, "write_vec: request to write %d bytes", towrite);
155 len1 = iov[1].iov_len;
156 while (towrite > 0) {
157 if (((len = writev(fd, iov, 2)) == -1 && errno == EINTR) || !len)
160 if ((size_t)len == towrite) /* wrote everything out */
162 else if (len < 0) { /* error */
167 if (towrite > len1) { /* skip part of header */
168 iov[0].iov_base = (char *) iov[0].iov_base + len;
169 iov[0].iov_len -= len;
170 } else { /* skip to data */
171 if (iov[0].iov_len) {
172 len -= iov[0].iov_len;
175 iov[1].iov_base = (char *) iov[1].iov_base + len;
176 iov[1].iov_len -= len;
180 LOG(log_maxdebug, logtype_cnid, "write_vec: wrote %d bytes", towrite);
185 /* --------------------- */
186 static int init_tsock(CNID_private *db)
192 LOG(log_debug, logtype_cnid, "init_tsock: BEGIN. Opening volume '%s'", db->db_dir);
194 if ((fd = tsock_getfd(Cnid_srv, Cnid_port)) < 0)
197 len = strlen(db->db_dir);
199 iov[0].iov_base = &len;
200 iov[0].iov_len = sizeof(int);
202 iov[1].iov_base = db->db_dir;
203 iov[1].iov_len = len;
205 if (write_vec(fd, iov, len + sizeof(int)) < 0) {
206 LOG(log_error, logtype_cnid, "init_tsock: Error/short write: %s", strerror(errno));
211 LOG(log_debug, logtype_cnid, "init_tsock: ok");
216 /* --------------------- */
217 static int send_packet(CNID_private *db, struct cnid_dbd_rqst *rqst)
222 LOG(log_maxdebug, logtype_cnid, "send_packet: BEGIN");
224 if (!rqst->namelen) {
225 if (write(db->fd, rqst, sizeof(struct cnid_dbd_rqst)) != sizeof(struct cnid_dbd_rqst)) {
226 LOG(log_warning, logtype_cnid, "send_packet: Error/short write rqst (db_dir %s): %s",
227 db->db_dir, strerror(errno));
230 LOG(log_maxdebug, logtype_cnid, "send_packet: OK");
234 iov[0].iov_base = rqst;
235 iov[0].iov_len = sizeof(struct cnid_dbd_rqst);
237 iov[1].iov_base = rqst->name;
238 iov[1].iov_len = rqst->namelen;
240 towrite = sizeof(struct cnid_dbd_rqst) +rqst->namelen;
242 if (write_vec(db->fd, iov, towrite) < 0) {
243 LOG(log_warning, logtype_cnid, "send_packet: Error writev rqst (db_dir %s): %s",
244 db->db_dir, strerror(errno));
248 LOG(log_maxdebug, logtype_cnid, "send_packet: OK");
252 /* ------------------- */
253 static void dbd_initstamp(struct cnid_dbd_rqst *rqst)
256 rqst->op = CNID_DBD_OP_GETSTAMP;
259 /* ------------------- */
260 static int dbd_reply_stamp(struct cnid_dbd_rply *rply)
262 switch (rply->result) {
263 case CNID_DBD_RES_OK:
265 case CNID_DBD_RES_NOTFOUND:
267 case CNID_DBD_RES_ERR_DB:
275 /* ------------------- */
276 static ssize_t dbd_read(int socket, void *data, const size_t length)
282 while (stored < length) {
283 len = read(socket, (u_int8_t *) data + stored, length - stored);
297 /* ---------------------
298 * send a request and get reply
299 * assume send is non blocking
300 * if no answer after sometime (at least MAX_DELAY secondes) return an error
302 static int dbd_rpc(CNID_private *db, struct cnid_dbd_rqst *rqst, struct cnid_dbd_rply *rply)
308 LOG(log_maxdebug, logtype_cnid, "dbd_rpc: BEGIN");
310 if (send_packet(db, rqst) < 0) {
314 nametmp = rply->name;
316 ret = dbd_read(db->fd, rply, sizeof(struct cnid_dbd_rply));
318 if (ret != sizeof(struct cnid_dbd_rply)) {
319 LOG(log_error, logtype_cnid, "dbd_rpc: Error reading header from fd (db_dir %s): %s",
320 db->db_dir, ret == -1?strerror(errno):"closed");
321 rply->name = nametmp;
324 rply->name = nametmp;
325 if (rply->namelen && rply->namelen > len) {
326 LOG(log_error, logtype_cnid,
327 "dbd_rpc: Error reading name (db_dir %s): %s name too long: %d. only wanted %d, garbage?",
328 db->db_dir, rply->name, rply->namelen, len);
331 if (rply->namelen && (ret = dbd_read(db->fd, rply->name, rply->namelen)) != (ssize_t)rply->namelen) {
332 LOG(log_error, logtype_cnid, "dbd_rpc: Error reading name from fd (db_dir %s): %s",
333 db->db_dir, ret == -1?strerror(errno):"closed");
337 LOG(log_maxdebug, logtype_cnid, "dbd_rpc: END");
342 /* -------------------- */
343 static int transmit(CNID_private *db, struct cnid_dbd_rqst *rqst, struct cnid_dbd_rply *rply)
346 int clean = 1; /* no errors so far - to prevent sleep on first try */
348 LOG(log_debug7, logtype_cnid, "transmit: BEGIN");
351 /* volume and db don't have the same timestamp
357 struct cnid_dbd_rqst rqst_stamp;
358 struct cnid_dbd_rply rply_stamp;
359 char stamp[ADEDLEN_PRIVSYN];
364 LOG(log_debug, logtype_cnid, "transmit: connecting to cnid_dbd ...");
365 if ((db->fd = init_tsock(db)) < 0) {
367 if (t - orig > MAX_DELAY)
371 dbd_initstamp(&rqst_stamp);
372 memset(stamp, 0, ADEDLEN_PRIVSYN);
373 rply_stamp.name = stamp;
374 rply_stamp.namelen = ADEDLEN_PRIVSYN;
376 if (dbd_rpc(db, &rqst_stamp, &rply_stamp) < 0)
378 if (dbd_reply_stamp(&rply_stamp ) < 0)
382 LOG(log_debug7, logtype_cnid, "transmit: reconnected to cnid_dbd, comparing database stamps...");
383 if (memcmp(stamp, db->stamp, ADEDLEN_PRIVSYN)) {
384 LOG(log_error, logtype_cnid, "transmit: ... not the same db!");
388 LOG(log_debug7, logtype_cnid, "transmit: ... OK.");
392 if (db->client_stamp)
393 memcpy(db->client_stamp, stamp, ADEDLEN_PRIVSYN);
394 memcpy(db->stamp, stamp, ADEDLEN_PRIVSYN);
396 LOG(log_debug, logtype_cnid, "transmit: succesfully attached to cnid_dbd for volume '%s' with stamp '%08lx'.",
397 db->db_dir, *(uint64_t *)stamp);
399 if (!dbd_rpc(db, rqst, rply)) {
400 LOG(log_debug7, logtype_cnid, "transmit: END OK");
406 db->fd = -1; /* FD not valid... will need to reconnect */
409 if (!clean) { /* don't sleep if just got disconnected by cnid server */
411 if (t - orig > MAX_DELAY) {
412 LOG(log_error, logtype_cnid, "transmit: Request to dbd daemon (db_dir %s) timed out.", db->db_dir);
415 /* sleep a little before retry */
418 clean = 0; /* false... next time sleep */
425 /* ---------------------- */
426 static struct _cnid_db *cnid_dbd_new(const char *volpath)
428 struct _cnid_db *cdb;
430 if ((cdb = (struct _cnid_db *)calloc(1, sizeof(struct _cnid_db))) == NULL)
433 if ((cdb->volpath = strdup(volpath)) == NULL) {
438 cdb->flags = CNID_FLAG_PERSISTENT | CNID_FLAG_LAZY_INIT;
440 cdb->cnid_add = cnid_dbd_add;
441 cdb->cnid_delete = cnid_dbd_delete;
442 cdb->cnid_get = cnid_dbd_get;
443 cdb->cnid_lookup = cnid_dbd_lookup;
444 cdb->cnid_nextid = NULL;
445 cdb->cnid_resolve = cnid_dbd_resolve;
446 cdb->cnid_getstamp = cnid_dbd_getstamp;
447 cdb->cnid_update = cnid_dbd_update;
448 cdb->cnid_rebuild_add = cnid_dbd_rebuild_add;
449 cdb->cnid_close = cnid_dbd_close;
454 /* ---------------------- */
455 struct _cnid_db *cnid_dbd_open(const char *dir, mode_t mask _U_)
457 CNID_private *db = NULL;
458 struct _cnid_db *cdb = NULL;
464 if ((cdb = cnid_dbd_new(dir)) == NULL) {
465 LOG(log_error, logtype_cnid, "cnid_open: Unable to allocate memory for database");
469 if ((db = (CNID_private *)calloc(1, sizeof(CNID_private))) == NULL) {
470 LOG(log_error, logtype_cnid, "cnid_open: Unable to allocate memory for database");
471 goto cnid_dbd_open_fail;
476 /* We keep a copy of the directory in the db structure so that we can
477 transparently reconnect later. */
478 strcpy(db->db_dir, dir);
479 db->magic = CNID_DB_MAGIC;
482 LOG(log_debug, logtype_cnid, "cnid_dbd_open: Finished initializing cnid dbd module for volume '%s'", db->db_dir);
488 if (cdb->volpath != NULL) {
499 /* ---------------------- */
500 void cnid_dbd_close(struct _cnid_db *cdb)
505 LOG(log_error, logtype_cnid, "cnid_close called with NULL argument !");
509 if ((db = cdb->_private) != NULL) {
510 LOG(log_info, logtype_cnid, "closing database connection for volume '%s'", db->db_dir);
523 /* ---------------------- */
524 cnid_t cnid_dbd_add(struct _cnid_db *cdb, const struct stat *st,
525 const cnid_t did, char *name, const size_t len,
529 struct cnid_dbd_rqst rqst;
530 struct cnid_dbd_rply rply;
533 if (!cdb || !(db = cdb->_private) || !st || !name) {
534 LOG(log_error, logtype_cnid, "cnid_add: Parameter error");
535 errno = CNID_ERR_PARAM;
539 if (len > MAXPATHLEN) {
540 LOG(log_error, logtype_cnid, "cnid_add: Path name is too long");
541 errno = CNID_ERR_PATH;
546 rqst.op = CNID_DBD_OP_ADD;
548 if (!(cdb->flags & CNID_FLAG_NODEV)) {
549 rqst.dev = st->st_dev;
552 rqst.ino = st->st_ino;
553 rqst.type = S_ISDIR(st->st_mode)?1:0;
558 LOG(log_debug, logtype_cnid, "cnid_dbd_add: CNID: %u, name: '%s', inode: 0x%llx, type: %d (0=file, 1=dir)",
559 ntohl(did), name, (long long)st->st_ino, rqst.type);
562 if (transmit(db, &rqst, &rply) < 0) {
567 switch(rply.result) {
568 case CNID_DBD_RES_OK:
570 LOG(log_debug, logtype_cnid, "cnid_dbd_add: got CNID: %u", ntohl(id));
572 case CNID_DBD_RES_ERR_MAX:
573 errno = CNID_ERR_MAX;
576 case CNID_DBD_RES_ERR_DB:
577 case CNID_DBD_RES_ERR_DUPLCNID:
588 /* ---------------------- */
589 cnid_t cnid_dbd_get(struct _cnid_db *cdb, const cnid_t did, char *name, const size_t len)
592 struct cnid_dbd_rqst rqst;
593 struct cnid_dbd_rply rply;
596 if (!cdb || !(db = cdb->_private) || !name) {
597 LOG(log_error, logtype_cnid, "cnid_dbd_get: Parameter error");
598 errno = CNID_ERR_PARAM;
602 if (len > MAXPATHLEN) {
603 LOG(log_error, logtype_cnid, "cnid_dbd_get: Path name is too long");
604 errno = CNID_ERR_PATH;
608 LOG(log_debug, logtype_cnid, "cnid_dbd_get: DID: %u, name: '%s'", ntohl(did), name);
611 rqst.op = CNID_DBD_OP_GET;
617 if (transmit(db, &rqst, &rply) < 0) {
622 switch(rply.result) {
623 case CNID_DBD_RES_OK:
625 LOG(log_debug, logtype_cnid, "cnid_dbd_get: got CNID: %u", ntohl(id));
627 case CNID_DBD_RES_NOTFOUND:
630 case CNID_DBD_RES_ERR_DB:
641 /* ---------------------- */
642 char *cnid_dbd_resolve(struct _cnid_db *cdb, cnid_t *id, void *buffer, size_t len)
645 struct cnid_dbd_rqst rqst;
646 struct cnid_dbd_rply rply;
649 if (!cdb || !(db = cdb->_private) || !id || !(*id)) {
650 LOG(log_error, logtype_cnid, "cnid_resolve: Parameter error");
651 errno = CNID_ERR_PARAM;
655 LOG(log_debug, logtype_cnid, "cnid_dbd_resolve: resolving CNID: %u", ntohl(*id));
657 /* TODO: We should maybe also check len. At the moment we rely on the caller
658 to provide a buffer that is large enough for MAXPATHLEN plus
659 CNID_HEADER_LEN plus 1 byte, which is large enough for the maximum that
660 can come from the database. */
663 rqst.op = CNID_DBD_OP_RESOLVE;
666 /* This mimicks the behaviour of the "regular" cnid_resolve. So far,
667 nobody uses the content of buffer. It only provides space for the
668 name in the caller. */
669 rply.name = (char *)buffer + CNID_HEADER_LEN;
670 rply.namelen = len - CNID_HEADER_LEN;
672 if (transmit(db, &rqst, &rply) < 0) {
678 switch (rply.result) {
679 case CNID_DBD_RES_OK:
682 LOG(log_debug, logtype_cnid, "cnid_dbd_resolve: resolved did: %u, name: '%s'", ntohl(*id), name);
684 case CNID_DBD_RES_NOTFOUND:
688 case CNID_DBD_RES_ERR_DB:
700 /* ---------------------- */
701 int cnid_dbd_getstamp(struct _cnid_db *cdb, void *buffer, const size_t len)
705 if (!cdb || !(db = cdb->_private) || len != ADEDLEN_PRIVSYN) {
706 LOG(log_error, logtype_cnid, "cnid_getstamp: Parameter error");
707 errno = CNID_ERR_PARAM;
710 db->client_stamp = buffer;
711 db->stamp_size = len;
712 memset(buffer,0, len);
716 /* ---------------------- */
717 cnid_t cnid_dbd_lookup(struct _cnid_db *cdb, const struct stat *st, const cnid_t did,
718 char *name, const size_t len)
721 struct cnid_dbd_rqst rqst;
722 struct cnid_dbd_rply rply;
725 if (!cdb || !(db = cdb->_private) || !st || !name) {
726 LOG(log_error, logtype_cnid, "cnid_lookup: Parameter error");
727 errno = CNID_ERR_PARAM;
731 if (len > MAXPATHLEN) {
732 LOG(log_error, logtype_cnid, "cnid_lookup: Path name is too long");
733 errno = CNID_ERR_PATH;
738 rqst.op = CNID_DBD_OP_LOOKUP;
740 if (!(cdb->flags & CNID_FLAG_NODEV)) {
741 rqst.dev = st->st_dev;
744 rqst.ino = st->st_ino;
745 rqst.type = S_ISDIR(st->st_mode)?1:0;
750 LOG(log_debug, logtype_cnid, "cnid_dbd_lookup: CNID: %u, name: '%s', inode: 0x%llx, type: %d (0=file, 1=dir)",
751 ntohl(did), name, (long long)st->st_ino, rqst.type);
754 if (transmit(db, &rqst, &rply) < 0) {
759 switch (rply.result) {
760 case CNID_DBD_RES_OK:
762 LOG(log_debug, logtype_cnid, "cnid_dbd_lookup: got CNID: %u", ntohl(id));
764 case CNID_DBD_RES_NOTFOUND:
767 case CNID_DBD_RES_ERR_DB:
778 /* ---------------------- */
779 int cnid_dbd_update(struct _cnid_db *cdb, const cnid_t id, const struct stat *st,
780 const cnid_t did, char *name, const size_t len)
783 struct cnid_dbd_rqst rqst;
784 struct cnid_dbd_rply rply;
786 if (!cdb || !(db = cdb->_private) || !id || !st || !name) {
787 LOG(log_error, logtype_cnid, "cnid_update: Parameter error");
788 errno = CNID_ERR_PARAM;
792 if (len > MAXPATHLEN) {
793 LOG(log_error, logtype_cnid, "cnid_update: Path name is too long");
794 errno = CNID_ERR_PATH;
799 rqst.op = CNID_DBD_OP_UPDATE;
801 if (!(cdb->flags & CNID_FLAG_NODEV)) {
802 rqst.dev = st->st_dev;
804 rqst.ino = st->st_ino;
805 rqst.type = S_ISDIR(st->st_mode)?1:0;
810 LOG(log_debug, logtype_cnid, "cnid_dbd_update: CNID: %u, name: '%s', inode: 0x%llx, type: %d (0=file, 1=dir)",
811 ntohl(id), name, (long long)st->st_ino, rqst.type);
814 if (transmit(db, &rqst, &rply) < 0) {
819 switch (rply.result) {
820 case CNID_DBD_RES_OK:
821 LOG(log_debug, logtype_cnid, "cnid_dbd_update: updated");
822 case CNID_DBD_RES_NOTFOUND:
824 case CNID_DBD_RES_ERR_DB:
832 /* ---------------------- */
833 cnid_t cnid_dbd_rebuild_add(struct _cnid_db *cdb, const struct stat *st,
834 const cnid_t did, char *name, const size_t len,
838 struct cnid_dbd_rqst rqst;
839 struct cnid_dbd_rply rply;
842 if (!cdb || !(db = cdb->_private) || !st || !name || hint == CNID_INVALID) {
843 LOG(log_error, logtype_cnid, "cnid_rebuild_add: Parameter error");
844 errno = CNID_ERR_PARAM;
848 if (len > MAXPATHLEN) {
849 LOG(log_error, logtype_cnid, "cnid_rebuild_add: Path name is too long");
850 errno = CNID_ERR_PATH;
855 rqst.op = CNID_DBD_OP_REBUILD_ADD;
857 if (!(cdb->flags & CNID_FLAG_NODEV)) {
858 rqst.dev = st->st_dev;
861 rqst.ino = st->st_ino;
862 rqst.type = S_ISDIR(st->st_mode)?1:0;
868 LOG(log_debug, logtype_cnid, "cnid_dbd_rebuild_add: CNID: %u, name: '%s', inode: 0x%llx, type: %d (0=file, 1=dir), hint: %u",
869 ntohl(did), name, (long long)st->st_ino, rqst.type, hint);
871 if (transmit(db, &rqst, &rply) < 0) {
876 switch(rply.result) {
877 case CNID_DBD_RES_OK:
879 LOG(log_debug, logtype_cnid, "cnid_dbd_rebuild_add: got CNID: %u", ntohl(id));
881 case CNID_DBD_RES_ERR_MAX:
882 errno = CNID_ERR_MAX;
885 case CNID_DBD_RES_ERR_DB:
886 case CNID_DBD_RES_ERR_DUPLCNID:
896 /* ---------------------- */
897 int cnid_dbd_delete(struct _cnid_db *cdb, const cnid_t id)
900 struct cnid_dbd_rqst rqst;
901 struct cnid_dbd_rply rply;
903 if (!cdb || !(db = cdb->_private) || !id) {
904 LOG(log_error, logtype_cnid, "cnid_delete: Parameter error");
905 errno = CNID_ERR_PARAM;
909 LOG(log_debug, logtype_cnid, "cnid_dbd_delete: delete CNID: %u", ntohl(id));
912 rqst.op = CNID_DBD_OP_DELETE;
916 if (transmit(db, &rqst, &rply) < 0) {
921 switch (rply.result) {
922 case CNID_DBD_RES_OK:
923 LOG(log_debug, logtype_cnid, "cnid_dbd_delete: deleted CNID: %u", ntohl(id));
924 case CNID_DBD_RES_NOTFOUND:
926 case CNID_DBD_RES_ERR_DB:
935 struct _cnid_module cnid_dbd_module = {
942 #endif /* CNID_DBD */