Changeset 221 for branches/samba-3.3.x/source/registry/reg_backend_db.c

Timestamp:

May 24, 2009, 7:17:10 AM (17 years ago)

Author:

Herwig Bauernfeind

Message:

Update Samba 3.3 to 3.3.1

File:

• branches/samba-3.3.x/source/registry/reg_backend_db.c (modified) (3 diffs)

branches/samba-3.3.x/source/registry/reg_backend_db.c

@@ -537 +537 @@
 
         for (i=0; i<num_subkeys; i++) {
-                len += tdb_pack(buffer+len, buflen-len, "f",
-                                regsubkey_ctr_specific_key(ctr, i));
-                if (len > buflen) {
-                        /* allocate some extra space */
-                        buffer = (uint8 *)SMB_REALLOC(buffer, len*2);
+                size_t thistime;
+
+                thistime = tdb_pack(buffer+len, buflen-len, "f",
+                                    regsubkey_ctr_specific_key(ctr, i));
+                if (len+thistime > buflen) {
+                        size_t thistime2;
+                        /*
+                         * tdb_pack hasn't done anything because of the short
+                         * buffer, allocate extra space.
+                         */
+                        buffer = SMB_REALLOC_ARRAY(buffer, uint8_t,
+                                                   (len+thistime)*2);
                         if(buffer == NULL) {
                                 DEBUG(0, ("regdb_store_keys: Failed to realloc "
-                                          "memory of size [%d]\n", len*2));
+                                          "memory of size [%u]\n",
+                                          (unsigned int)(len+thistime)*2));
                                 ret = false;
                                 goto done;
                         }
-                        buflen = len*2;
-                        len = tdb_pack(buffer+len, buflen-len, "f",
-                                       regsubkey_ctr_specific_key(ctr, i));
-                }
+                        buflen = (len+thistime)*2;
+                        thistime2 = tdb_pack(
+                                buffer+len, buflen-len, "f",
+                                regsubkey_ctr_specific_key(ctr, i));
+                        if (thistime2 != thistime) {
+                                DEBUG(0, ("tdb_pack failed\n"));
+                                ret = false;
+                                goto done;
+                        }
+                }
+                len += thistime;
         }
 
@@ -928 +943 @@
 int regdb_fetch_keys(const char *key, REGSUBKEY_CTR *ctr)
 {
-        WERROR werr;
         uint32 num_items;
         uint8 *buf;
@@ -959 +973 @@
         len = tdb_unpack( buf, buflen, "d", &num_items);
 
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
         for (i=0; i<num_items; i++) {
                 len += tdb_unpack(buf+len, buflen-len, "f", subkeyname);
-                werr = regsubkey_ctr_addkey(ctr, subkeyname);
-                if (!W_ERROR_IS_OK(werr)) {
-                        DEBUG(5, ("regdb_fetch_keys: regsubkey_ctr_addkey "
-                                  "failed: %s\n", dos_errstr(werr)));
+                ctr->subkeys[i] = talloc_strdup(ctr->subkeys, subkeyname);
+                if (ctr->subkeys[i] == NULL) {
+                        DEBUG(5, ("regdb_fetch_keys: could not allocate "
+                                  "subkeyname\n"));
+                        TALLOC_FREE(ctr->subkeys);
+                        ctr->num_subkeys = 0;
                         goto done;
                 }