2 * $Id: cnid_dbd.c,v 1.9 2009-07-21 13:41:16 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);
114 LOG(log_debug7, logtype_cnid, "tsock_getfd: using sockfd %d for cnid server '%s:%d'", sock, host, port);
119 /* --------------------- */
120 static int write_vec(int fd, struct iovec *iov, size_t towrite)
125 LOG(log_maxdebug, logtype_cnid, "write_vec: request to write %d bytes", towrite);
127 len1 = iov[1].iov_len;
128 while (towrite > 0) {
129 if (((len = writev(fd, iov, 2)) == -1 && errno == EINTR) || !len)
132 if ((size_t)len == towrite) /* wrote everything out */
134 else if (len < 0) { /* error */
139 if (towrite > len1) { /* skip part of header */
140 iov[0].iov_base = (char *) iov[0].iov_base + len;
141 iov[0].iov_len -= len;
142 } else { /* skip to data */
143 if (iov[0].iov_len) {
144 len -= iov[0].iov_len;
147 iov[1].iov_base = (char *) iov[1].iov_base + len;
148 iov[1].iov_len -= len;
152 LOG(log_maxdebug, logtype_cnid, "write_vec: wrote %d bytes", towrite);
157 /* --------------------- */
158 static int init_tsock(CNID_private *db)
164 LOG(log_debug, logtype_cnid, "init_tsock: BEGIN. Opening volume '%s'", db->db_dir);
166 if ((fd = tsock_getfd(Cnid_srv, Cnid_port)) < 0)
169 len = strlen(db->db_dir);
171 iov[0].iov_base = &len;
172 iov[0].iov_len = sizeof(int);
174 iov[1].iov_base = db->db_dir;
175 iov[1].iov_len = len;
177 if (write_vec(fd, iov, len + sizeof(int)) < 0) {
178 LOG(log_error, logtype_cnid, "init_tsock: Error/short write: %s", strerror(errno));
183 LOG(log_debug, logtype_cnid, "init_tsock: ok");
188 /* --------------------- */
189 static int send_packet(CNID_private *db, struct cnid_dbd_rqst *rqst)
194 LOG(log_maxdebug, logtype_cnid, "send_packet: BEGIN");
196 if (!rqst->namelen) {
197 if (write(db->fd, rqst, sizeof(struct cnid_dbd_rqst)) != sizeof(struct cnid_dbd_rqst)) {
198 LOG(log_warning, logtype_cnid, "send_packet: Error/short write rqst (db_dir %s): %s",
199 db->db_dir, strerror(errno));
202 LOG(log_maxdebug, logtype_cnid, "send_packet: OK");
206 iov[0].iov_base = rqst;
207 iov[0].iov_len = sizeof(struct cnid_dbd_rqst);
209 iov[1].iov_base = rqst->name;
210 iov[1].iov_len = rqst->namelen;
212 towrite = sizeof(struct cnid_dbd_rqst) +rqst->namelen;
214 if (write_vec(db->fd, iov, towrite) < 0) {
215 LOG(log_warning, logtype_cnid, "send_packet: Error writev rqst (db_dir %s): %s",
216 db->db_dir, strerror(errno));
220 LOG(log_maxdebug, logtype_cnid, "send_packet: OK");
224 /* ------------------- */
225 static void dbd_initstamp(struct cnid_dbd_rqst *rqst)
228 rqst->op = CNID_DBD_OP_GETSTAMP;
231 /* ------------------- */
232 static int dbd_reply_stamp(struct cnid_dbd_rply *rply)
234 switch (rply->result) {
235 case CNID_DBD_RES_OK:
237 case CNID_DBD_RES_NOTFOUND:
239 case CNID_DBD_RES_ERR_DB:
247 /* ---------------------
248 * send a request and get reply
249 * assume send is non blocking
250 * if no answer after sometime (at least MAX_DELAY secondes) return an error
253 static int dbd_rpc(CNID_private *db, struct cnid_dbd_rqst *rqst, struct cnid_dbd_rply *rply)
262 LOG(log_maxdebug, logtype_cnid, "dbd_rpc: BEGIN");
264 if (send_packet(db, rqst) < 0) {
268 FD_SET(db->fd, &readfds);
272 tv.tv_sec = MAX_DELAY;
273 while ((ret = select(maxfd + 1, &readfds, NULL, NULL, &tv)) < 0 && errno == EINTR);
276 LOG(log_error, logtype_cnid, "dbd_rpc: Error in select (db_dir %s): %s",
277 db->db_dir, strerror(errno));
283 LOG(log_error, logtype_cnid, "dbd_rpc: select timed out (db_dir %s)",
288 nametmp = rply->name;
289 /* assume that if we have something then everything is there (doesn't sleep) */
290 if ((ret = read(db->fd, rply, sizeof(struct cnid_dbd_rply))) != sizeof(struct cnid_dbd_rply)) {
291 LOG(log_error, logtype_cnid, "dbd_rpc: Error reading header from fd (db_dir %s): %s",
292 db->db_dir, ret == -1?strerror(errno):"closed");
293 rply->name = nametmp;
296 rply->name = nametmp;
297 if (rply->namelen && rply->namelen > len) {
298 LOG(log_error, logtype_cnid,
299 "dbd_rpc: Error reading name (db_dir %s): %s name too long: %d. only wanted %d, garbage?",
300 db->db_dir, rply->name, rply->namelen, len);
303 if (rply->namelen && (ret = read(db->fd, rply->name, rply->namelen)) != (ssize_t)rply->namelen) {
304 LOG(log_error, logtype_cnid, "dbd_rpc: Error reading name from fd (db_dir %s): %s",
305 db->db_dir, ret == -1?strerror(errno):"closed");
309 LOG(log_maxdebug, logtype_cnid, "dbd_rpc: END");
314 /* -------------------- */
315 static int transmit(CNID_private *db, struct cnid_dbd_rqst *rqst, struct cnid_dbd_rply *rply)
319 int clean = 1; /* no errors so far - to prevent sleep on first try */
321 LOG(log_debug7, logtype_cnid, "transmit: BEGIN");
324 /* volume and db don't have the same timestamp
331 struct cnid_dbd_rqst rqst_stamp;
332 struct cnid_dbd_rply rply_stamp;
333 char stamp[ADEDLEN_PRIVSYN];
335 LOG(log_debug, logtype_cnid, "transmit: connecting to cnid_dbd ...");
336 if ((db->fd = init_tsock(db)) < 0) {
338 if (t - orig > MAX_DELAY)
342 dbd_initstamp(&rqst_stamp);
343 memset(stamp, 0, ADEDLEN_PRIVSYN);
344 rply_stamp.name = stamp;
345 rply_stamp.namelen = ADEDLEN_PRIVSYN;
347 if (dbd_rpc(db, &rqst_stamp, &rply_stamp) < 0)
349 if (dbd_reply_stamp(&rply_stamp ) < 0)
353 LOG(log_debug7, logtype_cnid, "transmit: reconnected to cnid_dbd, comparing database stamps...");
354 if (memcmp(stamp, db->stamp, ADEDLEN_PRIVSYN)) {
355 LOG(log_error, logtype_cnid, "transmit: ... not the same db!");
359 LOG(log_debug7, logtype_cnid, "transmit: ... OK.");
363 if (db->client_stamp)
364 memcpy(db->client_stamp, stamp, ADEDLEN_PRIVSYN);
365 memcpy(db->stamp, stamp, ADEDLEN_PRIVSYN);
367 LOG(log_debug, logtype_cnid, "transmit: succesfully attached to cnid_dbd for volume '%s' with stamp '%08lx'.",
368 db->db_dir, *(uint64_t *)stamp);
370 if (!dbd_rpc(db, rqst, rply)) {
371 LOG(log_debug7, logtype_cnid, "transmit: END OK");
375 LOG(log_error, logtype_cnid, "transmit: something went wrong");
378 db->fd = -1; /* FD not valid... will need to reconnect */
381 if (t - orig > MAX_DELAY) {
382 LOG(log_error, logtype_cnid, "transmit: Request to dbd daemon (db_dir %s) timed out.", db->db_dir);
386 if (!clean) { /* don't sleep if just got disconnected by cnid server */
387 /* sleep a little before retry */
390 select(0, NULL, NULL, NULL, &tv); /* sleep for 5 seconds */
392 clean = 0; /* false... next time sleep */
398 /* ---------------------- */
399 static struct _cnid_db *cnid_dbd_new(const char *volpath)
401 struct _cnid_db *cdb;
403 if ((cdb = (struct _cnid_db *)calloc(1, sizeof(struct _cnid_db))) == NULL)
406 if ((cdb->volpath = strdup(volpath)) == NULL) {
411 cdb->flags = CNID_FLAG_PERSISTENT | CNID_FLAG_LAZY_INIT;
413 cdb->cnid_add = cnid_dbd_add;
414 cdb->cnid_delete = cnid_dbd_delete;
415 cdb->cnid_get = cnid_dbd_get;
416 cdb->cnid_lookup = cnid_dbd_lookup;
417 cdb->cnid_nextid = NULL;
418 cdb->cnid_resolve = cnid_dbd_resolve;
419 cdb->cnid_getstamp = cnid_dbd_getstamp;
420 cdb->cnid_update = cnid_dbd_update;
421 cdb->cnid_rebuild_add = cnid_dbd_rebuild_add;
422 cdb->cnid_close = cnid_dbd_close;
427 /* ---------------------- */
428 struct _cnid_db *cnid_dbd_open(const char *dir, mode_t mask _U_)
430 CNID_private *db = NULL;
431 struct _cnid_db *cdb = NULL;
437 if ((cdb = cnid_dbd_new(dir)) == NULL) {
438 LOG(log_error, logtype_cnid, "cnid_open: Unable to allocate memory for database");
442 if ((db = (CNID_private *)calloc(1, sizeof(CNID_private))) == NULL) {
443 LOG(log_error, logtype_cnid, "cnid_open: Unable to allocate memory for database");
444 goto cnid_dbd_open_fail;
449 /* We keep a copy of the directory in the db structure so that we can
450 transparently reconnect later. */
451 strcpy(db->db_dir, dir);
452 db->magic = CNID_DB_MAGIC;
455 LOG(log_debug, logtype_cnid, "cnid_dbd_open: Finished initializing cnid dbd module for volume '%s'", db->db_dir);
461 if (cdb->volpath != NULL) {
472 /* ---------------------- */
473 void cnid_dbd_close(struct _cnid_db *cdb)
478 LOG(log_error, logtype_cnid, "cnid_close called with NULL argument !");
482 if ((db = cdb->_private) != NULL) {
483 LOG(log_info, logtype_cnid, "closing database connection for volume '%s'", db->db_dir);
496 /* ---------------------- */
497 cnid_t cnid_dbd_add(struct _cnid_db *cdb, const struct stat *st,
498 const cnid_t did, char *name, const size_t len,
502 struct cnid_dbd_rqst rqst;
503 struct cnid_dbd_rply rply;
506 if (!cdb || !(db = cdb->_private) || !st || !name) {
507 LOG(log_error, logtype_cnid, "cnid_add: Parameter error");
508 errno = CNID_ERR_PARAM;
512 if (len > MAXPATHLEN) {
513 LOG(log_error, logtype_cnid, "cnid_add: Path name is too long");
514 errno = CNID_ERR_PATH;
519 rqst.op = CNID_DBD_OP_ADD;
521 if (!(cdb->flags & CNID_FLAG_NODEV)) {
522 rqst.dev = st->st_dev;
525 rqst.ino = st->st_ino;
526 rqst.type = S_ISDIR(st->st_mode)?1:0;
531 LOG(log_debug, logtype_cnid, "cnid_dbd_add: CNID: %u, name: '%s', inode: 0x%llx, type: %d (0=file, 1=dir)",
532 ntohl(did), name, (long long)st->st_ino, rqst.type);
535 if (transmit(db, &rqst, &rply) < 0) {
540 switch(rply.result) {
541 case CNID_DBD_RES_OK:
543 LOG(log_debug, logtype_cnid, "cnid_dbd_add: got CNID: %u", ntohl(id));
545 case CNID_DBD_RES_ERR_MAX:
546 errno = CNID_ERR_MAX;
549 case CNID_DBD_RES_ERR_DB:
550 case CNID_DBD_RES_ERR_DUPLCNID:
561 /* ---------------------- */
562 cnid_t cnid_dbd_get(struct _cnid_db *cdb, const cnid_t did, char *name, const size_t len)
565 struct cnid_dbd_rqst rqst;
566 struct cnid_dbd_rply rply;
569 if (!cdb || !(db = cdb->_private) || !name) {
570 LOG(log_error, logtype_cnid, "cnid_dbd_get: Parameter error");
571 errno = CNID_ERR_PARAM;
575 if (len > MAXPATHLEN) {
576 LOG(log_error, logtype_cnid, "cnid_dbd_get: Path name is too long");
577 errno = CNID_ERR_PATH;
581 LOG(log_debug, logtype_cnid, "cnid_dbd_get: DID: %u, name: '%s'", ntohl(did), name);
584 rqst.op = CNID_DBD_OP_GET;
590 if (transmit(db, &rqst, &rply) < 0) {
595 switch(rply.result) {
596 case CNID_DBD_RES_OK:
598 LOG(log_debug, logtype_cnid, "cnid_dbd_get: got CNID: %u", ntohl(id));
600 case CNID_DBD_RES_NOTFOUND:
603 case CNID_DBD_RES_ERR_DB:
614 /* ---------------------- */
615 char *cnid_dbd_resolve(struct _cnid_db *cdb, cnid_t *id, void *buffer, size_t len)
618 struct cnid_dbd_rqst rqst;
619 struct cnid_dbd_rply rply;
622 if (!cdb || !(db = cdb->_private) || !id || !(*id)) {
623 LOG(log_error, logtype_cnid, "cnid_resolve: Parameter error");
624 errno = CNID_ERR_PARAM;
628 LOG(log_debug, logtype_cnid, "cnid_dbd_resolve: resolving CNID: %u", ntohl(*id));
630 /* TODO: We should maybe also check len. At the moment we rely on the caller
631 to provide a buffer that is large enough for MAXPATHLEN plus
632 CNID_HEADER_LEN plus 1 byte, which is large enough for the maximum that
633 can come from the database. */
636 rqst.op = CNID_DBD_OP_RESOLVE;
639 /* This mimicks the behaviour of the "regular" cnid_resolve. So far,
640 nobody uses the content of buffer. It only provides space for the
641 name in the caller. */
642 rply.name = (char *)buffer + CNID_HEADER_LEN;
643 rply.namelen = len - CNID_HEADER_LEN;
645 if (transmit(db, &rqst, &rply) < 0) {
651 switch (rply.result) {
652 case CNID_DBD_RES_OK:
655 LOG(log_debug, logtype_cnid, "cnid_dbd_resolve: resolved did: %u, name: '%s'", ntohl(*id), name);
657 case CNID_DBD_RES_NOTFOUND:
661 case CNID_DBD_RES_ERR_DB:
673 /* ---------------------- */
674 int cnid_dbd_getstamp(struct _cnid_db *cdb, void *buffer, const size_t len)
678 if (!cdb || !(db = cdb->_private) || len != ADEDLEN_PRIVSYN) {
679 LOG(log_error, logtype_cnid, "cnid_getstamp: Parameter error");
680 errno = CNID_ERR_PARAM;
683 db->client_stamp = buffer;
684 db->stamp_size = len;
685 memset(buffer,0, len);
689 /* ---------------------- */
690 cnid_t cnid_dbd_lookup(struct _cnid_db *cdb, const struct stat *st, const cnid_t did,
691 char *name, const size_t len)
694 struct cnid_dbd_rqst rqst;
695 struct cnid_dbd_rply rply;
698 if (!cdb || !(db = cdb->_private) || !st || !name) {
699 LOG(log_error, logtype_cnid, "cnid_lookup: Parameter error");
700 errno = CNID_ERR_PARAM;
704 if (len > MAXPATHLEN) {
705 LOG(log_error, logtype_cnid, "cnid_lookup: Path name is too long");
706 errno = CNID_ERR_PATH;
711 rqst.op = CNID_DBD_OP_LOOKUP;
713 if (!(cdb->flags & CNID_FLAG_NODEV)) {
714 rqst.dev = st->st_dev;
717 rqst.ino = st->st_ino;
718 rqst.type = S_ISDIR(st->st_mode)?1:0;
723 LOG(log_debug, logtype_cnid, "cnid_dbd_lookup: CNID: %u, name: '%s', inode: 0x%llx, type: %d (0=file, 1=dir)",
724 ntohl(did), name, (long long)st->st_ino, rqst.type);
727 if (transmit(db, &rqst, &rply) < 0) {
732 switch (rply.result) {
733 case CNID_DBD_RES_OK:
735 LOG(log_debug, logtype_cnid, "cnid_dbd_lookup: got CNID: %u", ntohl(id));
737 case CNID_DBD_RES_NOTFOUND:
740 case CNID_DBD_RES_ERR_DB:
751 /* ---------------------- */
752 int cnid_dbd_update(struct _cnid_db *cdb, const cnid_t id, const struct stat *st,
753 const cnid_t did, char *name, const size_t len)
756 struct cnid_dbd_rqst rqst;
757 struct cnid_dbd_rply rply;
759 if (!cdb || !(db = cdb->_private) || !id || !st || !name) {
760 LOG(log_error, logtype_cnid, "cnid_update: Parameter error");
761 errno = CNID_ERR_PARAM;
765 if (len > MAXPATHLEN) {
766 LOG(log_error, logtype_cnid, "cnid_update: Path name is too long");
767 errno = CNID_ERR_PATH;
772 rqst.op = CNID_DBD_OP_UPDATE;
774 if (!(cdb->flags & CNID_FLAG_NODEV)) {
775 rqst.dev = st->st_dev;
777 rqst.ino = st->st_ino;
778 rqst.type = S_ISDIR(st->st_mode)?1:0;
783 LOG(log_debug, logtype_cnid, "cnid_dbd_update: CNID: %u, name: '%s', inode: 0x%llx, type: %d (0=file, 1=dir)",
784 ntohl(id), name, (long long)st->st_ino, rqst.type);
787 if (transmit(db, &rqst, &rply) < 0) {
792 switch (rply.result) {
793 case CNID_DBD_RES_OK:
794 LOG(log_debug, logtype_cnid, "cnid_dbd_update: updated");
795 case CNID_DBD_RES_NOTFOUND:
797 case CNID_DBD_RES_ERR_DB:
805 /* ---------------------- */
806 cnid_t cnid_dbd_rebuild_add(struct _cnid_db *cdb, const struct stat *st,
807 const cnid_t did, char *name, const size_t len,
811 struct cnid_dbd_rqst rqst;
812 struct cnid_dbd_rply rply;
815 if (!cdb || !(db = cdb->_private) || !st || !name || hint == CNID_INVALID) {
816 LOG(log_error, logtype_cnid, "cnid_rebuild_add: Parameter error");
817 errno = CNID_ERR_PARAM;
821 if (len > MAXPATHLEN) {
822 LOG(log_error, logtype_cnid, "cnid_rebuild_add: Path name is too long");
823 errno = CNID_ERR_PATH;
828 rqst.op = CNID_DBD_OP_REBUILD_ADD;
830 if (!(cdb->flags & CNID_FLAG_NODEV)) {
831 rqst.dev = st->st_dev;
834 rqst.ino = st->st_ino;
835 rqst.type = S_ISDIR(st->st_mode)?1:0;
841 LOG(log_debug, logtype_cnid, "cnid_dbd_rebuild_add: CNID: %u, name: '%s', inode: 0x%llx, type: %d (0=file, 1=dir), hint: %u",
842 ntohl(did), name, (long long)st->st_ino, rqst.type, hint);
844 if (transmit(db, &rqst, &rply) < 0) {
849 switch(rply.result) {
850 case CNID_DBD_RES_OK:
852 LOG(log_debug, logtype_cnid, "cnid_dbd_rebuild_add: got CNID: %u", ntohl(id));
854 case CNID_DBD_RES_ERR_MAX:
855 errno = CNID_ERR_MAX;
858 case CNID_DBD_RES_ERR_DB:
859 case CNID_DBD_RES_ERR_DUPLCNID:
869 /* ---------------------- */
870 int cnid_dbd_delete(struct _cnid_db *cdb, const cnid_t id)
873 struct cnid_dbd_rqst rqst;
874 struct cnid_dbd_rply rply;
876 if (!cdb || !(db = cdb->_private) || !id) {
877 LOG(log_error, logtype_cnid, "cnid_delete: Parameter error");
878 errno = CNID_ERR_PARAM;
882 LOG(log_debug, logtype_cnid, "cnid_dbd_delete: delete CNID: %u", ntohl(id));
885 rqst.op = CNID_DBD_OP_DELETE;
889 if (transmit(db, &rqst, &rply) < 0) {
894 switch (rply.result) {
895 case CNID_DBD_RES_OK:
896 LOG(log_debug, logtype_cnid, "cnid_dbd_delete: deleted CNID: %u", ntohl(id));
897 case CNID_DBD_RES_NOTFOUND:
899 case CNID_DBD_RES_ERR_DB:
908 struct _cnid_module cnid_dbd_module = {
915 #endif /* CNID_DBD */