Add ability to drop overly large packets

[onak.git] / keyid.c

diff --git a/keyid.c b/keyid.c
index 6a9b5580c34fa1c6a9df2100f565fd117ce382dc..d5244943f8e0c4480294900ad02c8577dc0b1fa2 100644 (file)
--- a/keyid.c
+++ b/keyid.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 <string.h>
@@ -38,6 +37,19 @@
 #include "sha1.h"
 #endif
 
+uint64_t fingerprint2keyid(struct openpgp_fingerprint *fingerprint)
+{
+       uint64_t keyid;
+       int i;
+
+       for (keyid = 0, i = 12; i < 20; i++) {
+               keyid <<= 8;
+               keyid += fingerprint->fp[i];
+       }
+
+       return keyid;
+}
+
 
 /**
  *     get_keyid - Given a public key returns the keyid.
@@ -132,7 +144,7 @@ onak_status_t get_packetid(struct openpgp_packet *packet, uint64_t *keyid)
        uint8_t         data;
 #endif
 
-       if (packet == NULL)
+       if (packet == NULL || packet->data == NULL)
                return ONAK_E_INVALID_PARAM;
 
        switch (packet->data[0]) {
@@ -159,11 +171,9 @@ onak_status_t get_packetid(struct openpgp_packet *packet, uint64_t *keyid)
                        ripemd160_digest(&ripemd160_context,
                                RIPEMD160_DIGEST_SIZE,
                                fingerprint.fp);
+                       fingerprint.length = RIPEMD160_DIGEST_SIZE;
 
-                       for (*keyid = 0, i = 12; i < 20; i++) {
-                               *keyid <<= 8;
-                               *keyid += fingerprint.fp[i];
-                       }
+                       *keyid = fingerprint2keyid(&fingerprint);
 
                        return ONAK_E_OK;
                }
@@ -194,11 +204,8 @@ onak_status_t get_packetid(struct openpgp_packet *packet, uint64_t *keyid)
                break;
        case 4:
                get_fingerprint(packet, &fingerprint);
-               
-               for (*keyid = 0, i = 12; i < 20; i++) {
-                       *keyid <<= 8;
-                       *keyid += fingerprint.fp[i];
-               }
+
+               *keyid = fingerprint2keyid(&fingerprint);
 
                break;
        default: