freebsd-skq/utils/VtableTest/gen.cc
2010-01-15 15:39:40 +00:00

351 lines
9.4 KiB
C++

#include <stdio.h>
#include <stdlib.h>
#define N_FIELDS 7
#define N_FUNCS 128
#define FUNCSPACING 20
#define N_STRUCTS 180 /* 1280 */
#define N_BASES 6
#define COVARIANT 0
const char *simple_types[] = { "bool", "char", "short", "int", "float",
"double", "long double", "wchar_t", "void *",
"char *"
};
void gl(const char *c) {
printf("%s\n", c);
}
void g(const char *c) {
printf("%s", c);
}
void g(int i) {
printf("%d", i);
}
int uuid = 0;
char base_present[N_STRUCTS][N_STRUCTS];
// The return type for each function when doing covariant testcase generation.
short ret_types[N_STRUCTS][N_FUNCS*FUNCSPACING];
bool is_ambiguous(int s, int base) {
for (int i = 0; i < N_STRUCTS; ++i) {
if ((base_present[base][i] & base_present[s][i]) == 1)
return true;
}
return false;
}
void add_bases(int s, int base) {
for (int i = 0; i < N_STRUCTS; ++i)
base_present[s][i] |= base_present[base][i];
if (!COVARIANT)
return;
for (int i = 0; i < N_FUNCS*FUNCSPACING; ++i) {
if (!ret_types[base][i])
continue;
if (!ret_types[s][i]) {
ret_types[s][i] = ret_types[base][i];
continue;
}
if (base_present[ret_types[base][i]][ret_types[s][i]])
// If the return type of the function from this base dominates
ret_types[s][i] = ret_types[base][i];
if (base_present[ret_types[s][i]][ret_types[base][i]])
// If a previous base dominates
continue;
// If neither dominates, we'll use this class.
ret_types[s][i] = s;
}
}
// This contains the class that has the final override for
// each class, for each function.
short final_override[N_STRUCTS][N_FUNCS*FUNCSPACING];
void gs(int s) {
bool polymorphic = false;
static int bases[N_BASES];
int i_bases = random() % (N_BASES*2);
if (i_bases >= N_BASES)
// PARAM: 1/2 of all clases should have no bases
i_bases = 0;
int n_bases = 0;
bool first_base = true;
// PARAM: 3/4 of all should be class, the rest are structs
if (random() % 4 == 0)
g("struct s");
else
g("class s");
g(s);
int old_base = -1;
if (s == 0 || s == 1)
i_bases = 0;
while (i_bases) {
--i_bases;
int base = random() % (s-1) + 1;
if (!base_present[s][base]) {
if (is_ambiguous(s, base))
continue;
if (first_base) {
first_base = false;
g(": ");
} else
g(", ");
int base_type = 1;
if (random()%8 == 0) {
// PARAM: 1/8th the bases are virtual
g("virtual ");
// We have a vtable and rtti, but technically we're not polymorphic
// polymorphic = true;
base_type = 3;
}
// PARAM: 1/4 are public, 1/8 are privare, 1/8 are protected, the reset, default
int base_protection = 0;
if (!COVARIANT)
base_protection = random()%8;
switch (base_protection) {
case 0:
case 1:
g("public "); break;
case 2:
case 3:
case 4:
case 5:
break;
case 6:
g("private "); break;
case 7:
g("protected "); break;
}
g("s");
add_bases(s, base);
bases[n_bases] = base;
base_present[s][base] = base_type;
++n_bases;
g(base);
old_base = base;
}
}
gl(" {");
/* Fields */
int n_fields = N_FIELDS == 0 ? 0 : random() % (N_FIELDS*4);
// PARAM: 3/4 of all structs should have no members
if (n_fields >= N_FIELDS)
n_fields = 0;
for (int i = 0; i < n_fields; ++i) {
int t = random() % (sizeof(simple_types) / sizeof(simple_types[0]));
g(" "); g(simple_types[t]); g(" field"); g(i); gl(";");
}
/* Virtual functions */
static int funcs[N_FUNCS*FUNCSPACING];
// PARAM: 1/2 of all structs should have no virtual functions
int n_funcs = random() % (N_FUNCS*2);
if (n_funcs > N_FUNCS)
n_funcs = 0;
int old_func = -1;
for (int i = 0; i < n_funcs; ++i) {
int fn = old_func + random() % FUNCSPACING + 1;
funcs[i] = fn;
int ret_type = 0;
if (COVARIANT) {
ret_type = random() % s + 1;
if (!base_present[s][ret_type]
|| !base_present[ret_type][ret_types[s][fn]])
if (ret_types[s][fn]) {
printf(" // Found one for s%d for s%d* fun%d.\n", s,
ret_types[s][fn], fn);
ret_type = ret_types[s][fn];
} else
ret_type = s;
else
printf(" // Wow found one for s%d for fun%d.\n", s, fn);
ret_types[s][fn] = ret_type;
}
if (ret_type) {
g(" virtual s"); g(ret_type); g("* fun");
} else
g(" virtual void fun");
g(fn); g("(char *t) { mix(\"vfn this offset\", (char *)this - t); mix(\"vfn uuid\", "); g(++uuid);
if (ret_type)
gl("); return 0; }");
else
gl("); }");
final_override[s][fn] = s;
old_func = fn;
}
// Add required overriders for correctness
for (int i = 0; i < n_bases; ++i) {
// For each base
int base = bases[i];
for (int fn = 0; fn < N_FUNCS*FUNCSPACING; ++fn) {
// For each possible function
int new_base = final_override[base][fn];
if (new_base == 0)
// If the base didn't have a final overrider, skip
continue;
int prev_base = final_override[s][fn];
if (prev_base == s)
// Skip functions defined in this class
continue;
// If we don't want to change the info, skip
if (prev_base == new_base)
continue;
if (prev_base == 0) {
// record the final override
final_override[s][fn] = new_base;
continue;
}
if (base_present[prev_base][new_base]) {
// The previous base dominates the new base, no update necessary
printf(" // No override for fun%d in s%d as s%d dominates s%d.\n",
fn, s, prev_base, new_base);
continue;
}
if (base_present[new_base][prev_base]) {
// The new base dominates the old base, no override necessary
printf(" // No override for fun%d in s%d as s%d dominates s%d.\n",
fn, s, new_base, prev_base);
// record the final override
final_override[s][fn] = new_base;
continue;
}
printf(" // Found we needed override for fun%d in s%d.\n", fn, s);
// record the final override
funcs[n_funcs++] = fn;
if (n_funcs == (N_FUNCS*FUNCSPACING-1))
abort();
int ret_type = 0;
if (COVARIANT) {
if (!ret_types[s][fn]) {
ret_types[s][fn] = ret_type = s;
} else {
ret_type = ret_types[s][fn];
if (ret_type != s)
printf(" // Calculated return type in s%d as s%d* fun%d.\n",
s, ret_type, fn);
}
}
if (ret_type) {
g(" virtual s"); g(ret_type); g("* fun");
} else
g(" virtual void fun");
g(fn); g("(char *t) { mix(\"vfn this offset\", (char *)this - t); mix(\"vfn uuid\", "); g(++uuid);
if (ret_type)
gl("); return 0; }");
else
gl("); }");
final_override[s][fn] = s;
}
}
gl("public:");
gl(" void calc(char *t) {");
// mix in the type number
g(" mix(\"type num\", "); g(s); gl(");");
// mix in the size
g(" mix(\"type size\", sizeof (s"); g(s); gl("));");
// mix in the this offset
gl(" mix(\"subobject offset\", (char *)this - t);");
if (n_funcs)
polymorphic = true;
if (polymorphic) {
// mix in offset to the complete object under construction
gl(" mix(\"real top v current top\", t - (char *)dynamic_cast<void*>(this));");
}
/* check base layout and overrides */
for (int i = 0; i < n_bases; ++i) {
g(" calc_s"); g(bases[i]); gl("(t);");
}
if (polymorphic) {
/* check dynamic_cast to each direct base */
for (int i = 0; i < n_bases; ++i) {
g(" if ((char *)dynamic_cast<s"); g(bases[i]); gl("*>(this))");
g(" mix(\"base dyn cast\", t - (char *)dynamic_cast<s"); g(bases[i]); gl("*>(this));");
g(" else mix(\"no dyncast\", "); g(++uuid); gl(");");
}
}
/* check field layout */
for (int i = 0; i < n_fields; ++i) {
g(" mix(\"field offset\", (char *)&field"); g(i); gl(" - (char *)this);");
}
if (n_fields == 0) {
g(" mix(\"no fields\", "); g(++uuid); gl(");");
}
/* check functions */
for (int i = 0; i < n_funcs; ++i) {
g(" fun"); g(funcs[i]); gl("(t);");
}
if (n_funcs == 0) {
g(" mix(\"no funcs\", "); g(++uuid); gl(");");
}
gl(" }");
// default ctor
g(" s"); g(s); g("() ");
first_base = true;
for (int i = 0; i < n_bases; ++i) {
if (first_base) {
g(": ");
first_base = false;
} else
g(", ");
g("s"); g(bases[i]); g("((char *)this)");
}
gl(" { calc((char *)this); }");
g(" ~s"); g(s); gl("() { calc((char *)this); }");
// ctor with this to the complete object
g(" s"); g(s); gl("(char *t) { calc(t); }");
g(" void calc_s"); g(s); gl("(char *t) { calc(t); }");
g("} a"); g(s); gl(";");
}
main(int argc, char **argv) {
unsigned seed = 0;
char state[16];
if (argc > 1)
seed = atol(argv[1]);
initstate(seed, state, sizeof(state));
gl("extern \"C\" int printf(const char *...);");
gl("");
gl("long long sum;");
gl("void mix(const char *desc, long long i) {");
// If this ever becomes too slow, we can remove this after we improve the
// mixing function
gl(" printf(\"%s: %lld\\n\", desc, i);");
gl(" sum += ((sum ^ i) << 3) + (sum<<1) - i;");
gl("}");
gl("");
// PARAM: Randomly size testcases or large testcases?
int n_structs = /* random() % */ N_STRUCTS;
for (int i = 1; i < n_structs; ++i)
gs(i);
gl("int main() {");
gl(" printf(\"%llx\\n\", sum);");
gl("}");
return 0;
}