+ /* Update digest in catalog */
+ char digestbuf[BASE64_SIZE(CRYPTO_DIGEST_MAX_SIZE)];
+ int len = 0;
+ int type = CRYPTO_DIGEST_NONE;
+
+ switch(Stream) {
+ case STREAM_MD5_DIGEST:
+ len = CRYPTO_DIGEST_MD5_SIZE;
+ type = CRYPTO_DIGEST_MD5;
+ break;
+ case STREAM_SHA1_DIGEST:
+ len = CRYPTO_DIGEST_SHA1_SIZE;
+ type = CRYPTO_DIGEST_SHA1;
+ break;
+ case STREAM_SHA256_DIGEST:
+ len = CRYPTO_DIGEST_SHA256_SIZE;
+ type = CRYPTO_DIGEST_SHA256;
+ break;
+ case STREAM_SHA512_DIGEST:
+ len = CRYPTO_DIGEST_SHA512_SIZE;
+ type = CRYPTO_DIGEST_SHA512;
+ break;
+ default:
+ /* Never reached ... */
+ Jmsg(jcr, M_ERROR, 0, _("Catalog error updating file digest. Unsupported digest stream type: %d"),
+ Stream);
+ }
+
+ bin_to_base64(digestbuf, sizeof(digestbuf), fname, len, true);
+ Dmsg3(400, "DigestLen=%d Digest=%s type=%d\n", strlen(digestbuf),
+ digestbuf, Stream);
+ if (jcr->cached_attribute) {
+ ar->Digest = digestbuf;
+ ar->DigestType = type;
+ Dmsg2(400, "Cached attr with digest. Stream=%d fname=%s\n",
+ ar->Stream, ar->fname);
+
+ /* Update BaseFile table */
+ if (!db_create_attributes_record(jcr, jcr->db, ar)) {
+ Jmsg1(jcr, M_FATAL, 0, _("attribute create error. %s"),
+ db_strerror(jcr->db));
+ }
+ jcr->cached_attribute = false;
+ } else {
+ if (!db_add_digest_to_file_record(jcr, jcr->db, ar->FileId, digestbuf, type)) {
+ Jmsg(jcr, M_ERROR, 0, _("Catalog error updating file digest. %s"),
+ db_strerror(jcr->db));
+ }
+ }
+ }
+ }
+}
+
+/*
+ * Update File Attributes in the catalog with data
+ * sent by the Storage daemon.
+ */
+void catalog_update(JCR *jcr, BSOCK *bs)
+{
+ if (!jcr->pool->catalog_files) {
+ return; /* user disabled cataloging */
+ }
+ if (jcr->is_job_canceled()) {
+ goto bail_out;
+ }
+ if (!jcr->db) {
+ POOLMEM *omsg = get_memory(bs->msglen+1);
+ pm_strcpy(omsg, bs->msg);
+ bs->fsend(_("1994 Invalid Catalog Update: %s"), omsg);
+ Jmsg1(jcr, M_FATAL, 0, _("Invalid Catalog Update; DB not open: %s"), omsg);
+ free_memory(omsg);
+ goto bail_out;
+ }
+ update_attribute(jcr, bs->msg, bs->msglen);
+
+bail_out:
+ if (jcr->is_job_canceled()) {
+ cancel_storage_daemon_job(jcr);
+ }
+}
+
+/*
+ * Update File Attributes in the catalog with data read from
+ * the storage daemon spool file. We receive the filename and
+ * we try to read it.
+ */
+bool despool_attributes_from_file(JCR *jcr, const char *file)
+{
+ bool ret=false;
+ int32_t pktsiz;
+ size_t nbytes;
+ ssize_t last = 0, size = 0;
+ int count = 0;
+ int32_t msglen; /* message length */
+ POOLMEM *msg = get_pool_memory(PM_MESSAGE);
+ FILE *spool_fd=NULL;
+
+ Dmsg0(100, "Begin despool_attributes_from_file\n");
+
+ if (jcr->is_job_canceled() || !jcr->pool->catalog_files || !jcr->db) {
+ goto bail_out; /* user disabled cataloging */
+ }
+
+ spool_fd = fopen(file, "rb");
+ if (!spool_fd) {
+ Dmsg0(100, "cancel despool_attributes_from_file\n");
+ /* send an error message */
+ goto bail_out;
+ }
+#if defined(HAVE_POSIX_FADVISE) && defined(POSIX_FADV_WILLNEED)
+ posix_fadvise(fileno(spool_fd), 0, 0, POSIX_FADV_WILLNEED);
+#endif
+
+ while (fread((char *)&pktsiz, 1, sizeof(int32_t), spool_fd) ==
+ sizeof(int32_t)) {
+ size += sizeof(int32_t);
+ msglen = ntohl(pktsiz);
+ if (msglen > 0) {
+ if (msglen > (int32_t) sizeof_pool_memory(msg)) {
+ msg = realloc_pool_memory(msg, msglen + 1);
+ }
+ nbytes = fread(msg, 1, msglen, spool_fd);
+ if (nbytes != (size_t) msglen) {
+ berrno be;
+ Dmsg2(400, "nbytes=%d msglen=%d\n", nbytes, msglen);
+ Qmsg1(jcr, M_FATAL, 0, _("fread attr spool error. ERR=%s\n"),
+ be.bstrerror());
+ goto bail_out;
+ }
+ size += nbytes;
+ if ((++count & 0x3F) == 0) {
+ last = size;
+ }