]> git.cryptolib.org Git - avr-crypto-lib.git/blob - scal/scal-basic.c
bug fixing and support for malloc instead of stack memory (some functions)
[avr-crypto-lib.git] / scal / scal-basic.c
1 /* scal-basic.c */
2 /*
3     This file is part of the AVR-Crypto-Lib.
4     Copyright (C) 2011  Daniel Otte (daniel.otte@rub.de)
5
6     This program is free software: you can redistribute it and/or modify
7     it under the terms of the GNU General Public License as published by
8     the Free Software Foundation, either version 3 of the License, or
9     (at your option) any later version.
10
11     This program is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14     GNU General Public License for more details.
15
16     You should have received a copy of the GNU General Public License
17     along with this program.  If not, see <http://www.gnu.org/licenses/>.
18 */
19
20 #include <stdlib.h>
21 #include <stdint.h>
22 #include <string.h>
23 #include <avr/pgmspace.h>
24 #include "streamcipher_descriptor.h"
25 #include "keysize_descriptor.h"
26
27 uint8_t scal_cipher_init(const scdesc_t *cipher_descriptor,
28                          const void *key, uint16_t keysize_b,
29                          const void *iv,  uint16_t ivsize_b, scgen_ctx_t *ctx){
30         ctx->buffer = NULL;
31         ctx->ctx = NULL;
32
33         if(!is_valid_keysize_P((PGM_VOID_P)pgm_read_word(&(cipher_descriptor->valid_keysize_desc)), keysize_b)){
34                 return 1;
35         }
36         if(!is_valid_keysize_P((PGM_VOID_P)pgm_read_word(&(cipher_descriptor->valid_ivsize_desc)), ivsize_b)){
37                 return 2;
38         }
39         uint8_t flags;
40         void_fpt init_fpt;
41         flags = pgm_read_byte(&(cipher_descriptor->flags));
42         ctx->desc_ptr = cipher_descriptor;
43         ctx->keysize = keysize_b;
44         ctx->ivsize  = ivsize_b;
45         ctx->ctx = malloc(pgm_read_word(&(cipher_descriptor->ctxsize_B)));
46         if(ctx->ctx==NULL){
47                 return 3;
48         }
49         init_fpt = (void_fpt)pgm_read_word(&(cipher_descriptor->init));
50         switch(flags&SC_INIT_TYPE){
51                 case SC_INIT_TYPE_1:
52                         ((sc_init1_fpt)init_fpt)(key, ctx->ctx);
53                         break;
54                 case SC_INIT_TYPE_2:
55                         ((sc_init2_fpt)init_fpt)(key, iv, ctx->ctx);
56                         break;
57                 case SC_INIT_TYPE_3:
58                         ((sc_init3_fpt)init_fpt)(key, keysize_b, ctx->ctx);
59                         break;
60                 case SC_INIT_TYPE_4:
61                         ((sc_init4_fpt)init_fpt)(key, keysize_b, iv, ctx->ctx);
62                         break;
63                 case SC_INIT_TYPE_5:
64                         ((sc_init5_fpt)init_fpt)(key, keysize_b, iv, ivsize_b, ctx->ctx);
65                         break;
66                 default:
67                         return 4;
68         }
69         uint16_t blocksize_b;
70         blocksize_b = pgm_read_word(&(cipher_descriptor->gensize_b));
71         if(blocksize_b>8){
72                 ctx->buffer=malloc((blocksize_b+7)/8);
73                 if(ctx->buffer==NULL){
74                         return 5;
75                 }
76                 ctx->index=0;
77         }
78         return 0;
79 }
80
81
82 void scal_cipher_free(scgen_ctx_t *ctx){
83         if(ctx->buffer){
84                 free(ctx->buffer);
85         }
86         if(ctx->ctx){
87                 free(ctx->ctx);
88         }
89 }
90
91 uint8_t scal_cipher_gen_byte(scgen_ctx_t *ctx){
92         uint8_t flags;
93         uint16_t blocksize_b;
94         void_fpt gen_fpt;
95         flags = pgm_read_byte(&(ctx->desc_ptr->flags));
96         blocksize_b = pgm_read_word(&(ctx->desc_ptr->gensize_b));
97         gen_fpt = (void_fpt)(pgm_read_word(&(ctx->desc_ptr->gen.genvoid)));
98
99         if(blocksize_b==8){
100                 if((flags&SC_GEN_TYPE)==SC_GEN_TYPE_1){
101                         return ((sc_gen1_fpt)gen_fpt)(ctx->ctx);
102                 }else{
103                         uint8_t r;
104                         ((sc_gen2_fpt)gen_fpt)(&r, ctx->ctx);
105                         return r;
106                 }
107         }
108         if(blocksize_b<8){
109                 uint8_t r=0;
110                 uint8_t fill=0;
111                 do{
112                         r |= ((((sc_gen1_fpt)gen_fpt)(ctx->ctx))&(0xff<<(8-blocksize_b)))>>fill;
113                         fill += blocksize_b;
114                 }while(fill<8);
115                 return r;
116         }else{
117                 uint8_t r;
118                 if(ctx->index==0){
119                         ((sc_gen2_fpt)gen_fpt)(ctx->buffer, ctx->ctx);
120                         ctx->index = blocksize_b;
121                 }
122                 r=ctx->buffer[(blocksize_b-ctx->index)/8];
123                 ctx->index -= 8;
124                 return r;
125         }
126 }
127
128 void scal_cipher_gen_block(void *block, scgen_ctx_t *ctx){
129         uint8_t flags;
130         void_fpt gen_fpt;
131         flags = pgm_read_byte(&(ctx->desc_ptr->flags));
132         gen_fpt = (void_fpt)pgm_read_word(&(ctx->desc_ptr->gen));
133         if((flags&SC_GEN_TYPE)==SC_GEN_TYPE_1){
134                 *((uint8_t*)block) = ((sc_gen1_fpt)gen_fpt)(ctx->ctx);
135         }else{
136                 ((sc_gen2_fpt)gen_fpt)(block, ctx->ctx);
137         }
138 }
139
140 void scal_cipher_gen_fillblock(void *block, uint16_t blocksize_B, scgen_ctx_t *ctx){
141         while(blocksize_B){
142                 *((uint8_t*)block) = scal_cipher_gen_byte(ctx);
143                 block = (uint8_t*)block + 1;
144                 blocksize_B -= 1;
145         }
146 }
147
148 uint16_t scal_cipher_getBlocksize_b(const scdesc_t *desc){
149         uint16_t blocksize_b;
150         blocksize_b = pgm_read_word(&(desc->gensize_b));
151         return blocksize_b;
152 }
153
154 PGM_VOID_P scal_cipher_getKeysizeDesc(const scdesc_t *desc){
155         return (PGM_VOID_P)pgm_read_word(&(desc->valid_keysize_desc));
156 }
157
158 PGM_VOID_P scal_cipher_getIVsizeDesc(const scdesc_t *desc){
159         return (PGM_VOID_P)pgm_read_word(&(desc->valid_ivsize_desc));
160 }
161
162