X-Git-Url: http://the.earth.li/gitweb/?a=blobdiff_plain;f=parsekey.c;h=ed61e24d16be5e556a527cfbfa80bc4f10e6beaa;hb=cd281cb130964e168d59f763a8888f56608048bd;hp=517b0294a1452d26f23717052df85e64697365a3;hpb=5e1b22d763640c4d7a09d07920403d8d491b4410;p=onak.git

diff --git a/parsekey.c b/parsekey.c
index 517b029..ed61e24 100644
--- a/parsekey.c
+++ b/parsekey.c
@@ -25,8 +25,8 @@
 #include "keyid.h"
 #include "keystructs.h"
 #include "ll.h"
-#include "log.h"
 #include "mem.h"
+#include "onak.h"
 #include "openpgp.h"
 #include "parsekey.h"
 
@@ -62,7 +62,8 @@ int parse_keys(struct openpgp_packet_list *packets,
 			 * It's a signature packet. Add it to either the public
 			 * key, to the current UID or the current subkey.
 			 */
-			log_assert(curkey != NULL);
+			if (curkey == NULL)
+				return ONAK_E_INVALID_PARAM;
 			if (curkey->subkeys != NULL) {
 				ADD_PACKET_TO_LIST_END(curkey->last_subkey,
 					sig,
@@ -117,8 +118,10 @@ int parse_keys(struct openpgp_packet_list *packets,
 			/*
 			 * It's a UID packet (or a photo id, which is similar).
 			 */
-			log_assert(curkey != NULL);
-			log_assert(curkey->subkeys == NULL);
+			if (curkey == NULL)
+				return ONAK_E_INVALID_PARAM;
+			if (curkey->subkeys != NULL)
+				return ONAK_E_INVALID_PARAM;
 			ADD_PACKET_TO_LIST_END(curkey,
 				uid,
 				packet_dup(packets->packet));
@@ -127,13 +130,14 @@ int parse_keys(struct openpgp_packet_list *packets,
 			/*
 			 * It's a subkey packet.
 			 */
-			log_assert(curkey != NULL);
+			if (curkey == NULL)
+				return ONAK_E_INVALID_PARAM;
 			ADD_PACKET_TO_LIST_END(curkey,
 				subkey,
 				packet_dup(packets->packet));
 			break;
 		case OPENPGP_PACKET_TRUST:
-		case 61:
+		case OPENPGP_PACKET_COMMENT:
 			/*
 			 * One of:
 			 *
@@ -142,9 +146,8 @@ int parse_keys(struct openpgp_packet_list *packets,
 			 */
 			break;
 		default:
-			logthing(LOGTHING_ERROR,
-					"Unsupported packet type: %d",
-					packets->packet->tag);
+			/* Unsupported packet. Do what? Ignore for now. */
+			break;
 		}
 		packets = packets->next;
 	}
@@ -181,19 +184,20 @@ int debug_packet(struct openpgp_packet *packet)
  *	packet stream and reads the packets into a linked list of packets
  *	ready for parsing as a public key or whatever.
  */
-int read_openpgp_stream(int (*getchar_func)(void *ctx, size_t count,
+onak_status_t read_openpgp_stream(int (*getchar_func)(void *ctx, size_t count,
 				void *c),
 				void *ctx,
 				struct openpgp_packet_list **packets,
 				int maxnum)
 {
 	unsigned char			 curchar = 0;
-	struct openpgp_packet_list	*curpacket = NULL;
-	int				 rc = 0;
+	struct openpgp_packet_list	*curpacket = NULL, **packetend = NULL;
+	onak_status_t			 rc = ONAK_E_OK;
 	int				 keys = 0;
-	bool				 inpacket = false;
 
-	log_assert(packets != NULL);
+	if (packets == NULL)
+		return ONAK_E_INVALID_PARAM;
+
 	curpacket = *packets;
 	if (curpacket != NULL) {
 		while (curpacket->next != NULL) {
@@ -201,20 +205,20 @@ int read_openpgp_stream(int (*getchar_func)(void *ctx, size_t count,
 		}
 	}
 
-	while (!rc && (maxnum == 0 || keys < maxnum) &&
+	while (rc == ONAK_E_OK && (maxnum == 0 || keys < maxnum) &&
 			!getchar_func(ctx, 1, &curchar)) {
-		if (!inpacket && (curchar & 0x80)) {
+		if (curchar & 0x80) {
 			/*
-			 * New packet. Record the fact we're in a packet and
-			 * allocate memory for it.
+			 * New packet. Allocate memory for it.
 			 */
-			inpacket = true;
 			if (curpacket != NULL) {
 				curpacket->next = malloc(sizeof (*curpacket));
+				packetend = &curpacket->next;
 				curpacket = curpacket->next;
 			} else {
 				*packets = curpacket =
 					malloc(sizeof (*curpacket));
+				packetend = packets;
 			}
 			memset(curpacket, 0, sizeof(*curpacket));
 			curpacket->packet =
@@ -229,7 +233,10 @@ int read_openpgp_stream(int (*getchar_func)(void *ctx, size_t count,
 			 */
 			if (curpacket->packet->newformat) {
 				curpacket->packet->tag = (curchar & 0x3F);
-				rc = getchar_func(ctx, 1, &curchar);
+				if (getchar_func(ctx, 1, &curchar)) {
+					rc = ONAK_E_INVALID_PKT;
+					break;
+				}
 				curpacket->packet->length = curchar;
 				if (curpacket->packet->length > 191 &&
 					curpacket->packet->length < 224) {
@@ -240,59 +247,90 @@ int read_openpgp_stream(int (*getchar_func)(void *ctx, size_t count,
 					curpacket->packet->length += 192;
 				} else if (curpacket->packet->length > 223 &&
 					curpacket->packet->length < 255) {
-					logthing(LOGTHING_NOTICE,
-						"Partial length;"
-						" not supported.");
+					free(curpacket->packet);
+					curpacket->packet = NULL;
+					rc = ONAK_E_UNSUPPORTED_FEATURE;
 				} else if (curpacket->packet->length == 255) {
 					/*
 					 * 5 byte length; ie 255 followed by 3
 					 * bytes of MSB length.
 					 */
-					rc = getchar_func(ctx, 1, &curchar);
+					if (getchar_func(ctx, 1, &curchar)) {
+						rc = ONAK_E_INVALID_PKT;
+						break;
+					}
 					curpacket->packet->length = curchar;
 					curpacket->packet->length <<= 8;
-					rc = getchar_func(ctx, 1, &curchar);
+					if (getchar_func(ctx, 1, &curchar)) {
+						rc = ONAK_E_INVALID_PKT;
+						break;
+					}
 					curpacket->packet->length += curchar;
 					curpacket->packet->length <<= 8;
-					rc = getchar_func(ctx, 1, &curchar);
+					if (getchar_func(ctx, 1, &curchar)) {
+						rc = ONAK_E_INVALID_PKT;
+						break;
+					}
 					curpacket->packet->length += curchar;
 					curpacket->packet->length <<= 8;
-					rc = getchar_func(ctx, 1, &curchar);
+					if (getchar_func(ctx, 1, &curchar)) {
+						rc = ONAK_E_INVALID_PKT;
+						break;
+					}
 					curpacket->packet->length += curchar;
 				}
 			} else {
 				curpacket->packet->tag = (curchar & 0x3C) >> 2;
 				switch (curchar & 3) {
 				case 0:
-					rc = getchar_func(ctx, 1, &curchar);
+					if (getchar_func(ctx, 1, &curchar)) {
+						rc = ONAK_E_INVALID_PKT;
+						break;
+					}
 					curpacket->packet->length = curchar;
 					break;
 				case 1:
-					rc = getchar_func(ctx, 1, &curchar);
+					if (getchar_func(ctx, 1, &curchar)) {
+						rc = ONAK_E_INVALID_PKT;
+						break;
+					}
 					curpacket->packet->length = curchar;
 					curpacket->packet->length <<= 8;
-					rc = getchar_func(ctx, 1, &curchar);
+					if (getchar_func(ctx, 1, &curchar)) {
+						rc = ONAK_E_INVALID_PKT;
+						break;
+					}
 					curpacket->packet->length += curchar;
 					break;
 				case 2:
-					rc = getchar_func(ctx, 1, &curchar);
+					if (getchar_func(ctx, 1, &curchar)) {
+						rc = ONAK_E_INVALID_PKT;
+						break;
+					}
 					curpacket->packet->length = 
 						(curchar << 24);
-					rc = getchar_func(ctx, 1, &curchar);
+					if (getchar_func(ctx, 1, &curchar)) {
+						rc = ONAK_E_INVALID_PKT;
+						break;
+					}
 					curpacket->packet->length +=
 						(curchar << 16);
-					rc = getchar_func(ctx, 1, &curchar);
+					if (getchar_func(ctx, 1, &curchar)) {
+						rc = ONAK_E_INVALID_PKT;
+						break;
+					}
 					curpacket->packet->length +=
 						(curchar << 8);
-					rc = getchar_func(ctx, 1, &curchar);
+					if (getchar_func(ctx, 1, &curchar)) {
+						rc = ONAK_E_INVALID_PKT;
+						break;
+					}
 					curpacket->packet->length += curchar;
 					break;
 				case 3:
-					logthing(LOGTHING_ERROR,
-						"Unsupported length type 3.");
-					curpacket->packet->length = 0;
-					curpacket->packet->data = NULL;
-					rc = -1;
+					rc = ONAK_E_UNSUPPORTED_FEATURE;
+					free(curpacket->packet);
+					curpacket->packet = NULL;
 					break;
 				}
 			}
@@ -306,24 +344,24 @@ int read_openpgp_stream(int (*getchar_func)(void *ctx, size_t count,
 					malloc(curpacket->packet->length *
 					sizeof(unsigned char));
 				if (curpacket->packet->data == NULL) {
-					logthing(LOGTHING_ERROR, 
-						"Can't allocate memory for "
-						"packet!");
-					rc = -1;
+					rc = ONAK_E_NOMEM;
 				} else {
 					rc = getchar_func(ctx,
 						curpacket->packet->length,
 						curpacket->packet->data);
 				}
 			}
-			inpacket = false;
 		} else {
-			logthing(LOGTHING_ERROR, "Unexpected character: 0x%X",
-				curchar);
-			rc = 1;
+			rc = ONAK_E_INVALID_PKT;
 		}
 	}
 
+	/* Trim the last packet if it doesn't actually exist */
+	if (packetend != NULL && (*packetend)->packet == NULL) {
+		free(*packetend);
+		*packetend = NULL;
+	}
+
 	return (rc);
 }
 
@@ -336,7 +374,7 @@ int read_openpgp_stream(int (*getchar_func)(void *ctx, size_t count,
  *	This function uses putchar_func to write characters to an OpenPGP
  *	packet stream from a linked list of packets.
  */
-int write_openpgp_stream(int (*putchar_func)(void *ctx, size_t count,
+onak_status_t write_openpgp_stream(int (*putchar_func)(void *ctx, size_t count,
 						void *c),
 				void *ctx,
 				struct openpgp_packet_list *packets)
@@ -364,8 +402,6 @@ int write_openpgp_stream(int (*putchar_func)(void *ctx, size_t count,
 				putchar_func(ctx, 1, &curchar);
 			} else if (packets->packet->length > 8382 &&
 				packets->packet->length < 0xFFFFFFFF) {
-				logthing(LOGTHING_DEBUG,
-					"Writing 5 byte length");
 				curchar = 255;
 				putchar_func(ctx, 1, &curchar);
 				
@@ -385,8 +421,7 @@ int write_openpgp_stream(int (*putchar_func)(void *ctx, size_t count,
 				curchar &= 0xFF;
 				putchar_func(ctx, 1, &curchar);
 			} else {
-				logthing(LOGTHING_ERROR,
-					"Unsupported new format length.");
+				return ONAK_E_UNSUPPORTED_FEATURE;
 			}
 		} else {
 			curchar |= (packets->packet->tag << 2);
@@ -419,7 +454,8 @@ int write_openpgp_stream(int (*putchar_func)(void *ctx, size_t count,
 				packets->packet->data);
 		packets = packets->next;
 	}
-	return 0;
+
+	return ONAK_E_OK;
 }
 
 /**