Add ability to drop overly large packets

[onak.git] / lookup.c

diff --git a/lookup.c b/lookup.c
index 13ffbcc9e11848557f454c96de6009673411d058..c9f94a9e8579c51c187659cecb08f36a742a86a3 100644 (file)
--- a/lookup.c
+++ b/lookup.c
@@ -13,8 +13,7 @@
  * more details.
  *
  * You should have received a copy of the GNU General Public License along with
- * this program; if not, write to the Free Software Foundation, Inc., 51
- * Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ * this program.  If not, see <https://www.gnu.org/licenses/>.
  */
 
 #include <inttypes.h>
@@ -46,29 +45,31 @@
 #define OP_PHOTO   4
 #define OP_HGET    5
 
-void find_keys(char *search, uint64_t keyid, uint8_t *fp, size_t fpsize,
-               bool ishex, bool isfp, bool fingerprint, bool skshash,
+void find_keys(struct onak_dbctx *dbctx,
+               char *search, uint64_t keyid,
+               struct openpgp_fingerprint *fingerprint,
+               bool ishex, bool isfp, bool dispfp, bool skshash,
                bool exact, bool verbose, bool mrhkp)
 {
        struct openpgp_publickey *publickey = NULL;
        int count = 0;
 
        if (ishex) {
-               count = config.dbbackend->fetch_key_id(keyid, &publickey,
+               count = dbctx->fetch_key_id(dbctx, keyid, &publickey,
                                false);
        } else if (isfp) {
-               count = config.dbbackend->fetch_key_fp(fp, fpsize, &publickey,
+               count = dbctx->fetch_key_fp(dbctx, fingerprint, &publickey,
                                false);
        } else {
-               count = config.dbbackend->fetch_key_text(search, &publickey);
+               count = dbctx->fetch_key_text(dbctx, search, &publickey);
        }
        if (publickey != NULL) {
                if (mrhkp) {
                        printf("info:1:%d\n", count);
                        mrkey_index(publickey);
                } else {
-                       key_index(publickey, verbose, fingerprint, skshash,
-                               true);
+                       key_index(dbctx, publickey, verbose, dispfp,
+                               skshash, true);
                }
                free_publickey(publickey);
        } else if (count == 0) {
@@ -107,16 +108,16 @@ int main(int argc, char *argv[])
 {
        char **params = NULL;
        int op = OP_UNKNOWN;
-       int i;
+       int i, j;
        int indx = 0;
-       bool fingerprint = false;
+       bool dispfp = false;
        bool skshash = false;
        bool exact = false;
        bool ishex = false;
        bool isfp = false;
        bool mrhkp = false;
        uint64_t keyid = 0;
-       uint8_t fp[MAX_FINGERPRINT_LEN];
+       struct openpgp_fingerprint fingerprint;
        char *search = NULL;
        char *end = NULL;
        struct openpgp_publickey *publickey = NULL;
@@ -124,6 +125,7 @@ int main(int argc, char *argv[])
        struct openpgp_packet_list *list_end = NULL;
        int result;
        struct skshash hash;
+       struct onak_dbctx *dbctx;
 
        params = getcgivars(argc, argv);
        for (i = 0; params != NULL && params[i] != NULL; i += 2) {
@@ -144,10 +146,12 @@ int main(int argc, char *argv[])
                        params[i+1] = NULL;
                        if (search != NULL && strlen(search) == 42 &&
                                        search[0] == '0' && search[1] == 'x') {
-                               for (i = 0; i < MAX_FINGERPRINT_LEN; i++) {
-                                       fp[i] = (hex2bin(search[2 + i * 2])
+                               fingerprint.length = MAX_FINGERPRINT_LEN;
+                               for (j = 0; j < MAX_FINGERPRINT_LEN; j++) {
+                                       fingerprint.fp[j] = (hex2bin(
+                                                       search[2 + j * 2])
                                                                << 4) +
-                                               hex2bin(search[3 + i * 2]);
+                                               hex2bin(search[3 + j * 2]);
                                }
                                isfp = true;
                        } else if (search != NULL) {
@@ -162,7 +166,7 @@ int main(int argc, char *argv[])
                        indx = atoi(params[i+1]);
                } else if (!strcmp(params[i], "fingerprint")) {
                        if (!strcmp(params[i+1], "on")) {
-                               fingerprint = true;
+                               dispfp = true;
                        }
                } else if (!strcmp(params[i], "hash")) {
                        if (!strcmp(params[i+1], "on")) {
@@ -210,22 +214,22 @@ int main(int argc, char *argv[])
                readconfig(NULL);
                initlogthing("lookup", config.logfile);
                catchsignals();
-               config.dbbackend->initdb(false);
+               dbctx = config.dbinit(config.backend, false);
                switch (op) {
                case OP_GET:
                case OP_HGET:
                        if (op == OP_HGET) {
                                parse_skshash(search, &hash);
-                               result = config.dbbackend->fetch_key_skshash(
+                               result = dbctx->fetch_key_skshash(dbctx,
                                        &hash, &publickey);
                        } else if (ishex) {
-                               result = config.dbbackend->fetch_key_id(keyid,
+                               result = dbctx->fetch_key_id(dbctx, keyid,
                                        &publickey, false);
                        } else if (isfp) {
-                               result = config.dbbackend->fetch_key_fp(fp,
-                                       MAX_FINGERPRINT_LEN, &publickey, false);
+                               result = dbctx->fetch_key_fp(dbctx,
+                                       &fingerprint, &publickey, false);
                        } else {
-                               result = config.dbbackend->fetch_key_text(
+                               result = dbctx->fetch_key_text(dbctx,
                                        search,
                                        &publickey);
                        }
@@ -235,7 +239,7 @@ int main(int argc, char *argv[])
                                        result,
                                        search);
                                puts("<pre>");
-                               cleankeys(publickey);
+                               cleankeys(&publickey, config.clean_policies);
                                flatten_publickey(publickey,
                                                        &packets,
                                                        &list_end);
@@ -251,22 +255,21 @@ int main(int argc, char *argv[])
                        }
                        break;
                case OP_INDEX:
-                       find_keys(search, keyid, fp, MAX_FINGERPRINT_LEN,
-                                       ishex, isfp, fingerprint, skshash,
+                       find_keys(dbctx, search, keyid, &fingerprint,
+                                       ishex, isfp, dispfp, skshash,
                                        exact, false, mrhkp);
                        break;
                case OP_VINDEX:
-                       find_keys(search, keyid, fp, MAX_FINGERPRINT_LEN,
-                                       ishex, isfp, fingerprint, skshash,
+                       find_keys(dbctx, search, keyid, &fingerprint,
+                                       ishex, isfp, dispfp, skshash,
                                        exact, true, mrhkp);
                        break;
                case OP_PHOTO:
                        if (isfp) {
-                               config.dbbackend->fetch_key_fp(fp,
-                                       MAX_FINGERPRINT_LEN,
+                               dbctx->fetch_key_fp(dbctx, &fingerprint,
                                        &publickey, false);
                        } else {
-                               config.dbbackend->fetch_key_id(keyid,
+                               dbctx->fetch_key_id(dbctx, keyid,
                                        &publickey, false);
                        }
                        if (publickey != NULL) {
@@ -287,7 +290,7 @@ int main(int argc, char *argv[])
                default:
                        puts("Unknown operation!");
                }
-               config.dbbackend->cleanupdb();
+               dbctx->cleanupdb(dbctx);
                cleanuplogthing();
                cleanupconfig();
        }