mirror of
https://github.com/cathugger/mkp224o.git
synced 2025-01-09 19:17:19 -03:00
split worker off
This commit is contained in:
cathugger
parent
bfd9d712c8
commit
5c58f03805
14 changed files with 1499 additions and 1338 deletions
39
Makefile.in
39
Makefile.in
|
|
@ -22,6 +22,7 @@ ED25519OBJ= $(ED25519_@ED25519IMPL@)
|
||||||
|
|
||||||
MAINOBJ= \
|
MAINOBJ= \
|
||||||
main.c.o \
|
main.c.o \
|
||||||
|
worker.c.o \
|
||||||
yaml.c.o \
|
yaml.c.o \
|
||||||
vec.c.o \
|
vec.c.o \
|
||||||
cpucount.c.o \
|
cpucount.c.o \
|
||||||
|
|
@ -392,23 +393,8 @@ ed25519/ref10/sign.c.o: ed25519/ref10/ge.h ed25519/ref10/fe.h
|
||||||
ed25519/ref10/sign.c.o: ed25519/ref10/crypto_int32.h ed25519/ref10/sc.h
|
ed25519/ref10/sign.c.o: ed25519/ref10/crypto_int32.h ed25519/ref10/sc.h
|
||||||
ioutil.c.o: types.h ioutil.h
|
ioutil.c.o: types.h ioutil.h
|
||||||
keccak.c.o: types.h keccak.h
|
keccak.c.o: types.h keccak.h
|
||||||
main.c.o: types.h likely.h vec.h base32.h cpucount.h keccak.h
|
main.c.o: types.h vec.h cpucount.h keccak.h ioutil.h common.h yaml.h
|
||||||
main.c.o: ed25519/ed25519.h ed25519/ref10/ed25519.h ed25519/ref10/ge.h
|
main.c.o: filters.h worker.h filters_main.inc.h filters_common.inc.h
|
||||||
main.c.o: ed25519/ref10/fe.h ed25519/ref10/crypto_int32.h
|
|
||||||
main.c.o: ed25519/amd64-51-30k/ed25519.h ed25519/amd64-51-30k/ge25519.h
|
|
||||||
main.c.o: ed25519/amd64-51-30k/fe25519.h ed25519/amd64-51-30k/sc25519.h
|
|
||||||
main.c.o: ed25519/amd64-64-24k/ed25519.h ed25519/amd64-64-24k/ge25519.h
|
|
||||||
main.c.o: ed25519/ed25519-donna/ed25519-donna.h
|
|
||||||
main.c.o: ed25519/ed25519-donna/ed25519-donna-portable.h
|
|
||||||
main.c.o: ed25519/ed25519-donna/ed25519-donna-portable-identify.h
|
|
||||||
main.c.o: ed25519/ed25519-donna/curve25519-donna-64bit.h
|
|
||||||
main.c.o: ed25519/ed25519-donna/curve25519-donna-helpers.h
|
|
||||||
main.c.o: ed25519/ed25519-donna/modm-donna-64bit.h
|
|
||||||
main.c.o: ed25519/ed25519-donna/ed25519-donna-basepoint-table.h
|
|
||||||
main.c.o: ed25519/ed25519-donna/ed25519-donna-64bit-tables.h
|
|
||||||
main.c.o: ed25519/ed25519-donna/ed25519-donna-64bit-x86.h
|
|
||||||
main.c.o: ed25519/ed25519-donna/ed25519-donna-impl-base.h ioutil.h common.h
|
|
||||||
main.c.o: yaml.h filters.h
|
|
||||||
test_base16.c.o: types.h base16.h
|
test_base16.c.o: types.h base16.h
|
||||||
test_base32.c.o: types.h base32.h
|
test_base32.c.o: types.h base32.h
|
||||||
test_base64.c.o: types.h base64.h
|
test_base64.c.o: types.h base64.h
|
||||||
|
|
@ -431,4 +417,23 @@ test_ed25519.c.o: ed25519/ed25519-donna/ed25519-donna-64bit-tables.h
|
||||||
test_ed25519.c.o: ed25519/ed25519-donna/ed25519-donna-64bit-x86.h
|
test_ed25519.c.o: ed25519/ed25519-donna/ed25519-donna-64bit-x86.h
|
||||||
test_ed25519.c.o: ed25519/ed25519-donna/ed25519-donna-impl-base.h
|
test_ed25519.c.o: ed25519/ed25519-donna/ed25519-donna-impl-base.h
|
||||||
vec.c.o: vec.h
|
vec.c.o: vec.h
|
||||||
|
worker.c.o: types.h likely.h vec.h base32.h keccak.h ed25519/ed25519.h
|
||||||
|
worker.c.o: ed25519/ref10/ed25519.h ed25519/ref10/ge.h ed25519/ref10/fe.h
|
||||||
|
worker.c.o: ed25519/ref10/crypto_int32.h ed25519/amd64-51-30k/ed25519.h
|
||||||
|
worker.c.o: ed25519/amd64-51-30k/ge25519.h ed25519/amd64-51-30k/fe25519.h
|
||||||
|
worker.c.o: ed25519/amd64-51-30k/sc25519.h ed25519/amd64-64-24k/ed25519.h
|
||||||
|
worker.c.o: ed25519/amd64-64-24k/ge25519.h
|
||||||
|
worker.c.o: ed25519/ed25519-donna/ed25519-donna.h
|
||||||
|
worker.c.o: ed25519/ed25519-donna/ed25519-donna-portable.h
|
||||||
|
worker.c.o: ed25519/ed25519-donna/ed25519-donna-portable-identify.h
|
||||||
|
worker.c.o: ed25519/ed25519-donna/curve25519-donna-64bit.h
|
||||||
|
worker.c.o: ed25519/ed25519-donna/curve25519-donna-helpers.h
|
||||||
|
worker.c.o: ed25519/ed25519-donna/modm-donna-64bit.h
|
||||||
|
worker.c.o: ed25519/ed25519-donna/ed25519-donna-basepoint-table.h
|
||||||
|
worker.c.o: ed25519/ed25519-donna/ed25519-donna-64bit-tables.h
|
||||||
|
worker.c.o: ed25519/ed25519-donna/ed25519-donna-64bit-x86.h
|
||||||
|
worker.c.o: ed25519/ed25519-donna/ed25519-donna-impl-base.h ioutil.h common.h
|
||||||
|
worker.c.o: yaml.h worker.h filters.h filters_worker.inc.h
|
||||||
|
worker.c.o: filters_common.inc.h worker_slow.inc.h worker_fast.inc.h
|
||||||
|
worker.c.o: worker_fast_pass.inc.h worker_batch.inc.h
|
||||||
yaml.c.o: types.h yaml.h ioutil.h base32.h base64.h common.h
|
yaml.c.o: types.h yaml.h ioutil.h base32.h base64.h common.h
|
||||||
|
|
|
||||||
|
|
@ -36,5 +36,6 @@ You should have received a copy of the CC0 Public Domain Dedication along with t
|
||||||
keccak.c is based on <https://github.com/gvanas/KeccakCodePackage/blob/master/Standalone/CompactFIPS202/Keccak-more-compact.c>.
|
keccak.c is based on <https://github.com/gvanas/KeccakCodePackage/blob/master/Standalone/CompactFIPS202/Keccak-more-compact.c>.
|
||||||
ed25519/{ref10,amd64-51-30k,amd64-64-24k} are adopted from SUPERCOP <https://bench.cr.yp.to/supercop.html>.
|
ed25519/{ref10,amd64-51-30k,amd64-64-24k} are adopted from SUPERCOP <https://bench.cr.yp.to/supercop.html>.
|
||||||
ed25519/ed25519-donna adopted from <https://github.com/floodyberry/ed25519-donna>.
|
ed25519/ed25519-donna adopted from <https://github.com/floodyberry/ed25519-donna>.
|
||||||
Idea used in main.c' dofastwork() is stolen from <https://github.com/Yawning/horse25519>.
|
Idea used in worker_fast() is stolen from <https://github.com/Yawning/horse25519>.
|
||||||
base64 routines and initial YAML processing work contributed by Alexander Khristoforov <heios@protonmail.com>.
|
base64 routines and initial YAML processing work contributed by Alexander Khristoforov <heios@protonmail.com>.
|
||||||
|
Passphrase-based generation code and idea used in worker_batch() contributed by <https://github.com/foobar2019>.
|
||||||
|
|
|
||||||
2
base64.h
2
base64.h
|
|
@ -6,7 +6,7 @@ char *base64_to(char *dst,const u8 *src,size_t slen);
|
||||||
size_t base64_from(u8 *dst,const char *src,size_t slen);
|
size_t base64_from(u8 *dst,const char *src,size_t slen);
|
||||||
// calculates length needed to store data converted from base64
|
// calculates length needed to store data converted from base64
|
||||||
#define BASE64_FROM_LEN(l) ((l) / 4 * 3)
|
#define BASE64_FROM_LEN(l) ((l) / 4 * 3)
|
||||||
// validates base32 string and optionally stores length of valid data
|
// validates base64 string and optionally stores length of valid data
|
||||||
// returns 1 if whole string is good, 0 if string contains invalid data
|
// returns 1 if whole string is good, 0 if string contains invalid data
|
||||||
int base64_valid(const char *src,size_t *count);
|
int base64_valid(const char *src,size_t *count);
|
||||||
// aligns data length to something base64 can represent without padding
|
// aligns data length to something base64 can represent without padding
|
||||||
|
|
|
||||||
736
filters.h
736
filters.h
|
|
@ -1,4 +1,3 @@
|
||||||
// filters stuff
|
|
||||||
|
|
||||||
#ifndef INTFILTER
|
#ifndef INTFILTER
|
||||||
# define BINFILTER
|
# define BINFILTER
|
||||||
|
|
@ -42,26 +41,37 @@ struct binfilter {
|
||||||
u8 mask;
|
u8 mask;
|
||||||
} ;
|
} ;
|
||||||
|
|
||||||
|
VEC_STRUCT(bfiltervec,struct binfilter);
|
||||||
|
|
||||||
|
#ifdef BINFILTER
|
||||||
|
extern struct bfiltervec filters;
|
||||||
|
#endif
|
||||||
|
|
||||||
#endif // NEEDBINFILTER
|
#endif // NEEDBINFILTER
|
||||||
|
|
||||||
|
|
||||||
#ifdef BINFILTER
|
|
||||||
static VEC_STRUCT(bfiltervec,struct binfilter) filters;
|
|
||||||
#endif // BINFILTER
|
|
||||||
|
|
||||||
#ifdef INTFILTER
|
#ifdef INTFILTER
|
||||||
|
|
||||||
struct intfilter {
|
struct intfilter {
|
||||||
IFT f;
|
IFT f;
|
||||||
# ifndef OMITMASK
|
# ifndef OMITMASK
|
||||||
IFT m;
|
IFT m;
|
||||||
# endif
|
# endif
|
||||||
} ;
|
} ;
|
||||||
static VEC_STRUCT(ifiltervec,struct intfilter) filters;
|
|
||||||
|
VEC_STRUCT(ifiltervec,struct intfilter);
|
||||||
|
|
||||||
|
extern struct ifiltervec filters;
|
||||||
|
|
||||||
# ifdef OMITMASK
|
# ifdef OMITMASK
|
||||||
IFT ifiltermask;
|
extern IFT ifiltermask;
|
||||||
# endif
|
# endif
|
||||||
|
|
||||||
#endif // INTFILTER
|
#endif // INTFILTER
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifdef PCRE2FILTER
|
#ifdef PCRE2FILTER
|
||||||
|
|
||||||
#define PCRE2_CODE_UNIT_WIDTH 8
|
#define PCRE2_CODE_UNIT_WIDTH 8
|
||||||
|
|
@ -71,719 +81,15 @@ struct pcre2filter {
|
||||||
char *str;
|
char *str;
|
||||||
pcre2_code *re;
|
pcre2_code *re;
|
||||||
} ;
|
} ;
|
||||||
static VEC_STRUCT(pfiltervec,struct pcre2filter) filters;
|
|
||||||
|
|
||||||
#endif // PCRE2FILTER
|
VEC_STRUCT(pfiltervec,struct pcre2filter);
|
||||||
|
|
||||||
static void filters_init()
|
extern struct pfiltervec filters;
|
||||||
{
|
|
||||||
VEC_INIT(filters);
|
|
||||||
}
|
|
||||||
|
|
||||||
#ifdef INTFILTER
|
|
||||||
|
|
||||||
static inline size_t filter_len(size_t i)
|
|
||||||
{
|
|
||||||
# ifndef OMITMASK
|
|
||||||
const u8 *m = (const u8 *)&VEC_BUF(filters,i).m;
|
|
||||||
# else // OMITMASK
|
|
||||||
const u8 *m = (const u8 *)&ifiltermask;
|
|
||||||
# endif // OMITMASK
|
|
||||||
size_t c = 0;
|
|
||||||
for (size_t j = 0;;) {
|
|
||||||
u8 v = m[j];
|
|
||||||
for (size_t k = 0;;) {
|
|
||||||
if (!v)
|
|
||||||
return c;
|
|
||||||
++c;
|
|
||||||
if (++k >= 8)
|
|
||||||
break;
|
|
||||||
v <<= 1;
|
|
||||||
}
|
|
||||||
if (++j >= sizeof(IFT))
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
return c;
|
|
||||||
}
|
|
||||||
|
|
||||||
# ifdef OMITMASK
|
|
||||||
|
|
||||||
static inline int filter_compare(const void *p1,const void *p2)
|
|
||||||
{
|
|
||||||
if (((const struct intfilter *)p1)->f < ((const struct intfilter *)p2)->f)
|
|
||||||
return -1;
|
|
||||||
if (((const struct intfilter *)p1)->f > ((const struct intfilter *)p2)->f)
|
|
||||||
return 1;
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
# ifdef EXPANDMASK
|
|
||||||
|
|
||||||
/*
|
|
||||||
* for mask expansion, we need to figure out how much bits
|
|
||||||
* we need to fill in with different values.
|
|
||||||
* while in big endian machines this is quite easy,
|
|
||||||
* representation we use for little endian ones may
|
|
||||||
* leave gap of bits we don't want to touch.
|
|
||||||
*
|
|
||||||
* initial idea draft:
|
|
||||||
*
|
|
||||||
* raw representation -- FF.FF.F0.00
|
|
||||||
* big endian -- 0xFFFFF000
|
|
||||||
* little endian -- 0x00F0FFFF
|
|
||||||
* b: 0xFFffF000 ^ 0xFFff0000 -> 0x0000F000
|
|
||||||
* 0x0000F000 + 1 -> 0x0000F001
|
|
||||||
* 0x0000F000 & 0x0000F001 -> 0x0000F000 <- shifted mask
|
|
||||||
* 0x0000F000 ^ 0x0000F000 -> 0x00000000 <- direct mask
|
|
||||||
* 0x0000F000 ^ 0x00000000 -> 0x0000F000 <- shifted mask
|
|
||||||
* l: 0x00f0FFff ^ 0x0000FFff -> 0x00f00000
|
|
||||||
* 0x00f00000 + 1 -> 0x00f00001
|
|
||||||
* 0x00f00000 & 0x00f00001 -> 0x00f00000 <- shifted mask
|
|
||||||
* 0x00f00000 ^ 0x00f00000 -> 0x00000000 <- direct mask
|
|
||||||
* 0x00f00000 ^ 0x00000000 -> 0x00f00000 <- shifted mask
|
|
||||||
*
|
|
||||||
* b: 0xFFffFFff ^ 0xF0000000 -> 0x0FffFFff
|
|
||||||
* 0x0FffFFff + 1 -> 0x10000000
|
|
||||||
* 0x0FffFFff & 0x10000000 -> 0x00000000 <- shifted mask
|
|
||||||
* 0x0FffFFff ^ 0x00000000 -> 0x0FffFFff <- direct mask
|
|
||||||
* 0x0FffFFff ^ 0x0FffFFff -> 0x00000000 <- shifted mask
|
|
||||||
* l: 0xFFffFFff ^ 0x000000f0 -> 0xFFffFF0f
|
|
||||||
* 0xFFffFF0f + 1 -> 0xFFffFF10
|
|
||||||
* 0xFFffFF0f & 0xFFffFF10 -> 0xFFffFF00 <- shifted mask
|
|
||||||
* 0xFFffFF0f ^ 0xFFffFF00 -> 0x0000000f <- direct mask
|
|
||||||
* 0xFFffFF0f ^ 0x0000000f -> 0xFFffFF00 <- shifted mask
|
|
||||||
*
|
|
||||||
* essentially, we have to make direct mask + shifted mask bits worth of information
|
|
||||||
* and then split it into 2 parts
|
|
||||||
* we do not need absolute shifted mask shifting value, just relative to direct mask
|
|
||||||
* 0x0sss00dd - shifted & direct mask combo
|
|
||||||
* 0x000sssdd - combined mask
|
|
||||||
* 8 - relshiftval
|
|
||||||
* generate values from 0x00000000 to 0x000sssdd
|
|
||||||
* for each value, realmask <- (val & 0x000000dd) | ((val & 0x000sss00) << relshiftval)
|
|
||||||
* or..
|
|
||||||
* realmask <- (val & 0x000000dd) | ((val << relshiftval) & 0x0sss0000)
|
|
||||||
* ...
|
|
||||||
*
|
|
||||||
* above method doesn't work in some cases. better way:
|
|
||||||
*
|
|
||||||
* l: 0x80ffFFff ^ 0x00f0FFff -> 0x800f0000
|
|
||||||
* 0x800f0000 >> 16 -> 0x0000800f
|
|
||||||
* 0x0000800f + 1 -> 0x00008010
|
|
||||||
* 0x0000800f & 0x00008010 -> 0x00008000 <- smask
|
|
||||||
* 0x0000800f ^ 0x00008000 -> 0x0000000f <- dmask
|
|
||||||
*
|
|
||||||
* cross <- difference between mask we desire and mask we currently have
|
|
||||||
* shift cross to left variable ammount of times to eliminate zeros
|
|
||||||
* save shift ammount as ishift (initial shift)
|
|
||||||
* then, we eliminate first area of ones; if there was no gap, result is already all zeros
|
|
||||||
* save this thing as smask. it's only higher bits.
|
|
||||||
* XOR smask and cross; result is only lower bits.
|
|
||||||
* shift smask to left variable ammount of times until gap is eliminated.
|
|
||||||
* save resulting mask as cmask;
|
|
||||||
* save resulting shift value as rshift.
|
|
||||||
*/
|
|
||||||
|
|
||||||
static int flattened = 0;
|
|
||||||
|
|
||||||
#define EXPVAL(init,j,dmask,smask,ishift,rshift) \
|
|
||||||
((init) | ((((j) & (dmask)) | (((j) << (rshift)) & (smask))) << (ishift)))
|
|
||||||
// add expanded set of values
|
|
||||||
// allocates space on its own
|
|
||||||
static void ifilter_addexpanded(
|
|
||||||
struct intfilter *ifltr,
|
|
||||||
IFT dmask,IFT smask,IFT cmask,
|
|
||||||
int ishift,int rshift)
|
|
||||||
{
|
|
||||||
flattened = 1;
|
|
||||||
size_t i = VEC_LENGTH(filters);
|
|
||||||
VEC_ADDN(filters,cmask + 1);
|
|
||||||
for (size_t j = 0;;++j) {
|
|
||||||
VEC_BUF(filters,i + j).f =
|
|
||||||
EXPVAL(ifltr->f,j,dmask,smask,ishift,rshift);
|
|
||||||
if (j == cmask)
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// expand existing stuff
|
|
||||||
// allocates needed stuff on its own
|
|
||||||
static void ifilter_expand(IFT dmask,IFT smask,IFT cmask,int ishift,int rshift)
|
|
||||||
{
|
|
||||||
flattened = 1;
|
|
||||||
size_t len = VEC_LENGTH(filters);
|
|
||||||
VEC_ADDN(filters,cmask * len);
|
|
||||||
size_t esz = cmask + 1; // size of expanded elements
|
|
||||||
for (size_t i = len - 1;;--i) {
|
|
||||||
for (IFT j = 0;;++j) {
|
|
||||||
VEC_BUF(filters,i * esz + j).f =
|
|
||||||
EXPVAL(VEC_BUF(filters,i).f,j,dmask,smask,ishift,rshift);
|
|
||||||
if (j == cmask)
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
if (i == 0)
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static inline void ifilter_addflatten(struct intfilter *ifltr,IFT mask)
|
|
||||||
{
|
|
||||||
if (VEC_LENGTH(filters) == 0) {
|
|
||||||
// simple
|
|
||||||
VEC_ADD(filters,*ifltr);
|
|
||||||
ifiltermask = mask;
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (ifiltermask == mask) {
|
|
||||||
// lucky
|
|
||||||
VEC_ADD(filters,*ifltr);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
IFT cross = ifiltermask ^ mask;
|
|
||||||
int ishift = 0;
|
|
||||||
while ((cross & 1) == 0) {
|
|
||||||
++ishift;
|
|
||||||
cross >>= 1;
|
|
||||||
}
|
|
||||||
IFT smask = cross & (cross + 1); // shift mask
|
|
||||||
IFT dmask = cross ^ smask; // direct mask
|
|
||||||
IFT cmask; // combined mask
|
|
||||||
int rshift = 0; // relative shift
|
|
||||||
while (cmask = (smask >> rshift) | dmask,(cmask & (cmask + 1)) != 0)
|
|
||||||
++rshift;
|
|
||||||
// preparations done
|
|
||||||
if (ifiltermask > mask) {
|
|
||||||
// already existing stuff has more precise mask than we
|
|
||||||
// so we need to expand our stuff
|
|
||||||
ifilter_addexpanded(ifltr,dmask,smask,cmask,ishift,rshift);
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
ifiltermask = mask;
|
|
||||||
ifilter_expand(dmask,smask,cmask,ishift,rshift);
|
|
||||||
VEC_ADD(filters,*ifltr);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
# endif // EXPANDMASK
|
|
||||||
|
|
||||||
# else // OMITMASK
|
|
||||||
|
|
||||||
/*
|
|
||||||
* struct intfilter layout: filter,mask
|
|
||||||
* stuff is compared in big-endian way, so memcmp
|
|
||||||
* filter needs to be compared first
|
|
||||||
* if its equal, mask needs to be compared
|
|
||||||
* memcmp is aplicable there too
|
|
||||||
* due to struct intfilter layout, it all can be stuffed into one memcmp call
|
|
||||||
*/
|
|
||||||
static inline int filter_compare(const void *p1,const void *p2)
|
|
||||||
{
|
|
||||||
return memcmp(p1,p2,sizeof(struct intfilter));
|
|
||||||
}
|
|
||||||
|
|
||||||
# endif // OMITMASK
|
|
||||||
|
|
||||||
static void filter_sort(void)
|
|
||||||
{
|
|
||||||
size_t len = VEC_LENGTH(filters);
|
|
||||||
if (len > 0)
|
|
||||||
qsort(&VEC_BUF(filters,0),len,sizeof(struct intfilter),&filter_compare);
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif // INTFILTER
|
|
||||||
|
|
||||||
#ifdef BINFILTER
|
|
||||||
|
|
||||||
static inline size_t filter_len(size_t i)
|
|
||||||
{
|
|
||||||
size_t c = VEC_BUF(filters,i).len * 8;
|
|
||||||
u8 v = VEC_BUF(filters,i).mask;
|
|
||||||
for (size_t k = 0;;) {
|
|
||||||
if (!v)
|
|
||||||
return c;
|
|
||||||
++c;
|
|
||||||
if (++k >= 8)
|
|
||||||
return c;
|
|
||||||
v <<= 1;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static inline int filter_compare(const void *p1,const void *p2)
|
|
||||||
{
|
|
||||||
const struct binfilter *b1 = (const struct binfilter *)p1;
|
|
||||||
const struct binfilter *b2 = (const struct binfilter *)p2;
|
|
||||||
|
|
||||||
size_t l = b1->len <= b2->len ? b1->len : b2->len;
|
|
||||||
|
|
||||||
int cmp = memcmp(b1->f,b2->f,l);
|
|
||||||
if (cmp != 0)
|
|
||||||
return cmp;
|
|
||||||
|
|
||||||
if (b1->len < b2->len)
|
|
||||||
return -1;
|
|
||||||
if (b1->len > b2->len)
|
|
||||||
return +1;
|
|
||||||
|
|
||||||
u8 cmask = b1->mask & b2->mask;
|
|
||||||
if ((b1->f[l] & cmask) < (b2->f[l] & cmask))
|
|
||||||
return -1;
|
|
||||||
if ((b1->f[l] & cmask) > (b2->f[l] & cmask))
|
|
||||||
return +1;
|
|
||||||
|
|
||||||
if (b1->mask < b2->mask)
|
|
||||||
return -1;
|
|
||||||
if (b1->mask > b2->mask)
|
|
||||||
return +1;
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void filter_sort(void)
|
|
||||||
{
|
|
||||||
size_t len = VEC_LENGTH(filters);
|
|
||||||
if (len > 0)
|
|
||||||
qsort(&VEC_BUF(filters,0),len,sizeof(struct binfilter),&filter_compare);
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif // BINFILTER
|
|
||||||
|
|
||||||
#ifdef PCRE2FILTER
|
|
||||||
|
|
||||||
#define filter_len(i) ((pcre2ovector[1] - pcre2ovector[0]) * 5)
|
|
||||||
|
|
||||||
#endif // PCRE2FILTER
|
|
||||||
|
|
||||||
static void filters_add(const char *filter)
|
|
||||||
{
|
|
||||||
#ifdef NEEDBINFILTER
|
|
||||||
struct binfilter bf;
|
|
||||||
size_t ret;
|
|
||||||
# ifdef INTFILTER
|
|
||||||
union intconv {
|
|
||||||
IFT i;
|
|
||||||
u8 b[sizeof(IFT)];
|
|
||||||
} fc,mc;
|
|
||||||
# endif
|
|
||||||
|
|
||||||
// skip regex start symbol. we do not support regex tho
|
|
||||||
if (*filter == '^')
|
|
||||||
++filter;
|
|
||||||
|
|
||||||
memset(&bf,0,sizeof(bf));
|
|
||||||
|
|
||||||
if (!base32_valid(filter,&ret)) {
|
|
||||||
fprintf(stderr,"filter \"%s\" is invalid\n",filter);
|
|
||||||
fprintf(stderr," ");
|
|
||||||
while (ret--)
|
|
||||||
fputc(' ',stderr);
|
|
||||||
fprintf(stderr,"^\n");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
ret = BASE32_FROM_LEN(ret);
|
|
||||||
if (!ret)
|
|
||||||
return;
|
|
||||||
# ifdef INTFILTER
|
|
||||||
size_t maxsz = sizeof(IFT);
|
|
||||||
# else
|
|
||||||
size_t maxsz = sizeof(bf.f);
|
|
||||||
# endif
|
|
||||||
if (ret > maxsz) {
|
|
||||||
fprintf(stderr,"filter \"%s\" is too long\n",filter);
|
|
||||||
fprintf(stderr," ");
|
|
||||||
maxsz = (maxsz * 8) / 5;
|
|
||||||
while (maxsz--)
|
|
||||||
fputc(' ',stderr);
|
|
||||||
fprintf(stderr,"^\n");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
base32_from(bf.f,&bf.mask,filter);
|
|
||||||
bf.len = ret - 1;
|
|
||||||
|
|
||||||
# ifdef INTFILTER
|
|
||||||
mc.i = 0;
|
|
||||||
for (size_t i = 0;i < bf.len;++i)
|
|
||||||
mc.b[i] = 0xFF;
|
|
||||||
mc.b[bf.len] = bf.mask;
|
|
||||||
memcpy(fc.b,bf.f,sizeof(fc.b));
|
|
||||||
fc.i &= mc.i;
|
|
||||||
|
|
||||||
struct intfilter ifltr = {
|
|
||||||
.f = fc.i,
|
|
||||||
# ifndef OMITMASK
|
|
||||||
.m = mc.i,
|
|
||||||
# endif
|
|
||||||
};
|
|
||||||
|
|
||||||
# ifdef OMITMASK
|
|
||||||
ifilter_addflatten(&ifltr,mc.i);
|
|
||||||
# else // OMITMASK
|
|
||||||
VEC_ADD(filters,ifltr);
|
|
||||||
# endif // OMITMASK
|
|
||||||
# endif // INTFILTER
|
|
||||||
|
|
||||||
# ifdef BINFILTER
|
|
||||||
VEC_ADD(filters,bf);
|
|
||||||
# endif // BINFILTER
|
|
||||||
#endif // NEEDBINFILTER
|
|
||||||
|
|
||||||
#ifdef PCRE2FILTER
|
|
||||||
int errornum;
|
|
||||||
PCRE2_SIZE erroroffset;
|
|
||||||
pcre2_code *re;
|
|
||||||
|
|
||||||
re = pcre2_compile((PCRE2_SPTR8)filter,PCRE2_ZERO_TERMINATED,
|
|
||||||
PCRE2_NO_UTF_CHECK | PCRE2_ANCHORED,&errornum,&erroroffset,0);
|
|
||||||
if (!re) {
|
|
||||||
PCRE2_UCHAR buffer[1024];
|
|
||||||
pcre2_get_error_message(errornum,buffer,sizeof(buffer));
|
|
||||||
fprintf(stderr,"PCRE2 compilation failed at offset " FSZ ": %s\n",
|
|
||||||
(size_t)erroroffset,buffer);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// attempt to JIT. ignore error
|
|
||||||
(void) pcre2_jit_compile(re,PCRE2_JIT_COMPLETE);
|
|
||||||
|
|
||||||
struct pcre2filter f;
|
|
||||||
memset(&f,0,sizeof(f));
|
|
||||||
f.re = re;
|
|
||||||
size_t fl = strlen(filter) + 1;
|
|
||||||
f.str = (char *) malloc(fl);
|
|
||||||
if (!f.str)
|
|
||||||
abort();
|
|
||||||
memcpy(f.str,filter,fl);
|
|
||||||
VEC_ADD(filters,f);
|
|
||||||
#endif // PCRE2FILTER
|
|
||||||
}
|
|
||||||
|
|
||||||
#ifndef PCRE2FILTER
|
|
||||||
static inline int filters_a_includes_b(size_t a,size_t b)
|
|
||||||
{
|
|
||||||
# ifdef INTFILTER
|
|
||||||
# ifdef OMITMASK
|
|
||||||
return VEC_BUF(filters,a).f == VEC_BUF(filters,b).f;
|
|
||||||
# else // OMITMASK
|
|
||||||
return VEC_BUF(filters,a).f == (VEC_BUF(filters,b).f & VEC_BUF(filters,a).m);
|
|
||||||
# endif // OMITMASK
|
|
||||||
# else // INTFILTER
|
|
||||||
const struct binfilter *fa = &VEC_BUF(filters,a);
|
|
||||||
const struct binfilter *fb = &VEC_BUF(filters,b);
|
|
||||||
|
|
||||||
if (fa->len > fb->len)
|
|
||||||
return 0;
|
|
||||||
size_t l = fa->len;
|
|
||||||
|
|
||||||
int cmp = memcmp(fa->f,fb->f,l);
|
|
||||||
if (cmp != 0)
|
|
||||||
return 0;
|
|
||||||
|
|
||||||
if (fa->len < fb->len)
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
if (fa->mask > fb->mask)
|
|
||||||
return 0;
|
|
||||||
|
|
||||||
return fa->f[l] == (fb->f[l] & fa->mask);
|
|
||||||
# endif // INTFILTER
|
|
||||||
}
|
|
||||||
|
|
||||||
static void filters_dedup(void)
|
|
||||||
{
|
|
||||||
size_t last = ~(size_t)0; // index after last matching element
|
|
||||||
size_t chk; // element to compare against
|
|
||||||
size_t st; // start of area to destroy
|
|
||||||
|
|
||||||
size_t len = VEC_LENGTH(filters);
|
|
||||||
for (size_t i = 1;i < len;++i) {
|
|
||||||
if (last != i) {
|
|
||||||
if (filters_a_includes_b(i - 1,i)) {
|
|
||||||
if (last != ~(size_t)0) {
|
|
||||||
memmove(&VEC_BUF(filters,st),
|
|
||||||
&VEC_BUF(filters,last),
|
|
||||||
(i - last) * VEC_ELSIZE(filters));
|
|
||||||
st += i - last;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
st = i;
|
|
||||||
chk = i - 1;
|
|
||||||
last = i + 1;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
if (filters_a_includes_b(chk,i))
|
|
||||||
last = i + 1;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (last != ~(size_t)0) {
|
|
||||||
memmove(&VEC_BUF(filters,st),
|
|
||||||
&VEC_BUF(filters,last),
|
|
||||||
(len - last) * VEC_ELSIZE(filters));
|
|
||||||
st += len - last;
|
|
||||||
VEC_SETLENGTH(filters,st);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
#endif // !PCRE2FILTER
|
|
||||||
|
|
||||||
static void filters_prepare(void)
|
|
||||||
{
|
|
||||||
#ifndef PCRE2FILTER
|
|
||||||
if (!quietflag)
|
|
||||||
fprintf(stderr,"sorting filters...");
|
|
||||||
filter_sort();
|
|
||||||
if (wantdedup) {
|
|
||||||
if (!quietflag)
|
|
||||||
fprintf(stderr," removing duplicates...");
|
|
||||||
filters_dedup();
|
|
||||||
}
|
|
||||||
if (!quietflag)
|
|
||||||
fprintf(stderr," done.\n");
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
static void filters_clean(void)
|
|
||||||
{
|
|
||||||
#ifdef PCRE2FILTER
|
|
||||||
for (size_t i = 0;i < VEC_LENGTH(filters);++i) {
|
|
||||||
pcre2_code_free(VEC_BUF(filters,i).re);
|
|
||||||
free(VEC_BUF(filters,i).str);
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
VEC_FREE(filters);
|
|
||||||
}
|
|
||||||
|
|
||||||
static size_t filters_count(void)
|
|
||||||
{
|
|
||||||
return VEC_LENGTH(filters);
|
|
||||||
}
|
|
||||||
|
|
||||||
#ifdef INTFILTER
|
|
||||||
|
|
||||||
# ifndef BINSEARCH
|
|
||||||
|
|
||||||
#define MATCHFILTER(it,pk) \
|
|
||||||
((*(IFT *)(pk) & VEC_BUF(filters,it).m) == VEC_BUF(filters,it).f)
|
|
||||||
|
|
||||||
#define DOFILTER(it,pk,code) \
|
|
||||||
do { \
|
|
||||||
for (it = 0;it < VEC_LENGTH(filters);++it) { \
|
|
||||||
if (unlikely(MATCHFILTER(it,pk))) { \
|
|
||||||
code; \
|
|
||||||
break; \
|
|
||||||
} \
|
|
||||||
} \
|
|
||||||
} while (0)
|
|
||||||
|
|
||||||
# else // BINSEARCH
|
|
||||||
|
|
||||||
# ifdef OMITMASK
|
|
||||||
|
|
||||||
#define DOFILTER(it,pk,code) \
|
|
||||||
do { \
|
|
||||||
register IFT maskedpk = *(IFT *)(pk) & ifiltermask; \
|
|
||||||
for (size_t down = 0,up = VEC_LENGTH(filters);down < up;) { \
|
|
||||||
it = (up + down) / 2; \
|
|
||||||
if (maskedpk < VEC_BUF(filters,it).f) \
|
|
||||||
up = it; \
|
|
||||||
else if (maskedpk > VEC_BUF(filters,it).f) \
|
|
||||||
down = it + 1; \
|
|
||||||
else { \
|
|
||||||
code; \
|
|
||||||
break; \
|
|
||||||
} \
|
|
||||||
} \
|
|
||||||
} while (0)
|
|
||||||
|
|
||||||
# else // OMITMASK
|
|
||||||
|
|
||||||
#define DOFILTER(it,pk,code) \
|
|
||||||
do { \
|
|
||||||
for (size_t down = 0,up = VEC_LENGTH(filters);down < up;) { \
|
|
||||||
it = (up + down) / 2; \
|
|
||||||
IFT maskedpk = *(IFT *)(pk) & VEC_BUF(filters,it).m; \
|
|
||||||
register int cmp = memcmp(&maskedpk,&VEC_BUF(filters,it).f,sizeof(IFT)); \
|
|
||||||
if (cmp < 0) \
|
|
||||||
up = it; \
|
|
||||||
else if (cmp > 0) \
|
|
||||||
down = it + 1; \
|
|
||||||
else { \
|
|
||||||
code; \
|
|
||||||
break; \
|
|
||||||
} \
|
|
||||||
} \
|
|
||||||
} while (0)
|
|
||||||
|
|
||||||
# endif // OMITMASK
|
|
||||||
|
|
||||||
# endif // BINSEARCH
|
|
||||||
|
|
||||||
#define PREFILTER
|
|
||||||
#define POSTFILTER
|
|
||||||
|
|
||||||
#endif // INTFILTER
|
|
||||||
|
|
||||||
|
|
||||||
#ifdef BINFILTER
|
|
||||||
|
|
||||||
# ifndef BINSEARCH
|
|
||||||
|
|
||||||
#define MATCHFILTER(it,pk) ( \
|
|
||||||
memcmp(pk,VEC_BUF(filters,it).f,VEC_BUF(filters,it).len) == 0 && \
|
|
||||||
(pk[VEC_BUF(filters,it).len] & VEC_BUF(filters,it).mask) == VEC_BUF(filters,it).f[VEC_BUF(filters,it).len])
|
|
||||||
|
|
||||||
#define DOFILTER(it,pk,code) \
|
|
||||||
do { \
|
|
||||||
for (it = 0;it < VEC_LENGTH(filters);++it) { \
|
|
||||||
if (unlikely(MATCHFILTER(it,pk))) { \
|
|
||||||
code; \
|
|
||||||
break; \
|
|
||||||
} \
|
|
||||||
} \
|
|
||||||
} while (0)
|
|
||||||
|
|
||||||
# else // BINSEARCH
|
|
||||||
|
|
||||||
#define DOFILTER(it,pk,code) \
|
|
||||||
do { \
|
|
||||||
for (size_t down = 0,up = VEC_LENGTH(filters);down < up;) { \
|
|
||||||
it = (up + down) / 2; \
|
|
||||||
{ \
|
|
||||||
register int filterdiff = memcmp(pk,VEC_BUF(filters,it).f,VEC_BUF(filters,it).len); \
|
|
||||||
if (filterdiff < 0) { \
|
|
||||||
up = it; \
|
|
||||||
continue; \
|
|
||||||
} \
|
|
||||||
if (filterdiff > 0) { \
|
|
||||||
down = it + 1; \
|
|
||||||
continue; \
|
|
||||||
} \
|
|
||||||
} \
|
|
||||||
if ((pk[VEC_BUF(filters,it).len] & VEC_BUF(filters,it).mask) < \
|
|
||||||
VEC_BUF(filters,it).f[VEC_BUF(filters,it).len]) \
|
|
||||||
{ \
|
|
||||||
up = it; \
|
|
||||||
continue; \
|
|
||||||
} \
|
|
||||||
if ((pk[VEC_BUF(filters,it).len] & VEC_BUF(filters,it).mask) > \
|
|
||||||
VEC_BUF(filters,it).f[VEC_BUF(filters,it).len]) \
|
|
||||||
{ \
|
|
||||||
down = it + 1; \
|
|
||||||
continue; \
|
|
||||||
} \
|
|
||||||
{ \
|
|
||||||
code; \
|
|
||||||
break; \
|
|
||||||
} \
|
|
||||||
} \
|
|
||||||
} while (0)
|
|
||||||
|
|
||||||
# endif // BINSEARCH
|
|
||||||
|
|
||||||
#define PREFILTER
|
|
||||||
#define POSTFILTER
|
|
||||||
|
|
||||||
#endif // BINFILTER
|
|
||||||
|
|
||||||
|
|
||||||
#ifdef PCRE2FILTER
|
|
||||||
|
|
||||||
#define PREFILTER \
|
|
||||||
char pkconvbuf[BASE32_TO_LEN(PUBLIC_LEN) + 1]; \
|
|
||||||
pcre2_match_data *pcre2md = pcre2_match_data_create(128,0); \
|
|
||||||
PCRE2_SIZE *pcre2ovector = 0;
|
|
||||||
|
|
||||||
#define POSTFILTER \
|
|
||||||
pcre2_match_data_free(pcre2md);
|
|
||||||
|
|
||||||
#define DOFILTER(it,pk,code) \
|
|
||||||
do { \
|
|
||||||
base32_to(pkconvbuf,pk,PUBLIC_LEN); \
|
|
||||||
size_t __l = VEC_LENGTH(filters); \
|
|
||||||
for (it = 0;it < __l;++it) { \
|
|
||||||
int rc = pcre2_match(VEC_BUF(filters,it).re,(PCRE2_SPTR8)pkconvbuf,BASE32_TO_LEN(PUBLIC_LEN),0, \
|
|
||||||
PCRE2_NO_UTF_CHECK,pcre2md,0); \
|
|
||||||
if (unlikely(rc >= 0)) { \
|
|
||||||
pcre2ovector = pcre2_get_ovector_pointer(pcre2md); \
|
|
||||||
code; \
|
|
||||||
break; \
|
|
||||||
} \
|
|
||||||
} \
|
|
||||||
} while (0)
|
|
||||||
|
|
||||||
#endif // PCRE2FILTER
|
#endif // PCRE2FILTER
|
||||||
|
|
||||||
|
|
||||||
static void loadfilterfile(const char *fname)
|
extern int flattened;
|
||||||
{
|
|
||||||
char buf[128];
|
|
||||||
FILE *f = fopen(fname,"r");
|
|
||||||
while (fgets(buf,sizeof(buf),f)) {
|
|
||||||
for (char *p = buf;*p;++p) {
|
|
||||||
if (*p == '\n') {
|
|
||||||
*p = 0;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (*buf && *buf != '#' && memcmp(buf,"//",2) != 0)
|
|
||||||
filters_add(buf);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void filters_print(void)
|
extern void filters_init(void);
|
||||||
{
|
extern size_t filters_count(void);
|
||||||
if (quietflag)
|
|
||||||
return;
|
|
||||||
size_t i,l;
|
|
||||||
l = VEC_LENGTH(filters);
|
|
||||||
if (l)
|
|
||||||
fprintf(stderr,"filters:\n");
|
|
||||||
|
|
||||||
for (i = 0;i < l;++i) {
|
|
||||||
#ifdef NEEDBINFILTER
|
|
||||||
char buf0[256],buf1[256];
|
|
||||||
u8 bufx[128];
|
|
||||||
#endif
|
|
||||||
|
|
||||||
if (!verboseflag && i >= 20) {
|
|
||||||
size_t notshown = l - i;
|
|
||||||
fprintf(stderr,"[another " FSZ " %s not shown]\n",
|
|
||||||
notshown,notshown == 1 ? "filter" : "filters");
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
#ifdef INTFILTER
|
|
||||||
size_t len = 0;
|
|
||||||
u8 *imraw;
|
|
||||||
|
|
||||||
# ifndef OMITMASK
|
|
||||||
imraw = (u8 *)&VEC_BUF(filters,i).m;
|
|
||||||
# else
|
|
||||||
imraw = (u8 *)&ifiltermask;
|
|
||||||
# endif
|
|
||||||
while (len < sizeof(IFT) && imraw[len] != 0x00) ++len;
|
|
||||||
u8 mask = imraw[len-1];
|
|
||||||
u8 *ifraw = (u8 *)&VEC_BUF(filters,i).f;
|
|
||||||
#endif // INTFILTER
|
|
||||||
|
|
||||||
#ifdef BINFILTER
|
|
||||||
size_t len = VEC_BUF(filters,i).len + 1;
|
|
||||||
u8 mask = VEC_BUF(filters,i).mask;
|
|
||||||
u8 *ifraw = VEC_BUF(filters,i).f;
|
|
||||||
#endif // BINFILTER
|
|
||||||
#ifdef NEEDBINFILTER
|
|
||||||
base32_to(buf0,ifraw,len);
|
|
||||||
memcpy(bufx,ifraw,len);
|
|
||||||
bufx[len - 1] |= ~mask;
|
|
||||||
base32_to(buf1,bufx,len);
|
|
||||||
char *a = buf0,*b = buf1;
|
|
||||||
while (*a && *a == *b)
|
|
||||||
++a, ++b;
|
|
||||||
*a = 0;
|
|
||||||
fprintf(stderr,"\t%s\n",buf0);
|
|
||||||
#endif // NEEDBINFILTER
|
|
||||||
#ifdef PCRE2FILTER
|
|
||||||
fprintf(stderr,"\t%s\n",VEC_BUF(filters,i).str);
|
|
||||||
#endif // PCRE2FILTER
|
|
||||||
}
|
|
||||||
fprintf(stderr,"in total, " FSZ " %s\n",l,l == 1 ? "filter" : "filters");
|
|
||||||
}
|
|
||||||
|
|
|
||||||
51
filters_common.inc.h
Normal file
51
filters_common.inc.h
Normal file
|
|
@ -0,0 +1,51 @@
|
||||||
|
#ifdef INTFILTER
|
||||||
|
|
||||||
|
static inline size_t filter_len(size_t i)
|
||||||
|
{
|
||||||
|
# ifndef OMITMASK
|
||||||
|
const u8 *m = (const u8 *)&VEC_BUF(filters,i).m;
|
||||||
|
# else // OMITMASK
|
||||||
|
const u8 *m = (const u8 *)&ifiltermask;
|
||||||
|
# endif // OMITMASK
|
||||||
|
size_t c = 0;
|
||||||
|
for (size_t j = 0;;) {
|
||||||
|
u8 v = m[j];
|
||||||
|
for (size_t k = 0;;) {
|
||||||
|
if (!v)
|
||||||
|
return c;
|
||||||
|
++c;
|
||||||
|
if (++k >= 8)
|
||||||
|
break;
|
||||||
|
v <<= 1;
|
||||||
|
}
|
||||||
|
if (++j >= sizeof(IFT))
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
return c;
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif // INTFILTER
|
||||||
|
|
||||||
|
#ifdef BINFILTER
|
||||||
|
|
||||||
|
static inline size_t filter_len(size_t i)
|
||||||
|
{
|
||||||
|
size_t c = VEC_BUF(filters,i).len * 8;
|
||||||
|
u8 v = VEC_BUF(filters,i).mask;
|
||||||
|
for (size_t k = 0;;) {
|
||||||
|
if (!v)
|
||||||
|
return c;
|
||||||
|
++c;
|
||||||
|
if (++k >= 8)
|
||||||
|
return c;
|
||||||
|
v <<= 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif // BINFILTER
|
||||||
|
|
||||||
|
#ifdef PCRE2FILTER
|
||||||
|
|
||||||
|
#define filter_len(i) ((pcre2ovector[1] - pcre2ovector[0]) * 5)
|
||||||
|
|
||||||
|
#endif // PCRE2FILTER
|
||||||
513
filters_main.inc.h
Normal file
513
filters_main.inc.h
Normal file
|
|
@ -0,0 +1,513 @@
|
||||||
|
|
||||||
|
#include "filters_common.inc.h"
|
||||||
|
|
||||||
|
#ifdef INTFILTER
|
||||||
|
|
||||||
|
# ifdef OMITMASK
|
||||||
|
|
||||||
|
static inline int filter_compare(const void *p1,const void *p2)
|
||||||
|
{
|
||||||
|
if (((const struct intfilter *)p1)->f < ((const struct intfilter *)p2)->f)
|
||||||
|
return -1;
|
||||||
|
if (((const struct intfilter *)p1)->f > ((const struct intfilter *)p2)->f)
|
||||||
|
return 1;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
# ifdef EXPANDMASK
|
||||||
|
|
||||||
|
/*
|
||||||
|
* for mask expansion, we need to figure out how much bits
|
||||||
|
* we need to fill in with different values.
|
||||||
|
* while in big endian machines this is quite easy,
|
||||||
|
* representation we use for little endian ones may
|
||||||
|
* leave gap of bits we don't want to touch.
|
||||||
|
*
|
||||||
|
* initial idea draft:
|
||||||
|
*
|
||||||
|
* raw representation -- FF.FF.F0.00
|
||||||
|
* big endian -- 0xFFFFF000
|
||||||
|
* little endian -- 0x00F0FFFF
|
||||||
|
* b: 0xFFffF000 ^ 0xFFff0000 -> 0x0000F000
|
||||||
|
* 0x0000F000 + 1 -> 0x0000F001
|
||||||
|
* 0x0000F000 & 0x0000F001 -> 0x0000F000 <- shifted mask
|
||||||
|
* 0x0000F000 ^ 0x0000F000 -> 0x00000000 <- direct mask
|
||||||
|
* 0x0000F000 ^ 0x00000000 -> 0x0000F000 <- shifted mask
|
||||||
|
* l: 0x00f0FFff ^ 0x0000FFff -> 0x00f00000
|
||||||
|
* 0x00f00000 + 1 -> 0x00f00001
|
||||||
|
* 0x00f00000 & 0x00f00001 -> 0x00f00000 <- shifted mask
|
||||||
|
* 0x00f00000 ^ 0x00f00000 -> 0x00000000 <- direct mask
|
||||||
|
* 0x00f00000 ^ 0x00000000 -> 0x00f00000 <- shifted mask
|
||||||
|
*
|
||||||
|
* b: 0xFFffFFff ^ 0xF0000000 -> 0x0FffFFff
|
||||||
|
* 0x0FffFFff + 1 -> 0x10000000
|
||||||
|
* 0x0FffFFff & 0x10000000 -> 0x00000000 <- shifted mask
|
||||||
|
* 0x0FffFFff ^ 0x00000000 -> 0x0FffFFff <- direct mask
|
||||||
|
* 0x0FffFFff ^ 0x0FffFFff -> 0x00000000 <- shifted mask
|
||||||
|
* l: 0xFFffFFff ^ 0x000000f0 -> 0xFFffFF0f
|
||||||
|
* 0xFFffFF0f + 1 -> 0xFFffFF10
|
||||||
|
* 0xFFffFF0f & 0xFFffFF10 -> 0xFFffFF00 <- shifted mask
|
||||||
|
* 0xFFffFF0f ^ 0xFFffFF00 -> 0x0000000f <- direct mask
|
||||||
|
* 0xFFffFF0f ^ 0x0000000f -> 0xFFffFF00 <- shifted mask
|
||||||
|
*
|
||||||
|
* essentially, we have to make direct mask + shifted mask bits worth of information
|
||||||
|
* and then split it into 2 parts
|
||||||
|
* we do not need absolute shifted mask shifting value, just relative to direct mask
|
||||||
|
* 0x0sss00dd - shifted & direct mask combo
|
||||||
|
* 0x000sssdd - combined mask
|
||||||
|
* 8 - relshiftval
|
||||||
|
* generate values from 0x00000000 to 0x000sssdd
|
||||||
|
* for each value, realmask <- (val & 0x000000dd) | ((val & 0x000sss00) << relshiftval)
|
||||||
|
* or..
|
||||||
|
* realmask <- (val & 0x000000dd) | ((val << relshiftval) & 0x0sss0000)
|
||||||
|
* ...
|
||||||
|
*
|
||||||
|
* above method doesn't work in some cases. better way:
|
||||||
|
*
|
||||||
|
* l: 0x80ffFFff ^ 0x00f0FFff -> 0x800f0000
|
||||||
|
* 0x800f0000 >> 16 -> 0x0000800f
|
||||||
|
* 0x0000800f + 1 -> 0x00008010
|
||||||
|
* 0x0000800f & 0x00008010 -> 0x00008000 <- smask
|
||||||
|
* 0x0000800f ^ 0x00008000 -> 0x0000000f <- dmask
|
||||||
|
*
|
||||||
|
* cross <- difference between mask we desire and mask we currently have
|
||||||
|
* shift cross to left variable ammount of times to eliminate zeros
|
||||||
|
* save shift ammount as ishift (initial shift)
|
||||||
|
* then, we eliminate first area of ones; if there was no gap, result is already all zeros
|
||||||
|
* save this thing as smask. it's only higher bits.
|
||||||
|
* XOR smask and cross; result is only lower bits.
|
||||||
|
* shift smask to left variable ammount of times until gap is eliminated.
|
||||||
|
* save resulting mask as cmask;
|
||||||
|
* save resulting shift value as rshift.
|
||||||
|
*/
|
||||||
|
|
||||||
|
int flattened = 0;
|
||||||
|
|
||||||
|
#define EXPVAL(init,j,dmask,smask,ishift,rshift) \
|
||||||
|
((init) | ((((j) & (dmask)) | (((j) << (rshift)) & (smask))) << (ishift)))
|
||||||
|
// add expanded set of values
|
||||||
|
// allocates space on its own
|
||||||
|
static void ifilter_addexpanded(
|
||||||
|
struct intfilter *ifltr,
|
||||||
|
IFT dmask,IFT smask,IFT cmask,
|
||||||
|
int ishift,int rshift)
|
||||||
|
{
|
||||||
|
flattened = 1;
|
||||||
|
size_t i = VEC_LENGTH(filters);
|
||||||
|
VEC_ADDN(filters,cmask + 1);
|
||||||
|
for (size_t j = 0;;++j) {
|
||||||
|
VEC_BUF(filters,i + j).f =
|
||||||
|
EXPVAL(ifltr->f,j,dmask,smask,ishift,rshift);
|
||||||
|
if (j == cmask)
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// expand existing stuff
|
||||||
|
// allocates needed stuff on its own
|
||||||
|
static void ifilter_expand(IFT dmask,IFT smask,IFT cmask,int ishift,int rshift)
|
||||||
|
{
|
||||||
|
flattened = 1;
|
||||||
|
size_t len = VEC_LENGTH(filters);
|
||||||
|
VEC_ADDN(filters,cmask * len);
|
||||||
|
size_t esz = cmask + 1; // size of expanded elements
|
||||||
|
for (size_t i = len - 1;;--i) {
|
||||||
|
for (IFT j = 0;;++j) {
|
||||||
|
VEC_BUF(filters,i * esz + j).f =
|
||||||
|
EXPVAL(VEC_BUF(filters,i).f,j,dmask,smask,ishift,rshift);
|
||||||
|
if (j == cmask)
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if (i == 0)
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline void ifilter_addflatten(struct intfilter *ifltr,IFT mask)
|
||||||
|
{
|
||||||
|
if (VEC_LENGTH(filters) == 0) {
|
||||||
|
// simple
|
||||||
|
VEC_ADD(filters,*ifltr);
|
||||||
|
ifiltermask = mask;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (ifiltermask == mask) {
|
||||||
|
// lucky
|
||||||
|
VEC_ADD(filters,*ifltr);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
IFT cross = ifiltermask ^ mask;
|
||||||
|
int ishift = 0;
|
||||||
|
while ((cross & 1) == 0) {
|
||||||
|
++ishift;
|
||||||
|
cross >>= 1;
|
||||||
|
}
|
||||||
|
IFT smask = cross & (cross + 1); // shift mask
|
||||||
|
IFT dmask = cross ^ smask; // direct mask
|
||||||
|
IFT cmask; // combined mask
|
||||||
|
int rshift = 0; // relative shift
|
||||||
|
while (cmask = (smask >> rshift) | dmask,(cmask & (cmask + 1)) != 0)
|
||||||
|
++rshift;
|
||||||
|
// preparations done
|
||||||
|
if (ifiltermask > mask) {
|
||||||
|
// already existing stuff has more precise mask than we
|
||||||
|
// so we need to expand our stuff
|
||||||
|
ifilter_addexpanded(ifltr,dmask,smask,cmask,ishift,rshift);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
ifiltermask = mask;
|
||||||
|
ifilter_expand(dmask,smask,cmask,ishift,rshift);
|
||||||
|
VEC_ADD(filters,*ifltr);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
# endif // EXPANDMASK
|
||||||
|
|
||||||
|
# else // OMITMASK
|
||||||
|
|
||||||
|
/*
|
||||||
|
* struct intfilter layout: filter,mask
|
||||||
|
* stuff is compared in big-endian way, so memcmp
|
||||||
|
* filter needs to be compared first
|
||||||
|
* if its equal, mask needs to be compared
|
||||||
|
* memcmp is aplicable there too
|
||||||
|
* due to struct intfilter layout, it all can be stuffed into one memcmp call
|
||||||
|
*/
|
||||||
|
static inline int filter_compare(const void *p1,const void *p2)
|
||||||
|
{
|
||||||
|
return memcmp(p1,p2,sizeof(struct intfilter));
|
||||||
|
}
|
||||||
|
|
||||||
|
# endif // OMITMASK
|
||||||
|
|
||||||
|
static void filter_sort(void)
|
||||||
|
{
|
||||||
|
size_t len = VEC_LENGTH(filters);
|
||||||
|
if (len > 0)
|
||||||
|
qsort(&VEC_BUF(filters,0),len,sizeof(struct intfilter),&filter_compare);
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif // INTFILTER
|
||||||
|
|
||||||
|
#ifdef BINFILTER
|
||||||
|
|
||||||
|
static inline int filter_compare(const void *p1,const void *p2)
|
||||||
|
{
|
||||||
|
const struct binfilter *b1 = (const struct binfilter *)p1;
|
||||||
|
const struct binfilter *b2 = (const struct binfilter *)p2;
|
||||||
|
|
||||||
|
size_t l = b1->len <= b2->len ? b1->len : b2->len;
|
||||||
|
|
||||||
|
int cmp = memcmp(b1->f,b2->f,l);
|
||||||
|
if (cmp != 0)
|
||||||
|
return cmp;
|
||||||
|
|
||||||
|
if (b1->len < b2->len)
|
||||||
|
return -1;
|
||||||
|
if (b1->len > b2->len)
|
||||||
|
return +1;
|
||||||
|
|
||||||
|
u8 cmask = b1->mask & b2->mask;
|
||||||
|
if ((b1->f[l] & cmask) < (b2->f[l] & cmask))
|
||||||
|
return -1;
|
||||||
|
if ((b1->f[l] & cmask) > (b2->f[l] & cmask))
|
||||||
|
return +1;
|
||||||
|
|
||||||
|
if (b1->mask < b2->mask)
|
||||||
|
return -1;
|
||||||
|
if (b1->mask > b2->mask)
|
||||||
|
return +1;
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void filter_sort(void)
|
||||||
|
{
|
||||||
|
size_t len = VEC_LENGTH(filters);
|
||||||
|
if (len > 0)
|
||||||
|
qsort(&VEC_BUF(filters,0),len,sizeof(struct binfilter),&filter_compare);
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif // BINFILTER
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
#ifndef PCRE2FILTER
|
||||||
|
static inline int filters_a_includes_b(size_t a,size_t b)
|
||||||
|
{
|
||||||
|
# ifdef INTFILTER
|
||||||
|
# ifdef OMITMASK
|
||||||
|
return VEC_BUF(filters,a).f == VEC_BUF(filters,b).f;
|
||||||
|
# else // OMITMASK
|
||||||
|
return VEC_BUF(filters,a).f == (VEC_BUF(filters,b).f & VEC_BUF(filters,a).m);
|
||||||
|
# endif // OMITMASK
|
||||||
|
# else // INTFILTER
|
||||||
|
const struct binfilter *fa = &VEC_BUF(filters,a);
|
||||||
|
const struct binfilter *fb = &VEC_BUF(filters,b);
|
||||||
|
|
||||||
|
if (fa->len > fb->len)
|
||||||
|
return 0;
|
||||||
|
size_t l = fa->len;
|
||||||
|
|
||||||
|
int cmp = memcmp(fa->f,fb->f,l);
|
||||||
|
if (cmp != 0)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
if (fa->len < fb->len)
|
||||||
|
return 1;
|
||||||
|
|
||||||
|
if (fa->mask > fb->mask)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
return fa->f[l] == (fb->f[l] & fa->mask);
|
||||||
|
# endif // INTFILTER
|
||||||
|
}
|
||||||
|
|
||||||
|
static void filters_dedup(void)
|
||||||
|
{
|
||||||
|
size_t last = ~(size_t)0; // index after last matching element
|
||||||
|
size_t chk; // element to compare against
|
||||||
|
size_t st; // start of area to destroy
|
||||||
|
|
||||||
|
size_t len = VEC_LENGTH(filters);
|
||||||
|
for (size_t i = 1;i < len;++i) {
|
||||||
|
if (last != i) {
|
||||||
|
if (filters_a_includes_b(i - 1,i)) {
|
||||||
|
if (last != ~(size_t)0) {
|
||||||
|
memmove(&VEC_BUF(filters,st),
|
||||||
|
&VEC_BUF(filters,last),
|
||||||
|
(i - last) * VEC_ELSIZE(filters));
|
||||||
|
st += i - last;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
st = i;
|
||||||
|
chk = i - 1;
|
||||||
|
last = i + 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
if (filters_a_includes_b(chk,i))
|
||||||
|
last = i + 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (last != ~(size_t)0) {
|
||||||
|
memmove(&VEC_BUF(filters,st),
|
||||||
|
&VEC_BUF(filters,last),
|
||||||
|
(len - last) * VEC_ELSIZE(filters));
|
||||||
|
st += len - last;
|
||||||
|
VEC_SETLENGTH(filters,st);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif // !PCRE2FILTER
|
||||||
|
|
||||||
|
static void filters_clean(void)
|
||||||
|
{
|
||||||
|
#ifdef PCRE2FILTER
|
||||||
|
for (size_t i = 0;i < VEC_LENGTH(filters);++i) {
|
||||||
|
pcre2_code_free(VEC_BUF(filters,i).re);
|
||||||
|
free(VEC_BUF(filters,i).str);
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
VEC_FREE(filters);
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t filters_count(void)
|
||||||
|
{
|
||||||
|
return VEC_LENGTH(filters);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static void filters_print(void)
|
||||||
|
{
|
||||||
|
if (quietflag)
|
||||||
|
return;
|
||||||
|
size_t i,l;
|
||||||
|
l = VEC_LENGTH(filters);
|
||||||
|
if (l)
|
||||||
|
fprintf(stderr,"filters:\n");
|
||||||
|
|
||||||
|
for (i = 0;i < l;++i) {
|
||||||
|
#ifdef NEEDBINFILTER
|
||||||
|
char buf0[256],buf1[256];
|
||||||
|
u8 bufx[128];
|
||||||
|
#endif
|
||||||
|
|
||||||
|
if (!verboseflag && i >= 20) {
|
||||||
|
size_t notshown = l - i;
|
||||||
|
fprintf(stderr,"[another " FSZ " %s not shown]\n",
|
||||||
|
notshown,notshown == 1 ? "filter" : "filters");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef INTFILTER
|
||||||
|
size_t len = 0;
|
||||||
|
u8 *imraw;
|
||||||
|
|
||||||
|
# ifndef OMITMASK
|
||||||
|
imraw = (u8 *)&VEC_BUF(filters,i).m;
|
||||||
|
# else
|
||||||
|
imraw = (u8 *)&ifiltermask;
|
||||||
|
# endif
|
||||||
|
while (len < sizeof(IFT) && imraw[len] != 0x00) ++len;
|
||||||
|
u8 mask = imraw[len-1];
|
||||||
|
u8 *ifraw = (u8 *)&VEC_BUF(filters,i).f;
|
||||||
|
#endif // INTFILTER
|
||||||
|
|
||||||
|
#ifdef BINFILTER
|
||||||
|
size_t len = VEC_BUF(filters,i).len + 1;
|
||||||
|
u8 mask = VEC_BUF(filters,i).mask;
|
||||||
|
u8 *ifraw = VEC_BUF(filters,i).f;
|
||||||
|
#endif // BINFILTER
|
||||||
|
#ifdef NEEDBINFILTER
|
||||||
|
base32_to(buf0,ifraw,len);
|
||||||
|
memcpy(bufx,ifraw,len);
|
||||||
|
bufx[len - 1] |= ~mask;
|
||||||
|
base32_to(buf1,bufx,len);
|
||||||
|
char *a = buf0,*b = buf1;
|
||||||
|
while (*a && *a == *b)
|
||||||
|
++a, ++b;
|
||||||
|
*a = 0;
|
||||||
|
fprintf(stderr,"\t%s\n",buf0);
|
||||||
|
#endif // NEEDBINFILTER
|
||||||
|
#ifdef PCRE2FILTER
|
||||||
|
fprintf(stderr,"\t%s\n",VEC_BUF(filters,i).str);
|
||||||
|
#endif // PCRE2FILTER
|
||||||
|
}
|
||||||
|
fprintf(stderr,"in total, " FSZ " %s\n",l,l == 1 ? "filter" : "filters");
|
||||||
|
}
|
||||||
|
|
||||||
|
void filters_add(const char *filter)
|
||||||
|
{
|
||||||
|
#ifdef NEEDBINFILTER
|
||||||
|
struct binfilter bf;
|
||||||
|
size_t ret;
|
||||||
|
# ifdef INTFILTER
|
||||||
|
union intconv {
|
||||||
|
IFT i;
|
||||||
|
u8 b[sizeof(IFT)];
|
||||||
|
} fc,mc;
|
||||||
|
# endif
|
||||||
|
|
||||||
|
// skip regex start symbol. we do not support regex tho
|
||||||
|
if (*filter == '^')
|
||||||
|
++filter;
|
||||||
|
|
||||||
|
memset(&bf,0,sizeof(bf));
|
||||||
|
|
||||||
|
if (!base32_valid(filter,&ret)) {
|
||||||
|
fprintf(stderr,"filter \"%s\" is invalid\n",filter);
|
||||||
|
fprintf(stderr," ");
|
||||||
|
while (ret--)
|
||||||
|
fputc(' ',stderr);
|
||||||
|
fprintf(stderr,"^\n");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
ret = BASE32_FROM_LEN(ret);
|
||||||
|
if (!ret)
|
||||||
|
return;
|
||||||
|
# ifdef INTFILTER
|
||||||
|
size_t maxsz = sizeof(IFT);
|
||||||
|
# else
|
||||||
|
size_t maxsz = sizeof(bf.f);
|
||||||
|
# endif
|
||||||
|
if (ret > maxsz) {
|
||||||
|
fprintf(stderr,"filter \"%s\" is too long\n",filter);
|
||||||
|
fprintf(stderr," ");
|
||||||
|
maxsz = (maxsz * 8) / 5;
|
||||||
|
while (maxsz--)
|
||||||
|
fputc(' ',stderr);
|
||||||
|
fprintf(stderr,"^\n");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
base32_from(bf.f,&bf.mask,filter);
|
||||||
|
bf.len = ret - 1;
|
||||||
|
|
||||||
|
# ifdef INTFILTER
|
||||||
|
mc.i = 0;
|
||||||
|
for (size_t i = 0;i < bf.len;++i)
|
||||||
|
mc.b[i] = 0xFF;
|
||||||
|
mc.b[bf.len] = bf.mask;
|
||||||
|
memcpy(fc.b,bf.f,sizeof(fc.b));
|
||||||
|
fc.i &= mc.i;
|
||||||
|
|
||||||
|
struct intfilter ifltr = {
|
||||||
|
.f = fc.i,
|
||||||
|
# ifndef OMITMASK
|
||||||
|
.m = mc.i,
|
||||||
|
# endif
|
||||||
|
};
|
||||||
|
|
||||||
|
# ifdef OMITMASK
|
||||||
|
ifilter_addflatten(&ifltr,mc.i);
|
||||||
|
# else // OMITMASK
|
||||||
|
VEC_ADD(filters,ifltr);
|
||||||
|
# endif // OMITMASK
|
||||||
|
# endif // INTFILTER
|
||||||
|
|
||||||
|
# ifdef BINFILTER
|
||||||
|
VEC_ADD(filters,bf);
|
||||||
|
# endif // BINFILTER
|
||||||
|
#endif // NEEDBINFILTER
|
||||||
|
|
||||||
|
#ifdef PCRE2FILTER
|
||||||
|
int errornum;
|
||||||
|
PCRE2_SIZE erroroffset;
|
||||||
|
pcre2_code *re;
|
||||||
|
|
||||||
|
re = pcre2_compile((PCRE2_SPTR8)filter,PCRE2_ZERO_TERMINATED,
|
||||||
|
PCRE2_NO_UTF_CHECK | PCRE2_ANCHORED,&errornum,&erroroffset,0);
|
||||||
|
if (!re) {
|
||||||
|
PCRE2_UCHAR buffer[1024];
|
||||||
|
pcre2_get_error_message(errornum,buffer,sizeof(buffer));
|
||||||
|
fprintf(stderr,"PCRE2 compilation failed at offset " FSZ ": %s\n",
|
||||||
|
(size_t)erroroffset,buffer);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// attempt to JIT. ignore error
|
||||||
|
(void) pcre2_jit_compile(re,PCRE2_JIT_COMPLETE);
|
||||||
|
|
||||||
|
struct pcre2filter f;
|
||||||
|
memset(&f,0,sizeof(f));
|
||||||
|
f.re = re;
|
||||||
|
size_t fl = strlen(filter) + 1;
|
||||||
|
f.str = (char *) malloc(fl);
|
||||||
|
if (!f.str)
|
||||||
|
abort();
|
||||||
|
memcpy(f.str,filter,fl);
|
||||||
|
VEC_ADD(filters,f);
|
||||||
|
#endif // PCRE2FILTER
|
||||||
|
}
|
||||||
|
|
||||||
|
static void filters_prepare(void)
|
||||||
|
{
|
||||||
|
#ifndef PCRE2FILTER
|
||||||
|
if (!quietflag)
|
||||||
|
fprintf(stderr,"sorting filters...");
|
||||||
|
filter_sort();
|
||||||
|
if (wantdedup) {
|
||||||
|
if (!quietflag)
|
||||||
|
fprintf(stderr," removing duplicates...");
|
||||||
|
filters_dedup();
|
||||||
|
}
|
||||||
|
if (!quietflag)
|
||||||
|
fprintf(stderr," done.\n");
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
static void loadfilterfile(const char *fname)
|
||||||
|
{
|
||||||
|
char buf[128];
|
||||||
|
FILE *f = fopen(fname,"r");
|
||||||
|
while (fgets(buf,sizeof(buf),f)) {
|
||||||
|
for (char *p = buf;*p;++p) {
|
||||||
|
if (*p == '\n') {
|
||||||
|
*p = 0;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (*buf && *buf != '#' && memcmp(buf,"//",2) != 0)
|
||||||
|
filters_add(buf);
|
||||||
|
}
|
||||||
|
}
|
||||||
194
filters_worker.inc.h
Normal file
194
filters_worker.inc.h
Normal file
|
|
@ -0,0 +1,194 @@
|
||||||
|
|
||||||
|
#ifdef BINFILTER
|
||||||
|
|
||||||
|
struct bfiltervec filters;
|
||||||
|
|
||||||
|
#endif // BINFILTER
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
#ifdef INTFILTER
|
||||||
|
|
||||||
|
struct ifiltervec filters;
|
||||||
|
|
||||||
|
# ifdef OMITMASK
|
||||||
|
IFT ifiltermask;
|
||||||
|
# endif
|
||||||
|
|
||||||
|
#endif // INTFILTER
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
#ifdef PCRE2FILTER
|
||||||
|
|
||||||
|
struct pfiltervec filters;
|
||||||
|
|
||||||
|
#endif // PCRE2FILTER
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
void filters_init(void)
|
||||||
|
{
|
||||||
|
VEC_INIT(filters);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
#include "filters_common.inc.h"
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
#ifdef INTFILTER
|
||||||
|
|
||||||
|
# ifndef BINSEARCH
|
||||||
|
|
||||||
|
#define MATCHFILTER(it,pk) \
|
||||||
|
((*(IFT *)(pk) & VEC_BUF(filters,it).m) == VEC_BUF(filters,it).f)
|
||||||
|
|
||||||
|
#define DOFILTER(it,pk,code) \
|
||||||
|
do { \
|
||||||
|
for (it = 0;it < VEC_LENGTH(filters);++it) { \
|
||||||
|
if (unlikely(MATCHFILTER(it,pk))) { \
|
||||||
|
code; \
|
||||||
|
break; \
|
||||||
|
} \
|
||||||
|
} \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
# else // BINSEARCH
|
||||||
|
|
||||||
|
# ifdef OMITMASK
|
||||||
|
|
||||||
|
#define DOFILTER(it,pk,code) \
|
||||||
|
do { \
|
||||||
|
register IFT maskedpk = *(IFT *)(pk) & ifiltermask; \
|
||||||
|
for (size_t down = 0,up = VEC_LENGTH(filters);down < up;) { \
|
||||||
|
it = (up + down) / 2; \
|
||||||
|
if (maskedpk < VEC_BUF(filters,it).f) \
|
||||||
|
up = it; \
|
||||||
|
else if (maskedpk > VEC_BUF(filters,it).f) \
|
||||||
|
down = it + 1; \
|
||||||
|
else { \
|
||||||
|
code; \
|
||||||
|
break; \
|
||||||
|
} \
|
||||||
|
} \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
# else // OMITMASK
|
||||||
|
|
||||||
|
#define DOFILTER(it,pk,code) \
|
||||||
|
do { \
|
||||||
|
for (size_t down = 0,up = VEC_LENGTH(filters);down < up;) { \
|
||||||
|
it = (up + down) / 2; \
|
||||||
|
IFT maskedpk = *(IFT *)(pk) & VEC_BUF(filters,it).m; \
|
||||||
|
register int cmp = memcmp(&maskedpk,&VEC_BUF(filters,it).f,sizeof(IFT)); \
|
||||||
|
if (cmp < 0) \
|
||||||
|
up = it; \
|
||||||
|
else if (cmp > 0) \
|
||||||
|
down = it + 1; \
|
||||||
|
else { \
|
||||||
|
code; \
|
||||||
|
break; \
|
||||||
|
} \
|
||||||
|
} \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
# endif // OMITMASK
|
||||||
|
|
||||||
|
# endif // BINSEARCH
|
||||||
|
|
||||||
|
#define PREFILTER
|
||||||
|
#define POSTFILTER
|
||||||
|
|
||||||
|
#endif // INTFILTER
|
||||||
|
|
||||||
|
|
||||||
|
#ifdef BINFILTER
|
||||||
|
|
||||||
|
# ifndef BINSEARCH
|
||||||
|
|
||||||
|
#define MATCHFILTER(it,pk) ( \
|
||||||
|
memcmp(pk,VEC_BUF(filters,it).f,VEC_BUF(filters,it).len) == 0 && \
|
||||||
|
(pk[VEC_BUF(filters,it).len] & VEC_BUF(filters,it).mask) == VEC_BUF(filters,it).f[VEC_BUF(filters,it).len])
|
||||||
|
|
||||||
|
#define DOFILTER(it,pk,code) \
|
||||||
|
do { \
|
||||||
|
for (it = 0;it < VEC_LENGTH(filters);++it) { \
|
||||||
|
if (unlikely(MATCHFILTER(it,pk))) { \
|
||||||
|
code; \
|
||||||
|
break; \
|
||||||
|
} \
|
||||||
|
} \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
# else // BINSEARCH
|
||||||
|
|
||||||
|
#define DOFILTER(it,pk,code) \
|
||||||
|
do { \
|
||||||
|
for (size_t down = 0,up = VEC_LENGTH(filters);down < up;) { \
|
||||||
|
it = (up + down) / 2; \
|
||||||
|
{ \
|
||||||
|
register int filterdiff = memcmp(pk,VEC_BUF(filters,it).f,VEC_BUF(filters,it).len); \
|
||||||
|
if (filterdiff < 0) { \
|
||||||
|
up = it; \
|
||||||
|
continue; \
|
||||||
|
} \
|
||||||
|
if (filterdiff > 0) { \
|
||||||
|
down = it + 1; \
|
||||||
|
continue; \
|
||||||
|
} \
|
||||||
|
} \
|
||||||
|
if ((pk[VEC_BUF(filters,it).len] & VEC_BUF(filters,it).mask) < \
|
||||||
|
VEC_BUF(filters,it).f[VEC_BUF(filters,it).len]) \
|
||||||
|
{ \
|
||||||
|
up = it; \
|
||||||
|
continue; \
|
||||||
|
} \
|
||||||
|
if ((pk[VEC_BUF(filters,it).len] & VEC_BUF(filters,it).mask) > \
|
||||||
|
VEC_BUF(filters,it).f[VEC_BUF(filters,it).len]) \
|
||||||
|
{ \
|
||||||
|
down = it + 1; \
|
||||||
|
continue; \
|
||||||
|
} \
|
||||||
|
{ \
|
||||||
|
code; \
|
||||||
|
break; \
|
||||||
|
} \
|
||||||
|
} \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
# endif // BINSEARCH
|
||||||
|
|
||||||
|
#define PREFILTER
|
||||||
|
#define POSTFILTER
|
||||||
|
|
||||||
|
#endif // BINFILTER
|
||||||
|
|
||||||
|
|
||||||
|
#ifdef PCRE2FILTER
|
||||||
|
|
||||||
|
#define PREFILTER \
|
||||||
|
char pkconvbuf[BASE32_TO_LEN(PUBLIC_LEN) + 1]; \
|
||||||
|
pcre2_match_data *pcre2md = pcre2_match_data_create(128,0); \
|
||||||
|
PCRE2_SIZE *pcre2ovector = 0;
|
||||||
|
|
||||||
|
#define POSTFILTER \
|
||||||
|
pcre2_match_data_free(pcre2md);
|
||||||
|
|
||||||
|
#define DOFILTER(it,pk,code) \
|
||||||
|
do { \
|
||||||
|
base32_to(pkconvbuf,pk,PUBLIC_LEN); \
|
||||||
|
size_t __l = VEC_LENGTH(filters); \
|
||||||
|
for (it = 0;it < __l;++it) { \
|
||||||
|
int rc = pcre2_match(VEC_BUF(filters,it).re,(PCRE2_SPTR8)pkconvbuf,BASE32_TO_LEN(PUBLIC_LEN),0, \
|
||||||
|
PCRE2_NO_UTF_CHECK,pcre2md,0); \
|
||||||
|
if (unlikely(rc >= 0)) { \
|
||||||
|
pcre2ovector = pcre2_get_ovector_pointer(pcre2md); \
|
||||||
|
code; \
|
||||||
|
break; \
|
||||||
|
} \
|
||||||
|
} \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
#endif // PCRE2FILTER
|
||||||
614
main.c
614
main.c
|
|
@ -12,48 +12,33 @@
|
||||||
#include <sodium/core.h>
|
#include <sodium/core.h>
|
||||||
#include <sodium/randombytes.h>
|
#include <sodium/randombytes.h>
|
||||||
#ifdef PASSPHRASE
|
#ifdef PASSPHRASE
|
||||||
#include <sodium/crypto_hash_sha256.h>
|
|
||||||
#include <sodium/crypto_pwhash.h>
|
#include <sodium/crypto_pwhash.h>
|
||||||
#endif
|
#endif
|
||||||
#include <sodium/utils.h>
|
#include <sodium/utils.h>
|
||||||
|
|
||||||
#include "types.h"
|
#include "types.h"
|
||||||
#include "likely.h"
|
|
||||||
#include "vec.h"
|
#include "vec.h"
|
||||||
#include "base32.h"
|
|
||||||
#include "cpucount.h"
|
#include "cpucount.h"
|
||||||
#include "keccak.h"
|
#include "keccak.h"
|
||||||
#include "ed25519/ed25519.h"
|
|
||||||
#include "ioutil.h"
|
#include "ioutil.h"
|
||||||
#include "common.h"
|
#include "common.h"
|
||||||
#include "yaml.h"
|
#include "yaml.h"
|
||||||
|
|
||||||
|
#include "filters.h"
|
||||||
|
|
||||||
|
#include "worker.h"
|
||||||
|
|
||||||
#ifndef _WIN32
|
#ifndef _WIN32
|
||||||
#define FSZ "%zu"
|
#define FSZ "%zu"
|
||||||
#else
|
#else
|
||||||
#define FSZ "%Iu"
|
#define FSZ "%Iu"
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// additional 0 terminator is added by C
|
|
||||||
static const char * const pkprefix = "== ed25519v1-public: type0 ==\0\0";
|
|
||||||
static const char * const skprefix = "== ed25519v1-secret: type0 ==\0\0";
|
|
||||||
|
|
||||||
static const char checksumstr[] = ".onion checksum";
|
|
||||||
#define checksumstrlen (sizeof(checksumstr) - 1) // 15
|
|
||||||
|
|
||||||
// How many times we loop before a reseed
|
|
||||||
#define DETERMINISTIC_LOOP_COUNT 1<<24
|
|
||||||
|
|
||||||
// Argon2 hashed passphrase stretching settings
|
// Argon2 hashed passphrase stretching settings
|
||||||
#define PWHASH_OPSLIMIT 64
|
#define PWHASH_OPSLIMIT 64
|
||||||
#define PWHASH_MEMLIMIT 64 * 1024 * 1024
|
#define PWHASH_MEMLIMIT 64 * 1024 * 1024
|
||||||
#define PWHASH_ALG crypto_pwhash_ALG_ARGON2ID13
|
#define PWHASH_ALG crypto_pwhash_ALG_ARGON2ID13
|
||||||
|
|
||||||
|
|
||||||
// output directory
|
|
||||||
static char *workdir = 0;
|
|
||||||
static size_t workdirlen = 0;
|
|
||||||
|
|
||||||
static int quietflag = 0;
|
static int quietflag = 0;
|
||||||
static int verboseflag = 0;
|
static int verboseflag = 0;
|
||||||
#ifndef PCRE2FILTER
|
#ifndef PCRE2FILTER
|
||||||
|
|
@ -68,19 +53,6 @@ size_t direndpos; // end of dir before .onion within string
|
||||||
size_t printstartpos; // where to start printing from
|
size_t printstartpos; // where to start printing from
|
||||||
size_t printlen; // precalculated, related to printstartpos
|
size_t printlen; // precalculated, related to printstartpos
|
||||||
|
|
||||||
static int yamloutput = 0;
|
|
||||||
static int numwords = 1;
|
|
||||||
static size_t numneedgenerate = 0;
|
|
||||||
|
|
||||||
static pthread_mutex_t keysgenerated_mutex;
|
|
||||||
static volatile size_t keysgenerated = 0;
|
|
||||||
static volatile int endwork = 0;
|
|
||||||
|
|
||||||
#ifdef PASSPHRASE
|
|
||||||
static pthread_mutex_t determseed_mutex;
|
|
||||||
static u8 determseed[SEED_LEN];
|
|
||||||
#endif
|
|
||||||
|
|
||||||
pthread_mutex_t fout_mutex;
|
pthread_mutex_t fout_mutex;
|
||||||
FILE *fout;
|
FILE *fout;
|
||||||
|
|
||||||
|
|
@ -94,32 +66,7 @@ static void termhandler(int sig)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#include "filters.h"
|
|
||||||
|
|
||||||
#ifdef STATISTICS
|
#ifdef STATISTICS
|
||||||
#define ADDNUMSUCCESS ++st->numsuccess.v
|
|
||||||
#else
|
|
||||||
#define ADDNUMSUCCESS do ; while (0)
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// statistics, if enabled
|
|
||||||
#ifdef STATISTICS
|
|
||||||
struct statstruct {
|
|
||||||
union {
|
|
||||||
u32 v;
|
|
||||||
size_t align;
|
|
||||||
} numcalc;
|
|
||||||
union {
|
|
||||||
u32 v;
|
|
||||||
size_t align;
|
|
||||||
} numsuccess;
|
|
||||||
union {
|
|
||||||
u32 v;
|
|
||||||
size_t align;
|
|
||||||
} numrestart;
|
|
||||||
} ;
|
|
||||||
VEC_STRUCT(statsvec,struct statstruct);
|
|
||||||
|
|
||||||
struct tstatstruct {
|
struct tstatstruct {
|
||||||
u64 numcalc;
|
u64 numcalc;
|
||||||
u64 numsuccess;
|
u64 numsuccess;
|
||||||
|
|
@ -131,549 +78,6 @@ struct tstatstruct {
|
||||||
VEC_STRUCT(tstatsvec,struct tstatstruct);
|
VEC_STRUCT(tstatsvec,struct tstatstruct);
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
static void onionready(char *sname,const u8 *secret,const u8 *pubonion)
|
|
||||||
{
|
|
||||||
if (endwork)
|
|
||||||
return;
|
|
||||||
|
|
||||||
if (numneedgenerate) {
|
|
||||||
pthread_mutex_lock(&keysgenerated_mutex);
|
|
||||||
if (keysgenerated >= numneedgenerate) {
|
|
||||||
pthread_mutex_unlock(&keysgenerated_mutex);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
++keysgenerated;
|
|
||||||
if (keysgenerated == numneedgenerate)
|
|
||||||
endwork = 1;
|
|
||||||
pthread_mutex_unlock(&keysgenerated_mutex);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Sanity check that the public key matches the private one.
|
|
||||||
ge_p3 point;
|
|
||||||
u8 testpk[PUBLIC_LEN];
|
|
||||||
ge_scalarmult_base(&point, secret);
|
|
||||||
ge_p3_tobytes(testpk, &point);
|
|
||||||
if (!memcmp(testpk, pubonion, PUBLIC_LEN))
|
|
||||||
abort();
|
|
||||||
|
|
||||||
if (!yamloutput) {
|
|
||||||
if (createdir(sname,1) != 0) {
|
|
||||||
pthread_mutex_lock(&fout_mutex);
|
|
||||||
fprintf(stderr,"ERROR: could not create directory for key output\n");
|
|
||||||
pthread_mutex_unlock(&fout_mutex);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
strcpy(&sname[onionendpos],"/hs_ed25519_secret_key");
|
|
||||||
writetofile(sname,secret,FORMATTED_SECRET_LEN,1);
|
|
||||||
|
|
||||||
strcpy(&sname[onionendpos],"/hs_ed25519_public_key");
|
|
||||||
writetofile(sname,pubonion,FORMATTED_PUBLIC_LEN,0);
|
|
||||||
|
|
||||||
strcpy(&sname[onionendpos],"/hostname");
|
|
||||||
FILE *hfile = fopen(sname,"w");
|
|
||||||
sname[onionendpos] = '\n';
|
|
||||||
if (hfile) {
|
|
||||||
fwrite(&sname[direndpos],ONION_LEN + 1,1,hfile);
|
|
||||||
fclose(hfile);
|
|
||||||
}
|
|
||||||
if (fout) {
|
|
||||||
pthread_mutex_lock(&fout_mutex);
|
|
||||||
fwrite(&sname[printstartpos],printlen,1,fout);
|
|
||||||
fflush(fout);
|
|
||||||
pthread_mutex_unlock(&fout_mutex);
|
|
||||||
}
|
|
||||||
} else
|
|
||||||
yamlout_writekeys(&sname[direndpos],pubonion,secret);
|
|
||||||
}
|
|
||||||
|
|
||||||
union pubonionunion {
|
|
||||||
u8 raw[PKPREFIX_SIZE + PUBLIC_LEN + 32];
|
|
||||||
struct {
|
|
||||||
u64 prefix[4];
|
|
||||||
u64 key[4];
|
|
||||||
u64 hash[4];
|
|
||||||
} i;
|
|
||||||
} ;
|
|
||||||
|
|
||||||
static char *makesname()
|
|
||||||
{
|
|
||||||
char *sname = (char *) malloc(workdirlen + ONION_LEN + 63 + 1);
|
|
||||||
if (!sname)
|
|
||||||
abort();
|
|
||||||
if (workdir)
|
|
||||||
memcpy(sname,workdir,workdirlen);
|
|
||||||
return sname;
|
|
||||||
}
|
|
||||||
|
|
||||||
// little endian inc
|
|
||||||
static void addsk32(u8 *sk)
|
|
||||||
{
|
|
||||||
register unsigned int c = 8;
|
|
||||||
for (size_t i = 0;i < 32;++i) {
|
|
||||||
c = (unsigned int)sk[i] + c; sk[i] = c & 0xFF; c >>= 8;
|
|
||||||
// unsure if needed
|
|
||||||
if (!c) break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 0123 4567 xxxx --3--> 3456 7xxx
|
|
||||||
// 0123 4567 xxxx --1--> 1234 567x
|
|
||||||
static inline void shiftpk(u8 *dst,const u8 *src,size_t sbits)
|
|
||||||
{
|
|
||||||
size_t i,sbytes = sbits / 8;
|
|
||||||
sbits %= 8;
|
|
||||||
for (i = 0;i + sbytes < PUBLIC_LEN;++i) {
|
|
||||||
dst[i] = (u8) ((src[i+sbytes] << sbits) |
|
|
||||||
(src[i+sbytes+1] >> (8 - sbits)));
|
|
||||||
}
|
|
||||||
for(;i < PUBLIC_LEN;++i)
|
|
||||||
dst[i] = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void *dowork(void *task)
|
|
||||||
{
|
|
||||||
union pubonionunion pubonion;
|
|
||||||
u8 * const pk = &pubonion.raw[PKPREFIX_SIZE];
|
|
||||||
u8 secret[SKPREFIX_SIZE + SECRET_LEN];
|
|
||||||
u8 * const sk = &secret[SKPREFIX_SIZE];
|
|
||||||
u8 seed[SEED_LEN];
|
|
||||||
u8 hashsrc[checksumstrlen + PUBLIC_LEN + 1];
|
|
||||||
u8 wpk[PUBLIC_LEN + 1];
|
|
||||||
char *sname;
|
|
||||||
|
|
||||||
size_t i;
|
|
||||||
|
|
||||||
#ifdef STATISTICS
|
|
||||||
struct statstruct *st = (struct statstruct *)task;
|
|
||||||
#endif
|
|
||||||
PREFILTER
|
|
||||||
|
|
||||||
memcpy(secret,skprefix,SKPREFIX_SIZE);
|
|
||||||
wpk[PUBLIC_LEN] = 0;
|
|
||||||
memset(&pubonion,0,sizeof(pubonion));
|
|
||||||
memcpy(pubonion.raw,pkprefix,PKPREFIX_SIZE);
|
|
||||||
// write version later as it will be overwritten by hash
|
|
||||||
memcpy(hashsrc,checksumstr,checksumstrlen);
|
|
||||||
hashsrc[checksumstrlen + PUBLIC_LEN] = 0x03; // version
|
|
||||||
|
|
||||||
sname = makesname();
|
|
||||||
|
|
||||||
initseed:
|
|
||||||
randombytes(seed,sizeof(seed));
|
|
||||||
ed25519_seckey_expand(sk,seed);
|
|
||||||
#ifdef STATISTICS
|
|
||||||
++st->numrestart.v;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
again:
|
|
||||||
if (unlikely(endwork))
|
|
||||||
goto end;
|
|
||||||
|
|
||||||
ed25519_pubkey(pk,sk);
|
|
||||||
|
|
||||||
#ifdef STATISTICS
|
|
||||||
++st->numcalc.v;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
DOFILTER(i,pk,{
|
|
||||||
if (numwords > 1) {
|
|
||||||
shiftpk(wpk,pk,filter_len(i));
|
|
||||||
size_t j;
|
|
||||||
for (int w = 1;;) {
|
|
||||||
DOFILTER(j,wpk,goto secondfind);
|
|
||||||
goto next;
|
|
||||||
secondfind:
|
|
||||||
if (++w >= numwords)
|
|
||||||
break;
|
|
||||||
shiftpk(wpk,wpk,filter_len(j));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// sanity check
|
|
||||||
if ((sk[0] & 248) != sk[0] || ((sk[31] & 63) | 64) != sk[31])
|
|
||||||
goto initseed;
|
|
||||||
|
|
||||||
ADDNUMSUCCESS;
|
|
||||||
|
|
||||||
// calc checksum
|
|
||||||
memcpy(&hashsrc[checksumstrlen],pk,PUBLIC_LEN);
|
|
||||||
FIPS202_SHA3_256(hashsrc,sizeof(hashsrc),&pk[PUBLIC_LEN]);
|
|
||||||
// version byte
|
|
||||||
pk[PUBLIC_LEN + 2] = 0x03;
|
|
||||||
// base32
|
|
||||||
strcpy(base32_to(&sname[direndpos],pk,PUBONION_LEN),".onion");
|
|
||||||
onionready(sname,secret,pubonion.raw);
|
|
||||||
pk[PUBLIC_LEN] = 0; // what is this for?
|
|
||||||
goto initseed;
|
|
||||||
});
|
|
||||||
next:
|
|
||||||
addsk32(sk);
|
|
||||||
goto again;
|
|
||||||
|
|
||||||
end:
|
|
||||||
free(sname);
|
|
||||||
POSTFILTER
|
|
||||||
sodium_memzero(secret,sizeof(secret));
|
|
||||||
sodium_memzero(seed,sizeof(seed));
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
// in little-endian order, 32 bytes aka 256 bits
|
|
||||||
static void addsztoscalar32(u8 *dst,size_t v)
|
|
||||||
{
|
|
||||||
int i;
|
|
||||||
u32 c = 0;
|
|
||||||
for (i = 0;i < 32;++i) {
|
|
||||||
c += *dst + (v & 0xFF); *dst = c & 0xFF; c >>= 8;
|
|
||||||
v >>= 8;
|
|
||||||
++dst;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void *dofastwork(void *task)
|
|
||||||
{
|
|
||||||
union pubonionunion pubonion;
|
|
||||||
u8 * const pk = &pubonion.raw[PKPREFIX_SIZE];
|
|
||||||
u8 secret[SKPREFIX_SIZE + SECRET_LEN];
|
|
||||||
u8 * const sk = &secret[SKPREFIX_SIZE];
|
|
||||||
u8 seed[SEED_LEN];
|
|
||||||
u8 hashsrc[checksumstrlen + PUBLIC_LEN + 1];
|
|
||||||
u8 wpk[PUBLIC_LEN + 1];
|
|
||||||
ge_p3 ge_public;
|
|
||||||
char *sname;
|
|
||||||
|
|
||||||
size_t counter;
|
|
||||||
size_t i;
|
|
||||||
|
|
||||||
#ifdef STATISTICS
|
|
||||||
struct statstruct *st = (struct statstruct *)task;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
PREFILTER
|
|
||||||
|
|
||||||
memcpy(secret,skprefix,SKPREFIX_SIZE);
|
|
||||||
wpk[PUBLIC_LEN] = 0;
|
|
||||||
memset(&pubonion,0,sizeof(pubonion));
|
|
||||||
memcpy(pubonion.raw,pkprefix,PKPREFIX_SIZE);
|
|
||||||
// write version later as it will be overwritten by hash
|
|
||||||
memcpy(hashsrc,checksumstr,checksumstrlen);
|
|
||||||
hashsrc[checksumstrlen + PUBLIC_LEN] = 0x03; // version
|
|
||||||
|
|
||||||
sname = makesname();
|
|
||||||
|
|
||||||
initseed:
|
|
||||||
#ifdef STATISTICS
|
|
||||||
++st->numrestart.v;
|
|
||||||
#endif
|
|
||||||
randombytes(seed,sizeof(seed));
|
|
||||||
ed25519_seckey_expand(sk,seed);
|
|
||||||
|
|
||||||
ge_scalarmult_base(&ge_public,sk);
|
|
||||||
ge_p3_tobytes(pk,&ge_public);
|
|
||||||
|
|
||||||
for (counter = 0;counter < SIZE_MAX-8;counter += 8) {
|
|
||||||
ge_p1p1 sum;
|
|
||||||
|
|
||||||
if (unlikely(endwork))
|
|
||||||
goto end;
|
|
||||||
|
|
||||||
DOFILTER(i,pk,{
|
|
||||||
if (numwords > 1) {
|
|
||||||
shiftpk(wpk,pk,filter_len(i));
|
|
||||||
size_t j;
|
|
||||||
for (int w = 1;;) {
|
|
||||||
DOFILTER(j,wpk,goto secondfind);
|
|
||||||
goto next;
|
|
||||||
secondfind:
|
|
||||||
if (++w >= numwords)
|
|
||||||
break;
|
|
||||||
shiftpk(wpk,wpk,filter_len(j));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// found!
|
|
||||||
// update secret key with counter
|
|
||||||
addsztoscalar32(sk,counter);
|
|
||||||
// sanity check
|
|
||||||
if ((sk[0] & 248) != sk[0] || ((sk[31] & 63) | 64) != sk[31])
|
|
||||||
goto initseed;
|
|
||||||
|
|
||||||
ADDNUMSUCCESS;
|
|
||||||
|
|
||||||
// calc checksum
|
|
||||||
memcpy(&hashsrc[checksumstrlen],pk,PUBLIC_LEN);
|
|
||||||
FIPS202_SHA3_256(hashsrc,sizeof(hashsrc),&pk[PUBLIC_LEN]);
|
|
||||||
// version byte
|
|
||||||
pk[PUBLIC_LEN + 2] = 0x03;
|
|
||||||
// full name
|
|
||||||
strcpy(base32_to(&sname[direndpos],pk,PUBONION_LEN),".onion");
|
|
||||||
onionready(sname,secret,pubonion.raw);
|
|
||||||
pk[PUBLIC_LEN] = 0; // what is this for?
|
|
||||||
// don't reuse same seed
|
|
||||||
goto initseed;
|
|
||||||
});
|
|
||||||
next:
|
|
||||||
ge_add(&sum,&ge_public,&ge_eightpoint);
|
|
||||||
ge_p1p1_to_p3(&ge_public,&sum);
|
|
||||||
ge_p3_tobytes(pk,&ge_public);
|
|
||||||
#ifdef STATISTICS
|
|
||||||
++st->numcalc.v;
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
goto initseed;
|
|
||||||
|
|
||||||
end:
|
|
||||||
free(sname);
|
|
||||||
POSTFILTER
|
|
||||||
sodium_memzero(secret,sizeof(secret));
|
|
||||||
sodium_memzero(seed,sizeof(seed));
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
#ifdef PASSPHRASE
|
|
||||||
static void reseedright(u8 sk[SECRET_LEN])
|
|
||||||
{
|
|
||||||
crypto_hash_sha256_state state;
|
|
||||||
crypto_hash_sha256_init(&state);
|
|
||||||
// old right side
|
|
||||||
crypto_hash_sha256_update(&state,&sk[32],32);
|
|
||||||
// new random data
|
|
||||||
randombytes(&sk[32],32);
|
|
||||||
crypto_hash_sha256_update(&state,&sk[32],32);
|
|
||||||
// put result in right side
|
|
||||||
crypto_hash_sha256_final(&state,&sk[32]);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void *dofastworkdeterministic(void *task)
|
|
||||||
{
|
|
||||||
union pubonionunion pubonion;
|
|
||||||
u8 * const pk = &pubonion.raw[PKPREFIX_SIZE];
|
|
||||||
u8 secret[SKPREFIX_SIZE + SECRET_LEN];
|
|
||||||
u8 * const sk = &secret[SKPREFIX_SIZE];
|
|
||||||
u8 seed[SEED_LEN];
|
|
||||||
u8 hashsrc[checksumstrlen + PUBLIC_LEN + 1];
|
|
||||||
u8 wpk[PUBLIC_LEN + 1];
|
|
||||||
ge_p3 ge_public;
|
|
||||||
char *sname;
|
|
||||||
|
|
||||||
size_t counter,oldcounter;
|
|
||||||
size_t i;
|
|
||||||
|
|
||||||
#ifdef STATISTICS
|
|
||||||
struct statstruct *st = (struct statstruct *)task;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
PREFILTER
|
|
||||||
|
|
||||||
memcpy(secret,skprefix,SKPREFIX_SIZE);
|
|
||||||
wpk[PUBLIC_LEN] = 0;
|
|
||||||
memset(&pubonion,0,sizeof(pubonion));
|
|
||||||
memcpy(pubonion.raw,pkprefix,PKPREFIX_SIZE);
|
|
||||||
// write version later as it will be overwritten by hash
|
|
||||||
memcpy(hashsrc,checksumstr,checksumstrlen);
|
|
||||||
hashsrc[checksumstrlen + PUBLIC_LEN] = 0x03; // version
|
|
||||||
|
|
||||||
sname = makesname();
|
|
||||||
|
|
||||||
initseed:
|
|
||||||
pthread_mutex_lock(&determseed_mutex);
|
|
||||||
for (int i = 0; i < SEED_LEN; i++)
|
|
||||||
if (++determseed[i])
|
|
||||||
break;
|
|
||||||
memcpy(seed, determseed, SEED_LEN);
|
|
||||||
pthread_mutex_unlock(&determseed_mutex);
|
|
||||||
ed25519_seckey_expand(sk,seed);
|
|
||||||
|
|
||||||
#ifdef STATISTICS
|
|
||||||
++st->numrestart.v;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
ge_scalarmult_base(&ge_public,sk);
|
|
||||||
ge_p3_tobytes(pk,&ge_public);
|
|
||||||
|
|
||||||
for (counter = oldcounter = 0;counter < DETERMINISTIC_LOOP_COUNT;counter += 8) {
|
|
||||||
ge_p1p1 sum;
|
|
||||||
|
|
||||||
if (unlikely(endwork))
|
|
||||||
goto end;
|
|
||||||
|
|
||||||
DOFILTER(i,pk,{
|
|
||||||
if (numwords > 1) {
|
|
||||||
shiftpk(wpk,pk,filter_len(i));
|
|
||||||
size_t j;
|
|
||||||
for (int w = 1;;) {
|
|
||||||
DOFILTER(j,wpk,goto secondfind);
|
|
||||||
goto next;
|
|
||||||
secondfind:
|
|
||||||
if (++w >= numwords)
|
|
||||||
break;
|
|
||||||
shiftpk(wpk,wpk,filter_len(j));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// found!
|
|
||||||
// update secret key with delta since last hit (if any)
|
|
||||||
addsztoscalar32(sk,counter-oldcounter);
|
|
||||||
oldcounter = counter;
|
|
||||||
// sanity check
|
|
||||||
if ((sk[0] & 248) != sk[0] || ((sk[31] & 63) | 64) != sk[31])
|
|
||||||
goto initseed;
|
|
||||||
|
|
||||||
// reseed right half of key to avoid reuse, it won't change public key anyway
|
|
||||||
reseedright(sk);
|
|
||||||
|
|
||||||
ADDNUMSUCCESS;
|
|
||||||
|
|
||||||
// calc checksum
|
|
||||||
memcpy(&hashsrc[checksumstrlen],pk,PUBLIC_LEN);
|
|
||||||
FIPS202_SHA3_256(hashsrc,sizeof(hashsrc),&pk[PUBLIC_LEN]);
|
|
||||||
// version byte
|
|
||||||
pk[PUBLIC_LEN + 2] = 0x03;
|
|
||||||
// full name
|
|
||||||
strcpy(base32_to(&sname[direndpos],pk,PUBONION_LEN),".onion");
|
|
||||||
onionready(sname,secret,pubonion.raw);
|
|
||||||
pk[PUBLIC_LEN] = 0; // what is this for?
|
|
||||||
});
|
|
||||||
next:
|
|
||||||
ge_add(&sum, &ge_public,&ge_eightpoint);
|
|
||||||
ge_p1p1_to_p3(&ge_public,&sum);
|
|
||||||
ge_p3_tobytes(pk,&ge_public);
|
|
||||||
#ifdef STATISTICS
|
|
||||||
++st->numcalc.v;
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
goto initseed;
|
|
||||||
|
|
||||||
end:
|
|
||||||
free(sname);
|
|
||||||
POSTFILTER
|
|
||||||
sodium_memzero(secret,sizeof(secret));
|
|
||||||
sodium_memzero(seed,sizeof(seed));
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
#endif // PASSPHRASE
|
|
||||||
|
|
||||||
#ifndef BATCHNUM
|
|
||||||
#define BATCHNUM 2048
|
|
||||||
#endif
|
|
||||||
|
|
||||||
static void *dobatchwork(void *task)
|
|
||||||
{
|
|
||||||
union pubonionunion pubonion;
|
|
||||||
u8 * const pk = &pubonion.raw[PKPREFIX_SIZE];
|
|
||||||
u8 secret[SKPREFIX_SIZE + SECRET_LEN];
|
|
||||||
u8 * const sk = &secret[SKPREFIX_SIZE];
|
|
||||||
u8 seed[SEED_LEN];
|
|
||||||
u8 hashsrc[checksumstrlen + PUBLIC_LEN + 1];
|
|
||||||
u8 wpk[PUBLIC_LEN + 1];
|
|
||||||
ge_p3 ge_public;
|
|
||||||
char *sname;
|
|
||||||
|
|
||||||
ge_p3 ge_batch[BATCHNUM];
|
|
||||||
fe *(batchgez)[BATCHNUM];
|
|
||||||
fe tmp_batch[BATCHNUM];
|
|
||||||
bytes32 pk_batch[BATCHNUM];
|
|
||||||
|
|
||||||
size_t counter;
|
|
||||||
size_t i;
|
|
||||||
|
|
||||||
#ifdef STATISTICS
|
|
||||||
struct statstruct *st = (struct statstruct *)task;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
for (size_t b = 0;b < BATCHNUM;++b)
|
|
||||||
batchgez[b] = &GEZ(ge_batch[b]);
|
|
||||||
|
|
||||||
PREFILTER
|
|
||||||
|
|
||||||
memcpy(secret,skprefix,SKPREFIX_SIZE);
|
|
||||||
wpk[PUBLIC_LEN] = 0;
|
|
||||||
memset(&pubonion,0,sizeof(pubonion));
|
|
||||||
memcpy(pubonion.raw,pkprefix,PKPREFIX_SIZE);
|
|
||||||
// write version later as it will be overwritten by hash
|
|
||||||
memcpy(hashsrc,checksumstr,checksumstrlen);
|
|
||||||
hashsrc[checksumstrlen + PUBLIC_LEN] = 0x03; // version
|
|
||||||
|
|
||||||
sname = makesname();
|
|
||||||
|
|
||||||
initseed:
|
|
||||||
#ifdef STATISTICS
|
|
||||||
++st->numrestart.v;
|
|
||||||
#endif
|
|
||||||
randombytes(seed,sizeof(seed));
|
|
||||||
ed25519_seckey_expand(sk,seed);
|
|
||||||
|
|
||||||
ge_scalarmult_base(&ge_public,sk);
|
|
||||||
|
|
||||||
for (counter = 0;counter < SIZE_MAX-(8*BATCHNUM);counter += 8*BATCHNUM) {
|
|
||||||
ge_p1p1 sum;
|
|
||||||
|
|
||||||
if (unlikely(endwork))
|
|
||||||
goto end;
|
|
||||||
|
|
||||||
for (size_t b = 0;b < BATCHNUM;++b) {
|
|
||||||
ge_batch[b] = ge_public;
|
|
||||||
ge_add(&sum,&ge_public,&ge_eightpoint);
|
|
||||||
ge_p1p1_to_p3(&ge_public,&sum);
|
|
||||||
}
|
|
||||||
// NOTE: leaves unfinished
|
|
||||||
ge_p3_batchtobytes_destructive_1(pk_batch,ge_batch,batchgez,tmp_batch,BATCHNUM);
|
|
||||||
|
|
||||||
#ifdef STATISTICS
|
|
||||||
st->numcalc.v += BATCHNUM;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
for (size_t b = 0;b < BATCHNUM;++b) {
|
|
||||||
DOFILTER(i,pk_batch[b],{
|
|
||||||
if (numwords > 1) {
|
|
||||||
shiftpk(wpk,pk_batch[b],filter_len(i));
|
|
||||||
size_t j;
|
|
||||||
for (int w = 1;;) {
|
|
||||||
DOFILTER(j,wpk,goto secondfind);
|
|
||||||
goto next;
|
|
||||||
secondfind:
|
|
||||||
if (++w >= numwords)
|
|
||||||
break;
|
|
||||||
shiftpk(wpk,wpk,filter_len(j));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// found!
|
|
||||||
// finish it up
|
|
||||||
ge_p3_batchtobytes_destructive_finish(pk_batch[b],&ge_batch[b]);
|
|
||||||
// copy public key
|
|
||||||
memcpy(pk,pk_batch[b],PUBLIC_LEN);
|
|
||||||
// update secret key with counter
|
|
||||||
addsztoscalar32(sk,counter + (b * 8));
|
|
||||||
// sanity check
|
|
||||||
if ((sk[0] & 248) != sk[0] || ((sk[31] & 63) | 64) != sk[31])
|
|
||||||
goto initseed;
|
|
||||||
|
|
||||||
ADDNUMSUCCESS;
|
|
||||||
|
|
||||||
// calc checksum
|
|
||||||
memcpy(&hashsrc[checksumstrlen],pk,PUBLIC_LEN);
|
|
||||||
FIPS202_SHA3_256(hashsrc,sizeof(hashsrc),&pk[PUBLIC_LEN]);
|
|
||||||
// version byte
|
|
||||||
pk[PUBLIC_LEN + 2] = 0x03;
|
|
||||||
// full name
|
|
||||||
strcpy(base32_to(&sname[direndpos],pk,PUBONION_LEN),".onion");
|
|
||||||
onionready(sname,secret,pubonion.raw);
|
|
||||||
pk[PUBLIC_LEN] = 0; // what is this for?
|
|
||||||
// don't reuse same seed
|
|
||||||
goto initseed;
|
|
||||||
});
|
|
||||||
next:
|
|
||||||
;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
goto initseed;
|
|
||||||
|
|
||||||
end:
|
|
||||||
free(sname);
|
|
||||||
POSTFILTER
|
|
||||||
sodium_memzero(secret,sizeof(secret));
|
|
||||||
sodium_memzero(seed,sizeof(seed));
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void printhelp(FILE *out,const char *progname)
|
static void printhelp(FILE *out,const char *progname)
|
||||||
{
|
{
|
||||||
fprintf(out,
|
fprintf(out,
|
||||||
|
|
@ -770,6 +174,8 @@ static void setpassphrase(const char *pass)
|
||||||
|
|
||||||
VEC_STRUCT(threadvec, pthread_t);
|
VEC_STRUCT(threadvec, pthread_t);
|
||||||
|
|
||||||
|
#include "filters_main.inc.h"
|
||||||
|
|
||||||
int main(int argc,char **argv)
|
int main(int argc,char **argv)
|
||||||
{
|
{
|
||||||
const char *outfile = 0;
|
const char *outfile = 0;
|
||||||
|
|
@ -799,7 +205,7 @@ int main(int argc,char **argv)
|
||||||
fprintf(stderr,"sodium_init() failed\n");
|
fprintf(stderr,"sodium_init() failed\n");
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
ge_initeightpoint();
|
worker_init();
|
||||||
filters_init();
|
filters_init();
|
||||||
|
|
||||||
setvbuf(stderr,0,_IONBF,0);
|
setvbuf(stderr,0,_IONBF,0);
|
||||||
|
|
@ -1110,10 +516,10 @@ int main(int argc,char **argv)
|
||||||
#endif
|
#endif
|
||||||
tret = pthread_create(&VEC_BUF(threads,i),0,
|
tret = pthread_create(&VEC_BUF(threads,i),0,
|
||||||
#ifdef PASSPHRASE
|
#ifdef PASSPHRASE
|
||||||
deterministic ? dofastworkdeterministic :
|
deterministic ? worker_fast_pass :
|
||||||
#endif
|
#endif
|
||||||
batchkeygen ? dobatchwork :
|
batchkeygen ? worker_batch :
|
||||||
(fastkeygen ? dofastwork : dowork),tp);
|
(fastkeygen ? worker_fast : worker_slow),tp);
|
||||||
if (tret) {
|
if (tret) {
|
||||||
fprintf(stderr,"error while making " FSZ "th thread: %s\n",i,strerror(tret));
|
fprintf(stderr,"error while making " FSZ "th thread: %s\n",i,strerror(tret));
|
||||||
exit(1);
|
exit(1);
|
||||||
|
|
|
||||||
219
worker.c
Normal file
219
worker.c
Normal file
|
|
@ -0,0 +1,219 @@
|
||||||
|
#ifdef __linux__
|
||||||
|
#define _POSIX_C_SOURCE 200112L
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <stdint.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <time.h>
|
||||||
|
#include <pthread.h>
|
||||||
|
#include <sodium/randombytes.h>
|
||||||
|
#ifdef PASSPHRASE
|
||||||
|
#include <sodium/crypto_hash_sha256.h>
|
||||||
|
#endif
|
||||||
|
#include <sodium/utils.h>
|
||||||
|
|
||||||
|
#include "types.h"
|
||||||
|
#include "likely.h"
|
||||||
|
#include "vec.h"
|
||||||
|
#include "base32.h"
|
||||||
|
#include "keccak.h"
|
||||||
|
#include "ed25519/ed25519.h"
|
||||||
|
#include "ioutil.h"
|
||||||
|
#include "common.h"
|
||||||
|
#include "yaml.h"
|
||||||
|
|
||||||
|
#include "worker.h"
|
||||||
|
|
||||||
|
#include "filters.h"
|
||||||
|
|
||||||
|
#ifndef _WIN32
|
||||||
|
#define FSZ "%zu"
|
||||||
|
#else
|
||||||
|
#define FSZ "%Iu"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// additional 0 terminator is added by C
|
||||||
|
static const char * const pkprefix = "== ed25519v1-public: type0 ==\0\0";
|
||||||
|
static const char * const skprefix = "== ed25519v1-secret: type0 ==\0\0";
|
||||||
|
|
||||||
|
static const char checksumstr[] = ".onion checksum";
|
||||||
|
#define checksumstrlen (sizeof(checksumstr) - 1) // 15
|
||||||
|
|
||||||
|
pthread_mutex_t keysgenerated_mutex;
|
||||||
|
volatile size_t keysgenerated = 0;
|
||||||
|
volatile int endwork = 0;
|
||||||
|
|
||||||
|
int yamloutput = 0;
|
||||||
|
int numwords = 1;
|
||||||
|
size_t numneedgenerate = 0;
|
||||||
|
|
||||||
|
// output directory
|
||||||
|
char *workdir = 0;
|
||||||
|
size_t workdirlen = 0;
|
||||||
|
|
||||||
|
void worker_init(void)
|
||||||
|
{
|
||||||
|
ge_initeightpoint();
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef PASSPHRASE
|
||||||
|
// How many times we loop before a reseed
|
||||||
|
#define DETERMINISTIC_LOOP_COUNT 1<<24
|
||||||
|
|
||||||
|
pthread_mutex_t determseed_mutex;
|
||||||
|
u8 determseed[SEED_LEN];
|
||||||
|
#endif
|
||||||
|
|
||||||
|
|
||||||
|
char *makesname(void)
|
||||||
|
{
|
||||||
|
char *sname = (char *) malloc(workdirlen + ONION_LEN + 63 + 1);
|
||||||
|
if (!sname)
|
||||||
|
abort();
|
||||||
|
if (workdir)
|
||||||
|
memcpy(sname,workdir,workdirlen);
|
||||||
|
return sname;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void onionready(char *sname,const u8 *secret,const u8 *pubonion)
|
||||||
|
{
|
||||||
|
if (endwork)
|
||||||
|
return;
|
||||||
|
|
||||||
|
if (numneedgenerate) {
|
||||||
|
pthread_mutex_lock(&keysgenerated_mutex);
|
||||||
|
if (keysgenerated >= numneedgenerate) {
|
||||||
|
pthread_mutex_unlock(&keysgenerated_mutex);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
++keysgenerated;
|
||||||
|
if (keysgenerated == numneedgenerate)
|
||||||
|
endwork = 1;
|
||||||
|
pthread_mutex_unlock(&keysgenerated_mutex);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sanity check that the public key matches the private one.
|
||||||
|
ge_p3 point;
|
||||||
|
u8 testpk[PUBLIC_LEN];
|
||||||
|
ge_scalarmult_base(&point, secret);
|
||||||
|
ge_p3_tobytes(testpk, &point);
|
||||||
|
if (!memcmp(testpk, pubonion, PUBLIC_LEN))
|
||||||
|
abort();
|
||||||
|
|
||||||
|
if (!yamloutput) {
|
||||||
|
if (createdir(sname,1) != 0) {
|
||||||
|
pthread_mutex_lock(&fout_mutex);
|
||||||
|
fprintf(stderr,"ERROR: could not create directory for key output\n");
|
||||||
|
pthread_mutex_unlock(&fout_mutex);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
strcpy(&sname[onionendpos],"/hs_ed25519_secret_key");
|
||||||
|
writetofile(sname,secret,FORMATTED_SECRET_LEN,1);
|
||||||
|
|
||||||
|
strcpy(&sname[onionendpos],"/hs_ed25519_public_key");
|
||||||
|
writetofile(sname,pubonion,FORMATTED_PUBLIC_LEN,0);
|
||||||
|
|
||||||
|
strcpy(&sname[onionendpos],"/hostname");
|
||||||
|
FILE *hfile = fopen(sname,"w");
|
||||||
|
sname[onionendpos] = '\n';
|
||||||
|
if (hfile) {
|
||||||
|
fwrite(&sname[direndpos],ONION_LEN + 1,1,hfile);
|
||||||
|
fclose(hfile);
|
||||||
|
}
|
||||||
|
if (fout) {
|
||||||
|
pthread_mutex_lock(&fout_mutex);
|
||||||
|
fwrite(&sname[printstartpos],printlen,1,fout);
|
||||||
|
fflush(fout);
|
||||||
|
pthread_mutex_unlock(&fout_mutex);
|
||||||
|
}
|
||||||
|
} else
|
||||||
|
yamlout_writekeys(&sname[direndpos],pubonion,secret);
|
||||||
|
}
|
||||||
|
|
||||||
|
#include "filters_worker.inc.h"
|
||||||
|
|
||||||
|
#ifdef STATISTICS
|
||||||
|
#define ADDNUMSUCCESS ++st->numsuccess.v
|
||||||
|
#else
|
||||||
|
#define ADDNUMSUCCESS do ; while (0)
|
||||||
|
#endif
|
||||||
|
|
||||||
|
|
||||||
|
union pubonionunion {
|
||||||
|
u8 raw[PKPREFIX_SIZE + PUBLIC_LEN + 32];
|
||||||
|
struct {
|
||||||
|
u64 prefix[4];
|
||||||
|
u64 key[4];
|
||||||
|
u64 hash[4];
|
||||||
|
} i;
|
||||||
|
} ;
|
||||||
|
|
||||||
|
// little endian inc
|
||||||
|
static void addsk32(u8 *sk)
|
||||||
|
{
|
||||||
|
register unsigned int c = 8;
|
||||||
|
for (size_t i = 0;i < 32;++i) {
|
||||||
|
c = (unsigned int)sk[i] + c; sk[i] = c & 0xFF; c >>= 8;
|
||||||
|
// unsure if needed
|
||||||
|
if (!c) break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 0123 4567 xxxx --3--> 3456 7xxx
|
||||||
|
// 0123 4567 xxxx --1--> 1234 567x
|
||||||
|
static inline void shiftpk(u8 *dst,const u8 *src,size_t sbits)
|
||||||
|
{
|
||||||
|
size_t i,sbytes = sbits / 8;
|
||||||
|
sbits %= 8;
|
||||||
|
for (i = 0;i + sbytes < PUBLIC_LEN;++i) {
|
||||||
|
dst[i] = (u8) ((src[i+sbytes] << sbits) |
|
||||||
|
(src[i+sbytes+1] >> (8 - sbits)));
|
||||||
|
}
|
||||||
|
for(;i < PUBLIC_LEN;++i)
|
||||||
|
dst[i] = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
#include "worker_slow.inc.h"
|
||||||
|
|
||||||
|
|
||||||
|
// in little-endian order, 32 bytes aka 256 bits
|
||||||
|
static void addsztoscalar32(u8 *dst,size_t v)
|
||||||
|
{
|
||||||
|
int i;
|
||||||
|
u32 c = 0;
|
||||||
|
for (i = 0;i < 32;++i) {
|
||||||
|
c += *dst + (v & 0xFF); *dst = c & 0xFF; c >>= 8;
|
||||||
|
v >>= 8;
|
||||||
|
++dst;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#include "worker_fast.inc.h"
|
||||||
|
|
||||||
|
|
||||||
|
#ifdef PASSPHRASE
|
||||||
|
static void reseedright(u8 sk[SECRET_LEN])
|
||||||
|
{
|
||||||
|
crypto_hash_sha256_state state;
|
||||||
|
crypto_hash_sha256_init(&state);
|
||||||
|
// old right side
|
||||||
|
crypto_hash_sha256_update(&state,&sk[32],32);
|
||||||
|
// new random data
|
||||||
|
randombytes(&sk[32],32);
|
||||||
|
crypto_hash_sha256_update(&state,&sk[32],32);
|
||||||
|
// put result in right side
|
||||||
|
crypto_hash_sha256_final(&state,&sk[32]);
|
||||||
|
}
|
||||||
|
#endif // PASSPHRASE
|
||||||
|
|
||||||
|
#include "worker_fast_pass.inc.h"
|
||||||
|
|
||||||
|
|
||||||
|
#ifndef BATCHNUM
|
||||||
|
#define BATCHNUM 2048
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include "worker_batch.inc.h"
|
||||||
46
worker.h
Normal file
46
worker.h
Normal file
|
|
@ -0,0 +1,46 @@
|
||||||
|
|
||||||
|
extern pthread_mutex_t keysgenerated_mutex;
|
||||||
|
extern volatile size_t keysgenerated;
|
||||||
|
extern volatile int endwork;
|
||||||
|
|
||||||
|
extern int yamloutput;
|
||||||
|
extern int numwords;
|
||||||
|
extern size_t numneedgenerate;
|
||||||
|
|
||||||
|
extern char *workdir;
|
||||||
|
extern size_t workdirlen;
|
||||||
|
|
||||||
|
// statistics, if enabled
|
||||||
|
#ifdef STATISTICS
|
||||||
|
struct statstruct {
|
||||||
|
union {
|
||||||
|
u32 v;
|
||||||
|
size_t align;
|
||||||
|
} numcalc;
|
||||||
|
union {
|
||||||
|
u32 v;
|
||||||
|
size_t align;
|
||||||
|
} numsuccess;
|
||||||
|
union {
|
||||||
|
u32 v;
|
||||||
|
size_t align;
|
||||||
|
} numrestart;
|
||||||
|
} ;
|
||||||
|
VEC_STRUCT(statsvec,struct statstruct);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#ifdef PASSPHRASE
|
||||||
|
extern pthread_mutex_t determseed_mutex;
|
||||||
|
extern u8 determseed[SEED_LEN];
|
||||||
|
#endif
|
||||||
|
|
||||||
|
extern void worker_init(void);
|
||||||
|
|
||||||
|
extern char *makesname(void);
|
||||||
|
|
||||||
|
extern void *worker_slow(void *task);
|
||||||
|
extern void *worker_fast(void *task);
|
||||||
|
extern void *worker_batch(void *task);
|
||||||
|
#ifdef PASSPHRASE
|
||||||
|
extern void *worker_fast_pass(void *task);
|
||||||
|
#endif
|
||||||
119
worker_batch.inc.h
Normal file
119
worker_batch.inc.h
Normal file
|
|
@ -0,0 +1,119 @@
|
||||||
|
|
||||||
|
void *worker_batch(void *task)
|
||||||
|
{
|
||||||
|
union pubonionunion pubonion;
|
||||||
|
u8 * const pk = &pubonion.raw[PKPREFIX_SIZE];
|
||||||
|
u8 secret[SKPREFIX_SIZE + SECRET_LEN];
|
||||||
|
u8 * const sk = &secret[SKPREFIX_SIZE];
|
||||||
|
u8 seed[SEED_LEN];
|
||||||
|
u8 hashsrc[checksumstrlen + PUBLIC_LEN + 1];
|
||||||
|
u8 wpk[PUBLIC_LEN + 1];
|
||||||
|
ge_p3 ge_public;
|
||||||
|
char *sname;
|
||||||
|
|
||||||
|
ge_p3 ge_batch[BATCHNUM];
|
||||||
|
fe *(batchgez)[BATCHNUM];
|
||||||
|
fe tmp_batch[BATCHNUM];
|
||||||
|
bytes32 pk_batch[BATCHNUM];
|
||||||
|
|
||||||
|
size_t counter;
|
||||||
|
size_t i;
|
||||||
|
|
||||||
|
#ifdef STATISTICS
|
||||||
|
struct statstruct *st = (struct statstruct *)task;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
for (size_t b = 0;b < BATCHNUM;++b)
|
||||||
|
batchgez[b] = &GEZ(ge_batch[b]);
|
||||||
|
|
||||||
|
PREFILTER
|
||||||
|
|
||||||
|
memcpy(secret,skprefix,SKPREFIX_SIZE);
|
||||||
|
wpk[PUBLIC_LEN] = 0;
|
||||||
|
memset(&pubonion,0,sizeof(pubonion));
|
||||||
|
memcpy(pubonion.raw,pkprefix,PKPREFIX_SIZE);
|
||||||
|
// write version later as it will be overwritten by hash
|
||||||
|
memcpy(hashsrc,checksumstr,checksumstrlen);
|
||||||
|
hashsrc[checksumstrlen + PUBLIC_LEN] = 0x03; // version
|
||||||
|
|
||||||
|
sname = makesname();
|
||||||
|
|
||||||
|
initseed:
|
||||||
|
#ifdef STATISTICS
|
||||||
|
++st->numrestart.v;
|
||||||
|
#endif
|
||||||
|
randombytes(seed,sizeof(seed));
|
||||||
|
ed25519_seckey_expand(sk,seed);
|
||||||
|
|
||||||
|
ge_scalarmult_base(&ge_public,sk);
|
||||||
|
|
||||||
|
for (counter = 0;counter < SIZE_MAX-(8*BATCHNUM);counter += 8*BATCHNUM) {
|
||||||
|
ge_p1p1 sum;
|
||||||
|
|
||||||
|
if (unlikely(endwork))
|
||||||
|
goto end;
|
||||||
|
|
||||||
|
for (size_t b = 0;b < BATCHNUM;++b) {
|
||||||
|
ge_batch[b] = ge_public;
|
||||||
|
ge_add(&sum,&ge_public,&ge_eightpoint);
|
||||||
|
ge_p1p1_to_p3(&ge_public,&sum);
|
||||||
|
}
|
||||||
|
// NOTE: leaves unfinished
|
||||||
|
ge_p3_batchtobytes_destructive_1(pk_batch,ge_batch,batchgez,tmp_batch,BATCHNUM);
|
||||||
|
|
||||||
|
#ifdef STATISTICS
|
||||||
|
st->numcalc.v += BATCHNUM;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
for (size_t b = 0;b < BATCHNUM;++b) {
|
||||||
|
DOFILTER(i,pk_batch[b],{
|
||||||
|
if (numwords > 1) {
|
||||||
|
shiftpk(wpk,pk_batch[b],filter_len(i));
|
||||||
|
size_t j;
|
||||||
|
for (int w = 1;;) {
|
||||||
|
DOFILTER(j,wpk,goto secondfind);
|
||||||
|
goto next;
|
||||||
|
secondfind:
|
||||||
|
if (++w >= numwords)
|
||||||
|
break;
|
||||||
|
shiftpk(wpk,wpk,filter_len(j));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// found!
|
||||||
|
// finish it up
|
||||||
|
ge_p3_batchtobytes_destructive_finish(pk_batch[b],&ge_batch[b]);
|
||||||
|
// copy public key
|
||||||
|
memcpy(pk,pk_batch[b],PUBLIC_LEN);
|
||||||
|
// update secret key with counter
|
||||||
|
addsztoscalar32(sk,counter + (b * 8));
|
||||||
|
// sanity check
|
||||||
|
if ((sk[0] & 248) != sk[0] || ((sk[31] & 63) | 64) != sk[31])
|
||||||
|
goto initseed;
|
||||||
|
|
||||||
|
ADDNUMSUCCESS;
|
||||||
|
|
||||||
|
// calc checksum
|
||||||
|
memcpy(&hashsrc[checksumstrlen],pk,PUBLIC_LEN);
|
||||||
|
FIPS202_SHA3_256(hashsrc,sizeof(hashsrc),&pk[PUBLIC_LEN]);
|
||||||
|
// version byte
|
||||||
|
pk[PUBLIC_LEN + 2] = 0x03;
|
||||||
|
// full name
|
||||||
|
strcpy(base32_to(&sname[direndpos],pk,PUBONION_LEN),".onion");
|
||||||
|
onionready(sname,secret,pubonion.raw);
|
||||||
|
pk[PUBLIC_LEN] = 0; // what is this for?
|
||||||
|
// don't reuse same seed
|
||||||
|
goto initseed;
|
||||||
|
});
|
||||||
|
next:
|
||||||
|
;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
goto initseed;
|
||||||
|
|
||||||
|
end:
|
||||||
|
free(sname);
|
||||||
|
POSTFILTER
|
||||||
|
sodium_memzero(secret,sizeof(secret));
|
||||||
|
sodium_memzero(seed,sizeof(seed));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
101
worker_fast.inc.h
Normal file
101
worker_fast.inc.h
Normal file
|
|
@ -0,0 +1,101 @@
|
||||||
|
|
||||||
|
void *worker_fast(void *task)
|
||||||
|
{
|
||||||
|
union pubonionunion pubonion;
|
||||||
|
u8 * const pk = &pubonion.raw[PKPREFIX_SIZE];
|
||||||
|
u8 secret[SKPREFIX_SIZE + SECRET_LEN];
|
||||||
|
u8 * const sk = &secret[SKPREFIX_SIZE];
|
||||||
|
u8 seed[SEED_LEN];
|
||||||
|
u8 hashsrc[checksumstrlen + PUBLIC_LEN + 1];
|
||||||
|
u8 wpk[PUBLIC_LEN + 1];
|
||||||
|
ge_p3 ge_public;
|
||||||
|
char *sname;
|
||||||
|
|
||||||
|
size_t counter;
|
||||||
|
size_t i;
|
||||||
|
|
||||||
|
#ifdef STATISTICS
|
||||||
|
struct statstruct *st = (struct statstruct *)task;
|
||||||
|
#else
|
||||||
|
(void) task;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
PREFILTER
|
||||||
|
|
||||||
|
memcpy(secret,skprefix,SKPREFIX_SIZE);
|
||||||
|
wpk[PUBLIC_LEN] = 0;
|
||||||
|
memset(&pubonion,0,sizeof(pubonion));
|
||||||
|
memcpy(pubonion.raw,pkprefix,PKPREFIX_SIZE);
|
||||||
|
// write version later as it will be overwritten by hash
|
||||||
|
memcpy(hashsrc,checksumstr,checksumstrlen);
|
||||||
|
hashsrc[checksumstrlen + PUBLIC_LEN] = 0x03; // version
|
||||||
|
|
||||||
|
sname = makesname();
|
||||||
|
|
||||||
|
initseed:
|
||||||
|
#ifdef STATISTICS
|
||||||
|
++st->numrestart.v;
|
||||||
|
#endif
|
||||||
|
randombytes(seed,sizeof(seed));
|
||||||
|
ed25519_seckey_expand(sk,seed);
|
||||||
|
|
||||||
|
ge_scalarmult_base(&ge_public,sk);
|
||||||
|
ge_p3_tobytes(pk,&ge_public);
|
||||||
|
|
||||||
|
for (counter = 0;counter < SIZE_MAX-8;counter += 8) {
|
||||||
|
ge_p1p1 sum;
|
||||||
|
|
||||||
|
if (unlikely(endwork))
|
||||||
|
goto end;
|
||||||
|
|
||||||
|
DOFILTER(i,pk,{
|
||||||
|
if (numwords > 1) {
|
||||||
|
shiftpk(wpk,pk,filter_len(i));
|
||||||
|
size_t j;
|
||||||
|
for (int w = 1;;) {
|
||||||
|
DOFILTER(j,wpk,goto secondfind);
|
||||||
|
goto next;
|
||||||
|
secondfind:
|
||||||
|
if (++w >= numwords)
|
||||||
|
break;
|
||||||
|
shiftpk(wpk,wpk,filter_len(j));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// found!
|
||||||
|
// update secret key with counter
|
||||||
|
addsztoscalar32(sk,counter);
|
||||||
|
// sanity check
|
||||||
|
if ((sk[0] & 248) != sk[0] || ((sk[31] & 63) | 64) != sk[31])
|
||||||
|
goto initseed;
|
||||||
|
|
||||||
|
ADDNUMSUCCESS;
|
||||||
|
|
||||||
|
// calc checksum
|
||||||
|
memcpy(&hashsrc[checksumstrlen],pk,PUBLIC_LEN);
|
||||||
|
FIPS202_SHA3_256(hashsrc,sizeof(hashsrc),&pk[PUBLIC_LEN]);
|
||||||
|
// version byte
|
||||||
|
pk[PUBLIC_LEN + 2] = 0x03;
|
||||||
|
// full name
|
||||||
|
strcpy(base32_to(&sname[direndpos],pk,PUBONION_LEN),".onion");
|
||||||
|
onionready(sname,secret,pubonion.raw);
|
||||||
|
pk[PUBLIC_LEN] = 0; // what is this for?
|
||||||
|
// don't reuse same seed
|
||||||
|
goto initseed;
|
||||||
|
});
|
||||||
|
next:
|
||||||
|
ge_add(&sum,&ge_public,&ge_eightpoint);
|
||||||
|
ge_p1p1_to_p3(&ge_public,&sum);
|
||||||
|
ge_p3_tobytes(pk,&ge_public);
|
||||||
|
#ifdef STATISTICS
|
||||||
|
++st->numcalc.v;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
goto initseed;
|
||||||
|
|
||||||
|
end:
|
||||||
|
free(sname);
|
||||||
|
POSTFILTER
|
||||||
|
sodium_memzero(secret,sizeof(secret));
|
||||||
|
sodium_memzero(seed,sizeof(seed));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
111
worker_fast_pass.inc.h
Normal file
111
worker_fast_pass.inc.h
Normal file
|
|
@ -0,0 +1,111 @@
|
||||||
|
|
||||||
|
#ifdef PASSPHRASE
|
||||||
|
void *worker_fast_pass(void *task)
|
||||||
|
{
|
||||||
|
union pubonionunion pubonion;
|
||||||
|
u8 * const pk = &pubonion.raw[PKPREFIX_SIZE];
|
||||||
|
u8 secret[SKPREFIX_SIZE + SECRET_LEN];
|
||||||
|
u8 * const sk = &secret[SKPREFIX_SIZE];
|
||||||
|
u8 seed[SEED_LEN];
|
||||||
|
u8 hashsrc[checksumstrlen + PUBLIC_LEN + 1];
|
||||||
|
u8 wpk[PUBLIC_LEN + 1];
|
||||||
|
ge_p3 ge_public;
|
||||||
|
char *sname;
|
||||||
|
|
||||||
|
size_t counter,oldcounter;
|
||||||
|
size_t i;
|
||||||
|
|
||||||
|
#ifdef STATISTICS
|
||||||
|
struct statstruct *st = (struct statstruct *)task;
|
||||||
|
#else
|
||||||
|
(void) task;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
PREFILTER
|
||||||
|
|
||||||
|
memcpy(secret,skprefix,SKPREFIX_SIZE);
|
||||||
|
wpk[PUBLIC_LEN] = 0;
|
||||||
|
memset(&pubonion,0,sizeof(pubonion));
|
||||||
|
memcpy(pubonion.raw,pkprefix,PKPREFIX_SIZE);
|
||||||
|
// write version later as it will be overwritten by hash
|
||||||
|
memcpy(hashsrc,checksumstr,checksumstrlen);
|
||||||
|
hashsrc[checksumstrlen + PUBLIC_LEN] = 0x03; // version
|
||||||
|
|
||||||
|
sname = makesname();
|
||||||
|
|
||||||
|
initseed:
|
||||||
|
pthread_mutex_lock(&determseed_mutex);
|
||||||
|
for (int i = 0; i < SEED_LEN; i++)
|
||||||
|
if (++determseed[i])
|
||||||
|
break;
|
||||||
|
memcpy(seed, determseed, SEED_LEN);
|
||||||
|
pthread_mutex_unlock(&determseed_mutex);
|
||||||
|
ed25519_seckey_expand(sk,seed);
|
||||||
|
|
||||||
|
#ifdef STATISTICS
|
||||||
|
++st->numrestart.v;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
ge_scalarmult_base(&ge_public,sk);
|
||||||
|
ge_p3_tobytes(pk,&ge_public);
|
||||||
|
|
||||||
|
for (counter = oldcounter = 0;counter < DETERMINISTIC_LOOP_COUNT;counter += 8) {
|
||||||
|
ge_p1p1 sum;
|
||||||
|
|
||||||
|
if (unlikely(endwork))
|
||||||
|
goto end;
|
||||||
|
|
||||||
|
DOFILTER(i,pk,{
|
||||||
|
if (numwords > 1) {
|
||||||
|
shiftpk(wpk,pk,filter_len(i));
|
||||||
|
size_t j;
|
||||||
|
for (int w = 1;;) {
|
||||||
|
DOFILTER(j,wpk,goto secondfind);
|
||||||
|
goto next;
|
||||||
|
secondfind:
|
||||||
|
if (++w >= numwords)
|
||||||
|
break;
|
||||||
|
shiftpk(wpk,wpk,filter_len(j));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// found!
|
||||||
|
// update secret key with delta since last hit (if any)
|
||||||
|
addsztoscalar32(sk,counter-oldcounter);
|
||||||
|
oldcounter = counter;
|
||||||
|
// sanity check
|
||||||
|
if ((sk[0] & 248) != sk[0] || ((sk[31] & 63) | 64) != sk[31])
|
||||||
|
goto initseed;
|
||||||
|
|
||||||
|
// reseed right half of key to avoid reuse, it won't change public key anyway
|
||||||
|
reseedright(sk);
|
||||||
|
|
||||||
|
ADDNUMSUCCESS;
|
||||||
|
|
||||||
|
// calc checksum
|
||||||
|
memcpy(&hashsrc[checksumstrlen],pk,PUBLIC_LEN);
|
||||||
|
FIPS202_SHA3_256(hashsrc,sizeof(hashsrc),&pk[PUBLIC_LEN]);
|
||||||
|
// version byte
|
||||||
|
pk[PUBLIC_LEN + 2] = 0x03;
|
||||||
|
// full name
|
||||||
|
strcpy(base32_to(&sname[direndpos],pk,PUBONION_LEN),".onion");
|
||||||
|
onionready(sname,secret,pubonion.raw);
|
||||||
|
pk[PUBLIC_LEN] = 0; // what is this for?
|
||||||
|
});
|
||||||
|
next:
|
||||||
|
ge_add(&sum, &ge_public,&ge_eightpoint);
|
||||||
|
ge_p1p1_to_p3(&ge_public,&sum);
|
||||||
|
ge_p3_tobytes(pk,&ge_public);
|
||||||
|
#ifdef STATISTICS
|
||||||
|
++st->numcalc.v;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
goto initseed;
|
||||||
|
|
||||||
|
end:
|
||||||
|
free(sname);
|
||||||
|
POSTFILTER
|
||||||
|
sodium_memzero(secret,sizeof(secret));
|
||||||
|
sodium_memzero(seed,sizeof(seed));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
#endif // PASSPHRASE
|
||||||
89
worker_slow.inc.h
Normal file
89
worker_slow.inc.h
Normal file
|
|
@ -0,0 +1,89 @@
|
||||||
|
|
||||||
|
void *worker_slow(void *task)
|
||||||
|
{
|
||||||
|
union pubonionunion pubonion;
|
||||||
|
u8 * const pk = &pubonion.raw[PKPREFIX_SIZE];
|
||||||
|
u8 secret[SKPREFIX_SIZE + SECRET_LEN];
|
||||||
|
u8 * const sk = &secret[SKPREFIX_SIZE];
|
||||||
|
u8 seed[SEED_LEN];
|
||||||
|
u8 hashsrc[checksumstrlen + PUBLIC_LEN + 1];
|
||||||
|
u8 wpk[PUBLIC_LEN + 1];
|
||||||
|
char *sname;
|
||||||
|
|
||||||
|
size_t i;
|
||||||
|
|
||||||
|
#ifdef STATISTICS
|
||||||
|
struct statstruct *st = (struct statstruct *)task;
|
||||||
|
#else
|
||||||
|
(void) task;
|
||||||
|
#endif
|
||||||
|
PREFILTER
|
||||||
|
|
||||||
|
memcpy(secret,skprefix,SKPREFIX_SIZE);
|
||||||
|
wpk[PUBLIC_LEN] = 0;
|
||||||
|
memset(&pubonion,0,sizeof(pubonion));
|
||||||
|
memcpy(pubonion.raw,pkprefix,PKPREFIX_SIZE);
|
||||||
|
// write version later as it will be overwritten by hash
|
||||||
|
memcpy(hashsrc,checksumstr,checksumstrlen);
|
||||||
|
hashsrc[checksumstrlen + PUBLIC_LEN] = 0x03; // version
|
||||||
|
|
||||||
|
sname = makesname();
|
||||||
|
|
||||||
|
initseed:
|
||||||
|
randombytes(seed,sizeof(seed));
|
||||||
|
ed25519_seckey_expand(sk,seed);
|
||||||
|
#ifdef STATISTICS
|
||||||
|
++st->numrestart.v;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
again:
|
||||||
|
if (unlikely(endwork))
|
||||||
|
goto end;
|
||||||
|
|
||||||
|
ed25519_pubkey(pk,sk);
|
||||||
|
|
||||||
|
#ifdef STATISTICS
|
||||||
|
++st->numcalc.v;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
DOFILTER(i,pk,{
|
||||||
|
if (numwords > 1) {
|
||||||
|
shiftpk(wpk,pk,filter_len(i));
|
||||||
|
size_t j;
|
||||||
|
for (int w = 1;;) {
|
||||||
|
DOFILTER(j,wpk,goto secondfind);
|
||||||
|
goto next;
|
||||||
|
secondfind:
|
||||||
|
if (++w >= numwords)
|
||||||
|
break;
|
||||||
|
shiftpk(wpk,wpk,filter_len(j));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// sanity check
|
||||||
|
if ((sk[0] & 248) != sk[0] || ((sk[31] & 63) | 64) != sk[31])
|
||||||
|
goto initseed;
|
||||||
|
|
||||||
|
ADDNUMSUCCESS;
|
||||||
|
|
||||||
|
// calc checksum
|
||||||
|
memcpy(&hashsrc[checksumstrlen],pk,PUBLIC_LEN);
|
||||||
|
FIPS202_SHA3_256(hashsrc,sizeof(hashsrc),&pk[PUBLIC_LEN]);
|
||||||
|
// version byte
|
||||||
|
pk[PUBLIC_LEN + 2] = 0x03;
|
||||||
|
// base32
|
||||||
|
strcpy(base32_to(&sname[direndpos],pk,PUBONION_LEN),".onion");
|
||||||
|
onionready(sname,secret,pubonion.raw);
|
||||||
|
pk[PUBLIC_LEN] = 0; // what is this for?
|
||||||
|
goto initseed;
|
||||||
|
});
|
||||||
|
next:
|
||||||
|
addsk32(sk);
|
||||||
|
goto again;
|
||||||
|
|
||||||
|
end:
|
||||||
|
free(sname);
|
||||||
|
POSTFILTER
|
||||||
|
sodium_memzero(secret,sizeof(secret));
|
||||||
|
sodium_memzero(seed,sizeof(seed));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
Loading…
Reference in a new issue