1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
#!/usr/bin/perl
use lib 'lib';
use lib "$ENV{RQG_HOME}/lib";
use strict;
use GenTest;
use GenTest::Constants;
use GenTest::Generator::FromGrammar;
use GenTest::Executor;
use Getopt::Long;
$| = 1;
my ($gendata, $help, $grammar_file, $mask, $mask_level, $dsn);
my $queries = my $default_queries = 1000;
my $seed = 1;
my @ARGV_saved = @ARGV;
my $opt_result = GetOptions(
'grammar=s' => \$grammar_file,
'queries=i' => \$queries,
'help' => \$help,
'seed=s' => \$seed,
'mask=i' => \$mask,
'mask-level=i' => \$mask_level,
'dsn=s' => \$dsn
);
help() if !$opt_result || $help || not defined $grammar_file;
my $generator = GenTest::Generator::FromGrammar->new(
grammar_file => $grammar_file,
seed => ($seed eq 'time') ? time() : $seed,
mask => $mask,
mask_level => $mask_level
);
return STATUS_ENVIRONMENT_FAILURE if not defined $generator;
my $executor;
if (defined $dsn) {
$executor = GenTest::Executor->newFromDSN($dsn);
exit (STATUS_ENVIRONMENT_FAILURE) if not defined $executor;
}
if (defined $executor) {
my $init_result = $executor->init();
exit ($init_result) if $init_result > STATUS_OK;
}
foreach my $i (1..$queries) {
my $queries = $generator->next([$executor]);
if (
(not defined $queries) ||
($queries->[0] eq '')
) {
say("Grammar produced an empty query. Terminating.");
exit(STATUS_ENVIRONMENT_FAILURE);
}
my $sql = join('; ',@$queries);
print $sql.";\n";
}
exit(0);
sub help {
print <<EOF
$0 - Generate random queries from an SQL grammar and pipe them to STDOUT
--grammar : Grammar file to use for generating the queries (REQUIRED);
--seed : Seed for the pseudo-random generator
--queries : Numer of queries to generate (default $default_queries);
--dsn : The DSN of the database that will be used to resolve rules such as _table , _field
--mask : A seed to a random mask used to mask (reeduce) the grammar.
--mask-level: How many levels deep the mask is applied (default 1)
--help : This help message
EOF
;
exit(1);
}
|