X-Git-Url: https://git.cryptolib.org/?p=avr-crypto-lib.git;a=blobdiff_plain;f=keysize_descriptor.c;h=253100ff63f4e82b35e749d99ddc383f12271ed5;hp=045ff65f4c24c2c5188fad812ebbf24886f94095;hb=e542ff92d053ecf40b42364a44bc887431cecae2;hpb=6483e60140be5fe4ab60de85cdf679fc0518dc99

diff --git a/keysize_descriptor.c b/keysize_descriptor.c
index 045ff65..253100f 100644
--- a/keysize_descriptor.c
+++ b/keysize_descriptor.c
@@ -25,6 +25,7 @@
  */
 
 #include <stdint.h>
+#include <stdlib.h>
 #include <avr/pgmspace.h>
 #include "keysize_descriptor.h"
 
@@ -76,11 +77,85 @@ uint16_t get_keysize(PGM_VOID_P ks_desc){
 	uint8_t type;
 	uint16_t keysize;
 	type = pgm_read_byte(ks_desc);
-	if(type==KS_TYPE_LIST)
+	if(type==KS_TYPE_LIST){
 		ks_desc = (uint8_t*)ks_desc + 1;
+	}
 	ks_desc = (uint8_t*)ks_desc + 1;
 	keysize = pgm_read_word(ks_desc);
 	return keysize;
 }
 
-
+uint16_t get_keysizes(PGM_VOID_P ks_desc, uint16_t** list){
+	uint8_t type;
+	uint16_t items;
+	uint8_t i;
+	type = pgm_read_byte(ks_desc);
+	ks_desc = (uint8_t*)ks_desc + 1;
+	if(type==KS_TYPE_LIST){
+		items = pgm_read_byte(ks_desc);
+		ks_desc = (uint8_t*)ks_desc + 1;
+		if(!*list){
+			*list = malloc(items*2);
+			if(!*list){
+				return 0;
+			}
+		}
+		for(i=0; i<items; ++i){
+			((uint16_t*)(*list))[i] = pgm_read_word(ks_desc);
+			ks_desc = (uint8_t*)ks_desc + 2;
+		}
+		return items;
+	}
+	if(type==KS_TYPE_ARG_RANGE){
+		uint16_t min, max, distance, offset;
+		min = pgm_read_word(ks_desc);
+		ks_desc = (uint8_t*)ks_desc + 2;
+		max = pgm_read_word(ks_desc);
+		ks_desc = (uint8_t*)ks_desc + 2;
+		distance = pgm_read_word(ks_desc);
+		ks_desc = (uint8_t*)ks_desc + 2;
+		offset = pgm_read_word(ks_desc);
+		ks_desc = (uint8_t*)ks_desc + 2;
+		items = (max-min)/distance+1;
+		if(min%distance!=offset){
+			--items;
+			min += (distance-(min%distance-offset))%distance;
+		}
+		if(!*list){
+			*list = malloc(items*2);
+			if(!*list){
+				return 0;
+			}
+		}
+		i=0;
+		while(min<=max){
+			((uint16_t*)*list)[i++] = min;
+			min += distance;
+		}
+		return i;
+	}
+	if(type==KS_TYPE_RANGE){
+		uint16_t min, max, distance=8, offset=0;
+		min = pgm_read_word(ks_desc);
+		ks_desc = (uint8_t*)ks_desc + 2;
+		max = pgm_read_word(ks_desc);
+		items = (max-min)/distance+1;
+		if(min%distance!=offset){
+			--items;
+			min += (distance-(min%distance-offset))%distance;
+		}
+		if(!*list){
+			*list = malloc(items*2);
+			if(!*list){
+				return 0;
+			}
+		}
+		i=0;
+		while(min<=max){
+			((uint16_t*)*list)[i++] = min;
+			min += distance;
+		}
+		return i;
+	}
+	return 0;
+}