2 * $Id: cnid_dbd.c,v 1.1.4.18.2.1 2005-01-30 20:56:22 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 init_tsock(CNID_private *db)
122 if ((fd = tsock_getfd(Cnid_srv, Cnid_port)) < 0)
124 len = strlen(db->db_dir);
125 if (write(fd, &len, sizeof(int)) != sizeof(int)) {
126 LOG(log_error, logtype_cnid, "init_tsock: Error/short write: %s", strerror(errno));
130 if (write(fd, db->db_dir, len) != len) {
131 LOG(log_error, logtype_cnid, "init_tsock: Error/short write dir: %s", strerror(errno));
138 /* --------------------- */
139 static int send_packet(CNID_private *db, struct cnid_dbd_rqst *rqst, int silent)
145 iov[0].iov_base = rqst;
146 iov[0].iov_len = sizeof(struct cnid_dbd_rqst);
148 if (!rqst->namelen) {
149 if (write(db->fd, rqst, sizeof(struct cnid_dbd_rqst)) != sizeof(struct cnid_dbd_rqst)) {
151 LOG(log_warning, logtype_cnid, "send_packet: Error/short write rqst (db_dir %s): %s",
152 db->db_dir, strerror(errno));
157 iov[1].iov_base = rqst->name;
158 iov[1].iov_len = rqst->namelen;
160 towrite = sizeof(struct cnid_dbd_rqst) +rqst->namelen;
161 while (towrite > 0) {
162 if (((len = writev(db->fd, iov, 2)) == -1 && errno == EINTR) || !len)
165 if (len == towrite) /* wrote everything out */
167 else if (len < 0) { /* error */
169 LOG(log_warning, logtype_cnid, "send_packet: Error writev rqst (db_dir %s): %s",
170 db->db_dir, strerror(errno));
175 if (towrite > rqst->namelen) { /* skip part of header */
176 iov[0].iov_base = (char *) iov[0].iov_base + len;
177 iov[0].iov_len -= len;
178 } else { /* skip to data */
179 if (iov[0].iov_len) {
180 len -= iov[0].iov_len;
183 iov[1].iov_base = (char *) iov[1].iov_base + len;
184 iov[1].iov_len -= len;
190 /* ------------------- */
191 static void dbd_initstamp(struct cnid_dbd_rqst *rqst)
194 rqst->op = CNID_DBD_OP_GETSTAMP;
197 /* ------------------- */
198 static int dbd_reply_stamp(struct cnid_dbd_rply *rply)
200 switch (rply->result) {
201 case CNID_DBD_RES_OK:
203 case CNID_DBD_RES_NOTFOUND:
205 case CNID_DBD_RES_ERR_DB:
213 /* ---------------------
214 * send a request and get reply
215 * assume send is non blocking
216 * if no answer after sometime (at least MAX_DELAY secondes) return an error
219 static int dbd_rpc(CNID_private *db, struct cnid_dbd_rqst *rqst, struct cnid_dbd_rply *rply, int silent)
228 if (send_packet(db, rqst, silent) < 0) {
232 FD_SET(db->fd, &readfds);
236 tv.tv_sec = MAX_DELAY;
237 while ((ret = select(maxfd + 1, &readfds, NULL, NULL, &tv)) < 0 && errno == EINTR);
241 LOG(log_error, logtype_cnid, "dbd_rpc: Error in select (db_dir %s): %s",
242 db->db_dir, strerror(errno));
249 LOG(log_error, logtype_cnid, "dbd_rpc: select timed out (db_dir %s)",
255 nametmp = rply->name;
256 /* assume that if we have something then everything is there (doesn't sleep) */
257 if ((ret = read(db->fd, rply, sizeof(struct cnid_dbd_rply))) != sizeof(struct cnid_dbd_rply)) {
259 LOG(log_error, logtype_cnid, "dbd_rpc: Error reading header from fd (db_dir %s): %s",
260 db->db_dir, ret == -1?strerror(errno):"closed");
261 rply->name = nametmp;
264 rply->name = nametmp;
265 if (rply->namelen && rply->namelen > len) {
267 LOG(log_error, logtype_cnid,
268 "dbd_rpc: Error reading name (db_dir %s): %s name too long wanted %d only %d, garbage?",
269 db->db_dir, rply->namelen, len);
272 if (rply->namelen && (ret = read(db->fd, rply->name, rply->namelen)) != rply->namelen) {
274 LOG(log_error, logtype_cnid, "dbd_rpc: Error reading name from fd (db_dir %s): %s",
275 db->db_dir, ret == -1?strerror(errno):"closed");
281 /* -------------------- */
282 static int transmit(CNID_private *db, struct cnid_dbd_rqst *rqst, struct cnid_dbd_rply *rply)
289 /* volume and db don't have the same timestamp
297 if ((db->fd = init_tsock(db)) < 0) {
299 if (t - orig > MAX_DELAY)
304 struct cnid_dbd_rqst rqst_stamp;
305 struct cnid_dbd_rply rply_stamp;
306 char stamp[ADEDLEN_PRIVSYN];
308 dbd_initstamp(&rqst_stamp);
309 memset(stamp, 0, ADEDLEN_PRIVSYN);
310 rply_stamp.name = stamp;
311 rply_stamp.namelen = ADEDLEN_PRIVSYN;
313 if (dbd_rpc(db, &rqst_stamp, &rply_stamp, silent) < 0)
315 if (dbd_reply_stamp(&rply_stamp ) < 0)
317 if (memcmp(stamp, db->stamp, ADEDLEN_PRIVSYN)) {
318 LOG(log_error, logtype_cnid, "transmit: not the same db!");
325 if (!dbd_rpc(db, rqst, rply, silent))
329 silent = 0; /* From now on dbd_rpc and subroutines called from there
330 will log messages if something goes wrong again */
335 if (t - orig > MAX_DELAY) {
336 LOG(log_error, logtype_cnid, "transmit: Request to dbd daemon (db_dir %s) timed out.", db->db_dir);
340 /* sleep a little before retry */
344 select(0, NULL, NULL, NULL, &tv);
349 /* ---------------------- */
350 static struct _cnid_db *cnid_dbd_new(const char *volpath)
352 struct _cnid_db *cdb;
354 if ((cdb = (struct _cnid_db *)calloc(1, sizeof(struct _cnid_db))) == NULL)
357 if ((cdb->volpath = strdup(volpath)) == NULL) {
362 cdb->flags = CNID_FLAG_PERSISTENT;
364 cdb->cnid_add = cnid_dbd_add;
365 cdb->cnid_delete = cnid_dbd_delete;
366 cdb->cnid_get = cnid_dbd_get;
367 cdb->cnid_lookup = cnid_dbd_lookup;
368 cdb->cnid_nextid = NULL;
369 cdb->cnid_resolve = cnid_dbd_resolve;
370 cdb->cnid_getstamp = cnid_dbd_getstamp;
371 cdb->cnid_update = cnid_dbd_update;
372 cdb->cnid_close = cnid_dbd_close;
377 /* ---------------------- */
378 struct _cnid_db *cnid_dbd_open(const char *dir, mode_t mask)
380 CNID_private *db = NULL;
381 struct _cnid_db *cdb = NULL;
387 if ((cdb = cnid_dbd_new(dir)) == NULL) {
388 LOG(log_error, logtype_default, "cnid_open: Unable to allocate memory for database");
392 if ((db = (CNID_private *)calloc(1, sizeof(CNID_private))) == NULL) {
393 LOG(log_error, logtype_cnid, "cnid_open: Unable to allocate memory for database");
394 goto cnid_dbd_open_fail;
399 /* We keep a copy of the directory in the db structure so that we can
400 transparently reconnect later. */
401 strcpy(db->db_dir, dir);
402 db->magic = CNID_DB_MAGIC;
405 LOG(log_info, logtype_cnid, "opening database connection to %s", db->db_dir);
411 if (cdb->volpath != NULL) {
422 /* ---------------------- */
423 void cnid_dbd_close(struct _cnid_db *cdb)
428 LOG(log_error, logtype_afpd, "cnid_close called with NULL argument !");
432 if ((db = cdb->_private) != NULL) {
434 LOG(log_info, logtype_cnid, "closing database connection to %s", db->db_dir);
447 /* ---------------------- */
448 cnid_t cnid_dbd_add(struct _cnid_db *cdb, const struct stat *st,
449 const cnid_t did, char *name, const int len,
453 struct cnid_dbd_rqst rqst;
454 struct cnid_dbd_rply rply;
457 if (!cdb || !(db = cdb->_private) || !st || !name) {
458 LOG(log_error, logtype_cnid, "cnid_add: Parameter error");
459 errno = CNID_ERR_PARAM;
463 if (len > MAXPATHLEN) {
464 LOG(log_error, logtype_cnid, "cnid_add: Path name is too long");
465 errno = CNID_ERR_PATH;
470 rqst.op = CNID_DBD_OP_ADD;
472 if (!(cdb->flags & CNID_FLAG_NODEV)) {
473 rqst.dev = st->st_dev;
476 rqst.ino = st->st_ino;
477 rqst.type = S_ISDIR(st->st_mode)?1:0;
482 if (transmit(db, &rqst, &rply) < 0) {
487 switch(rply.result) {
488 case CNID_DBD_RES_OK:
491 case CNID_DBD_RES_ERR_MAX:
492 errno = CNID_ERR_MAX;
495 case CNID_DBD_RES_ERR_DB:
496 case CNID_DBD_RES_ERR_DUPLCNID:
506 /* ---------------------- */
507 cnid_t cnid_dbd_get(struct _cnid_db *cdb, const cnid_t did, char *name,
511 struct cnid_dbd_rqst rqst;
512 struct cnid_dbd_rply rply;
516 if (!cdb || !(db = cdb->_private) || !name) {
517 LOG(log_error, logtype_cnid, "cnid_get: Parameter error");
518 errno = CNID_ERR_PARAM;
522 if (len > MAXPATHLEN) {
523 LOG(log_error, logtype_cnid, "cnid_add: Path name is too long");
524 errno = CNID_ERR_PATH;
529 rqst.op = CNID_DBD_OP_GET;
534 if (transmit(db, &rqst, &rply) < 0) {
539 switch(rply.result) {
540 case CNID_DBD_RES_OK:
543 case CNID_DBD_RES_NOTFOUND:
546 case CNID_DBD_RES_ERR_DB:
557 /* ---------------------- */
558 char *cnid_dbd_resolve(struct _cnid_db *cdb, cnid_t *id, void *buffer, u_int32_t len)
561 struct cnid_dbd_rqst rqst;
562 struct cnid_dbd_rply rply;
565 if (!cdb || !(db = cdb->_private) || !id || !(*id)) {
566 LOG(log_error, logtype_cnid, "cnid_resolve: Parameter error");
567 errno = CNID_ERR_PARAM;
571 /* TODO: We should maybe also check len. At the moment we rely on the caller
572 to provide a buffer that is large enough for MAXPATHLEN plus
573 CNID_HEADER_LEN plus 1 byte, which is large enough for the maximum that
574 can come from the database. */
577 rqst.op = CNID_DBD_OP_RESOLVE;
580 /* This mimicks the behaviour of the "regular" cnid_resolve. So far,
581 nobody uses the content of buffer. It only provides space for the
582 name in the caller. */
583 rply.name = (char *)buffer + CNID_HEADER_LEN;
584 rply.namelen = len - CNID_HEADER_LEN;
586 if (transmit(db, &rqst, &rply) < 0) {
592 switch (rply.result) {
593 case CNID_DBD_RES_OK:
597 case CNID_DBD_RES_NOTFOUND:
601 case CNID_DBD_RES_ERR_DB:
613 /* --------------------- */
614 static int dbd_getstamp(CNID_private *db, void *buffer, const size_t len)
616 struct cnid_dbd_rqst rqst;
617 struct cnid_dbd_rply rply;
619 memset(buffer, 0, len);
620 dbd_initstamp(&rqst);
625 if (transmit(db, &rqst, &rply) < 0) {
629 return dbd_reply_stamp(&rply);
632 /* ---------------------- */
633 int cnid_dbd_getstamp(struct _cnid_db *cdb, void *buffer, const int len)
638 if (!cdb || !(db = cdb->_private) || len != ADEDLEN_PRIVSYN) {
639 LOG(log_error, logtype_cnid, "cnid_getstamp: Parameter error");
640 errno = CNID_ERR_PARAM;
643 ret = dbd_getstamp(db, buffer, len);
646 memcpy(db->stamp, buffer, len);
651 /* ---------------------- */
652 cnid_t cnid_dbd_lookup(struct _cnid_db *cdb, const struct stat *st, const cnid_t did,
653 char *name, const int len)
656 struct cnid_dbd_rqst rqst;
657 struct cnid_dbd_rply rply;
660 if (!cdb || !(db = cdb->_private) || !st || !name) {
661 LOG(log_error, logtype_cnid, "cnid_lookup: Parameter error");
662 errno = CNID_ERR_PARAM;
666 if (len > MAXPATHLEN) {
667 LOG(log_error, logtype_cnid, "cnid_lookup: Path name is too long");
668 errno = CNID_ERR_PATH;
673 rqst.op = CNID_DBD_OP_LOOKUP;
675 if (!(cdb->flags & CNID_FLAG_NODEV)) {
676 rqst.dev = st->st_dev;
679 rqst.ino = st->st_ino;
680 rqst.type = S_ISDIR(st->st_mode)?1:0;
685 if (transmit(db, &rqst, &rply) < 0) {
690 switch (rply.result) {
691 case CNID_DBD_RES_OK:
694 case CNID_DBD_RES_NOTFOUND:
697 case CNID_DBD_RES_ERR_DB:
708 /* ---------------------- */
709 int cnid_dbd_update(struct _cnid_db *cdb, const cnid_t id, const struct stat *st,
710 const cnid_t did, char *name, const int len)
713 struct cnid_dbd_rqst rqst;
714 struct cnid_dbd_rply rply;
717 if (!cdb || !(db = cdb->_private) || !id || !st || !name) {
718 LOG(log_error, logtype_cnid, "cnid_update: Parameter error");
719 errno = CNID_ERR_PARAM;
723 if (len > MAXPATHLEN) {
724 LOG(log_error, logtype_cnid, "cnid_update: Path name is too long");
725 errno = CNID_ERR_PATH;
730 rqst.op = CNID_DBD_OP_UPDATE;
732 if (!(cdb->flags & CNID_FLAG_NODEV)) {
733 rqst.dev = st->st_dev;
735 rqst.ino = st->st_ino;
736 rqst.type = S_ISDIR(st->st_mode)?1:0;
741 if (transmit(db, &rqst, &rply) < 0) {
746 switch (rply.result) {
747 case CNID_DBD_RES_OK:
748 case CNID_DBD_RES_NOTFOUND:
750 case CNID_DBD_RES_ERR_DB:
758 /* ---------------------- */
759 int cnid_dbd_delete(struct _cnid_db *cdb, const cnid_t id)
762 struct cnid_dbd_rqst rqst;
763 struct cnid_dbd_rply rply;
766 if (!cdb || !(db = cdb->_private) || !id) {
767 LOG(log_error, logtype_cnid, "cnid_delete: Parameter error");
768 errno = CNID_ERR_PARAM;
773 rqst.op = CNID_DBD_OP_DELETE;
776 if (transmit(db, &rqst, &rply) < 0) {
781 switch (rply.result) {
782 case CNID_DBD_RES_OK:
783 case CNID_DBD_RES_NOTFOUND:
785 case CNID_DBD_RES_ERR_DB:
793 struct _cnid_module cnid_dbd_module = {
799 #endif /* CNID_DBD */