Add ability to drop overly large packets

[onak.git] / lookup.c

diff --git a/lookup.c b/lookup.c
index 533433a3c1d66d9fee2e6221e095e5f1b61944ab..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>
@@ -47,8 +46,9 @@
 #define OP_HGET    5
 
 void find_keys(struct onak_dbctx *dbctx,
-               char *search, uint64_t keyid, uint8_t *fp, size_t fpsize,
-               bool ishex, bool isfp, bool fingerprint, bool skshash,
+               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;
@@ -58,7 +58,7 @@ void find_keys(struct onak_dbctx *dbctx,
                count = dbctx->fetch_key_id(dbctx, keyid, &publickey,
                                false);
        } else if (isfp) {
-               count = dbctx->fetch_key_fp(dbctx, fp, fpsize, &publickey,
+               count = dbctx->fetch_key_fp(dbctx, fingerprint, &publickey,
                                false);
        } else {
                count = dbctx->fetch_key_text(dbctx, search, &publickey);
@@ -68,7 +68,7 @@ void find_keys(struct onak_dbctx *dbctx,
                        printf("info:1:%d\n", count);
                        mrkey_index(publickey);
                } else {
-                       key_index(dbctx, publickey, verbose, fingerprint,
+                       key_index(dbctx, publickey, verbose, dispfp,
                                skshash, true);
                }
                free_publickey(publickey);
@@ -108,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;
@@ -146,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) {
@@ -164,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")) {
@@ -212,7 +214,7 @@ int main(int argc, char *argv[])
                readconfig(NULL);
                initlogthing("lookup", config.logfile);
                catchsignals();
-               dbctx = config.dbinit(false);
+               dbctx = config.dbinit(config.backend, false);
                switch (op) {
                case OP_GET:
                case OP_HGET:
@@ -224,8 +226,8 @@ int main(int argc, char *argv[])
                                result = dbctx->fetch_key_id(dbctx, keyid,
                                        &publickey, false);
                        } else if (isfp) {
-                               result = dbctx->fetch_key_fp(dbctx, fp,
-                                       MAX_FINGERPRINT_LEN, &publickey, false);
+                               result = dbctx->fetch_key_fp(dbctx,
+                                       &fingerprint, &publickey, false);
                        } else {
                                result = dbctx->fetch_key_text(dbctx,
                                        search,
@@ -237,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);
@@ -253,19 +255,18 @@ int main(int argc, char *argv[])
                        }
                        break;
                case OP_INDEX:
-                       find_keys(dbctx, 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(dbctx, 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) {
-                               dbctx->fetch_key_fp(dbctx, fp,
-                                       MAX_FINGERPRINT_LEN,
+                               dbctx->fetch_key_fp(dbctx, &fingerprint,
                                        &publickey, false);
                        } else {
                                dbctx->fetch_key_id(dbctx, keyid,