unit tests!

This commit is contained in:
leitner
2023-09-17 13:45:04 +00:00
parent fc2b16c556
commit c6c8ccba35

View File

@@ -2,7 +2,7 @@
#include <libowfat/byte.h>
#include "ldap.h"
#if 0
/*
ModifyRequest ::= [APPLICATION 6] SEQUENCE {
object LDAPDN,
modification SEQUENCE OF SEQUENCE {
@@ -15,62 +15,81 @@
AttributeTypeAndValues ::= SEQUENCE {
type AttributeDescription,
vals SET OF AttributeValue }
*/
#ifdef UNITTEST
size_t mallocfail=0;
static void* mymalloc(size_t n) {
if (--mallocfail == 0) return 0;
return malloc(n);
}
#define malloc mymalloc
#endif
size_t scan_ldapmodifyrequest(const char* src,const char* max,struct ModifyRequest* m) {
size_t res,tmp,oslen; /* outer sequence length */
struct Modification* last=0;
byte_zero(m,sizeof(*m));
if (!(res=scan_ldapstring(src,max,&m->object))) goto error;
if (!(tmp=scan_asn1SEQUENCE(src+res,max,&oslen))) goto error;
if (!(res=scan_ldapstring(src,max,&m->object))) // fail01
goto error;
if (!(tmp=scan_asn1SEQUENCE(src+res,max,&oslen))) // fail02
goto error;
res+=tmp;
if (src+res+oslen>max) goto error;
max=src+res+oslen;
if (src+res>=max) goto error; /* need at least one record */
if (src+res>=max) // fail03
goto error; /* need at least one record */
do {
size_t islen;
unsigned long etmp;
if (last) {
struct Modification* cur;
if (!(cur=malloc(sizeof(struct Modification)))) goto error;
if (!(cur=malloc(sizeof(struct Modification)))) // fail04
goto error;
byte_zero(cur,sizeof(*cur));
last->next=cur; last=cur;
} else
last=&m->m;
last->next=0;
if (!(tmp=scan_asn1SEQUENCE(src+res,max,&islen))) goto error;
if (!(tmp=scan_asn1SEQUENCE(src+res,max,&islen))) // fail05
goto error;
res+=tmp;
if (!(tmp=scan_asn1ENUMERATED(src+res,max,&etmp))) goto error;
if (etmp>2) goto error;
if (!(tmp=scan_asn1ENUMERATED(src+res,max,&etmp))) // fail06
goto error;
if (etmp>2) // fail06b
goto error;
last->operation=etmp; res+=tmp;
{
size_t iislen; /* urgh, _three_ levels of indirection */
const char* imax;
if (!(tmp=scan_asn1SEQUENCE(src+res,max,&iislen))) goto error;
if (!(tmp=scan_asn1SEQUENCE(src+res,max,&iislen))) // fail07
goto error;
res+=tmp;
imax=src+res+iislen;
if (imax>max) goto error;
if (!(tmp=scan_ldapstring(src+res,imax,&last->AttributeDescription))) goto error;
if (!(tmp=scan_ldapstring(src+res,imax,&last->AttributeDescription))) // fail08
goto error;
res+=tmp;
{
size_t iiislen; /* waah, _four_ levels of indirection! It doesn't get more inefficient than this */
const char* iimax;
struct AttributeDescriptionList** ilast=0;
if (!(tmp=scan_asn1SET(src+res,max,&iiislen))) goto error;
if (!(tmp=scan_asn1SET(src+res,max,&iiislen))) // fail09
goto error;
res+=tmp;
iimax=src+res+iiislen;
if (src+res+iiislen!=imax) goto error;
if (src+res+iiislen!=imax) // fail10
goto error;
ilast=&last->vals;
while (src+res<iimax) {
if (!(*ilast=malloc(sizeof(struct AttributeDescriptionList)))) goto error;
if (!(*ilast=malloc(sizeof(struct AttributeDescriptionList)))) // fail11
goto error;
byte_zero(*ilast,sizeof(**ilast));
if (!(tmp=scan_ldapstring(src+res,imax,&(*ilast)->a))) goto error;
if (!(tmp=scan_ldapstring(src+res,imax,&(*ilast)->a))) // fail12
goto error;
ilast=&(*ilast)->next;
res+=tmp;
}
}
}
break;
} while (src+res<max);
return res;
error:
@@ -81,6 +100,7 @@ error:
static void free_mod(struct Modification* m) {
while (m) {
struct Modification* tmp=m->next;
free_ldapadl(m->vals);
free(m);
m=tmp;
}
@@ -90,3 +110,81 @@ void free_ldapmodifyrequest(struct ModifyRequest* m) {
free_ldapadl(m->m.vals);
free_mod(m->m.next);
}
#ifdef UNITTEST
#undef UNITTEST
#include <assert.h>
#include <string.h>
#include "scan_ldapstring.c"
#include "scan_asn1STRING.c"
#include "scan_asn1tag.c"
#include "scan_asn1tagint.c"
#include "scan_asn1length.c"
#include "scan_asn1string.c"
#include "scan_asn1SEQUENCE.c"
#include "scan_asn1SET.c"
#include "scan_asn1ENUMERATED.c"
#include "scan_asn1int.c"
#include "scan_asn1rawint.c"
#include "free_ldapadl.c"
void byte_zero(void* out,size_t n) {
memset(out,0,n);
}
#include <stdio.h>
int main() {
char buf[500];
memcpy(buf, "\x04\x21ts=1234567890,ou=blog,d=fefe,c=de" // 0 string (33)
"\x30\x16" // 35 sequence (22)
"\x30\x14" // 37 sequence (20)
"\x0a\x01\x02" // 39 enumerated (1)
"\x30\x0f" // 42 sequence (15)
"\x04\x05""danke" // 44 string (5)
"\x31\x06" // 51 set (6)
"\x04\x04""Fefe", // 53 string (4)
59);
struct ModifyRequest m = { 0 };
assert(scan_ldapmodifyrequest(buf,buf+500,&m) == 59);
assert(m.object.l==33 && m.object.s==buf+2);
assert(m.m.operation==Replace && m.m.AttributeDescription.l==5 && m.m.AttributeDescription.s==buf+46);
assert(m.m.vals && m.m.vals->a.l==4 && m.m.vals->a.s==buf+55 && m.m.vals->next==0);
free_ldapmodifyrequest(&m);
memcpy(buf+53,"\x04\x01X\x04\x01Y", 6);
assert(scan_ldapmodifyrequest(buf,buf+500,&m) == 59);
assert(m.m.vals && m.m.vals->a.l==1 && m.m.vals->a.s==buf+55);
assert(m.m.vals->next && m.m.vals->next->a.l==1 && m.m.vals->next->a.s==buf+58 && m.m.vals->next->next==0);
free_ldapmodifyrequest(&m);
memcpy(buf+59,
"\x30\x14" // 59 sequence (20)
"\x0a\x01\x02" // 61 enumerated (1)
"\x30\x0f" // 64 sequence (15)
"\x04\x05""danke" // 66 string (5)
"\x31\x06" // 73 set (6)
"\x04\x04""Fefe", // 75 string (4)
22);
buf[36]+=22; // adjust length of outer sequence
assert(scan_ldapmodifyrequest(buf,buf+500,&m) == 81);
free_ldapmodifyrequest(&m);
mallocfail=2; assert(scan_ldapmodifyrequest(buf,buf+500,&m) == 0); // fail11
mallocfail=3; assert(scan_ldapmodifyrequest(buf,buf+500,&m) == 0); // fail04
buf[75]=0; assert(scan_ldapmodifyrequest(buf,buf+500,&m) == 0); // fail12
buf[36]++; buf[65]++; assert(scan_ldapmodifyrequest(buf,buf+500,&m) == 0); // fail10
buf[73]=0; assert(scan_ldapmodifyrequest(buf,buf+500,&m) == 0); // fail09
buf[66]=0; assert(scan_ldapmodifyrequest(buf,buf+500,&m) == 0); // fail08
buf[64]=0; assert(scan_ldapmodifyrequest(buf,buf+500,&m) == 0); // fail07
buf[62]=3; assert(scan_ldapmodifyrequest(buf,buf+500,&m) == 0); // fail06b
buf[61]=0; assert(scan_ldapmodifyrequest(buf,buf+500,&m) == 0); // fail06
buf[59]=0; assert(scan_ldapmodifyrequest(buf,buf+500,&m) == 0); // fail05
memcpy(buf, "\x04\x21ts=1234567890,ou=blog,d=fefe,c=de" // 0 string (33)
"\x30\x00", // 35 sequence (0)
37);
assert(scan_ldapmodifyrequest(buf,buf+500,&m) == 0); // fail03
buf[35]=0; assert(scan_ldapmodifyrequest(buf,buf+500,&m) == 0); // fail02
buf[0]=0; assert(scan_ldapmodifyrequest(buf,buf+500,&m) == 0); // fail01
return 0;
}
#endif