#
use File::Basename;
-my $file = shift;
-my $basename = basename($file, ".h");
+use Data::Dumper;
+
+#
+# Generate parse tree for header file
+#
+my $file = shift;
require smb_interfaces;
my $parser = new smb_interfaces;
$header = $parser->parse($file);
-use Data::Dumper;
-#print Dumper($header);
+#
+# Make second pass over tree to make it easier to process.
+#
+
+sub flatten_structs($) {
+ my $obj = shift;
+ my $s = { %$obj };
-# Create header
+ # Map NAME, STRUCT_NAME and UNION_NAME elements into a more likeable
+ # property.
-open(FILE, ">ejs_${basename}.h");
+ if (defined($obj->{STRUCT_NAME}) or defined($obj->{UNION_NAME})) {
-print FILE "/* header auto-generated by build_smb_interfaces.pl */\n\n";
+ $s->{TYPE_DEFINED} = defined($obj->{STRUCT_NAME}) ? $obj->{STRUCT_NAME}
+ : $obj->{UNION_NAME};
-print FILE "#ifndef _ejs_${basename}_h\n";
-print FILE "#define _ejs_${basename}_h\n\n";
+ delete $s->{STRUCT_NAME};
+ delete $s->{UNION_NAME};
+ }
-sub struct_name($)
-{
- my $obj = shift;
- return defined($obj->{STRUCT_NAME}) ? $obj->{STRUCT_NAME} : $obj->{UNION_NAME};
-}
+ # Create a new list of structure fields with flattened names
-sub prototypes_for($)
-{
- my $obj = shift;
- my $name = struct_name($obj);
+ foreach my $elt (@{$obj->{DATA}}) {
+ foreach my $name (@{$elt->{NAME}}) {
+ my $new_elt = { %$elt };
+ $new_elt->{NAME} = $name;
+# $new_elt->{PARENT} = $s;
+ push(@{$s->{FIELDS}}, flatten_structs($new_elt));
+ }
+ }
+
+ delete $s->{DATA};
- print FILE "NTSTATUS ejs_push_$name(struct ejs_rpc *, struct MprVar *, const char *, const uint32_t *);\n";
- print FILE "NTSTATUS ejs_pull_$name(struct ejs_rpc *, struct MprVar *, const char *, const uint32_t *);\n";
+ return $s;
}
-sub pushpull_for($)
-{
- my $obj = shift;
- my $name = struct_name($obj);
+@newheader = map { flatten_structs($_) } @{$header};
- print FILE "NTSTATUS ejs_push_$name(struct ejs_rpc *ejs, struct MprVar *v, const char *name, const uint32_t *r)\n";
- print FILE "{\n";
+#
+# Generate implementation
+#
- print FILE "\tNDR_CHECK(ejs_push_struct_start(ejs, &v, \"output\"));\n";
+my $basename = basename($file, ".h");
+stat "libcli/gen_raw" || mkdir("libcli/gen_raw") || die("mkdir");
- print FILE "\n\treturn NT_STATUS_OK;\n";
- print FILE "}\n\n";
+open(FILE, ">libcli/gen_raw/ejs_${basename}.c");
- print FILE "NTSTATUS ejs_pull_$name(struct ejs_rpc *ejs, struct MprVar *v, const char *name, const uint32_t *r)\n";
- print FILE "{\n";
+print FILE "/* EJS wrapper functions auto-generated by build_smb_interfaces.pl */\n\n";
- print FILE "\tNDR_CHECK(ejs_pull_struct_start(ejs, &v, \"input\"));\n";
+print FILE "#include \"includes.h\"\n";
+print FILE "#include \"scripting/ejs/smbcalls.h\"\n";
+print FILE "#include \"lib/appweb/ejs/ejs.h\"\n";
+print FILE "#include \"scripting/ejs/ejsrpc.h\"\n"; # TODO: remove this
+print FILE "\n";
- print FILE "\treturn NT_STATUS_OK;\n";
- print FILE "}\n\n";
-}
+sub transfer_element($$$) {
+ my $dir = shift;
+ my $prefix = shift;
+ my $elt = shift;
-foreach my $x (@{$header}) {
+ $type = $elt->{TYPE};
+ $type =~ s/_t$//;
- # Prototypes for top level structures and unions
+ print FILE "\tNDR_CHECK(ejs_${dir}_$type(ejs, v, \"$prefix.$elt->{NAME}\"));\n";
+}
- prototypes_for($x);
+sub transfer_struct($$) {
+ my $dir = shift;
+ my $struct = shift;
- # Prototypes for non-anonymous nested structures and unions
+ foreach my $field (@{$struct->{FIELDS}}) {
+ next if $dir eq "pull" and $field->{NAME} eq "out";
+ next if $dir eq "push" and $field->{NAME} eq "in";
- foreach my $e1 (@{$x->{DATA}}) {
- foreach my $e2 (@{$e1->{DATA}}) {
- if (defined($e2->{STRUCT_NAME}) or defined($e2->{UNION_NAME})) {
- prototypes_for($e2);
+ if ($field->{TYPE} eq "struct") {
+ foreach $subfield (@{$field->{FIELDS}}) {
+ transfer_element($dir, $field->{NAME}, $subfield);
}
+ } else {
+ transfer_element($dir, $struct->{NAME}, $field);
}
}
}
-print FILE "#endif\n";
+# Top level call functions
-close(FILE);
+foreach my $s (@newheader) {
-# Create file
+ if ($s->{TYPE} eq "struct") {
-open(FILE, ">ejs_${basename}.c");
+ # Push/pull top level struct
-print FILE "/* EJS wrapper functions auto-generated by build_smb_interfaces.pl */\n\n";
+ print FILE "NTSTATUS ejs_pull_$s->{TYPE_DEFINED}(struct ejs_rpc *ejs, struct MprVar *v, struct $s->{TYPE_DEFINED} *r)\n";
+ print FILE "{\n";
-# Top level functions
+ transfer_struct("pull", $s);
-foreach my $x (@{$header}) {
- next, if $x->{STRUCT_NAME} eq "";
- print FILE "static int ejs_$x->{STRUCT_NAME}(int eid, int argc, struct MprVar **argv)\n";
- print FILE "{\n";
- print FILE "\tejsSetErrorMsg(eid, \"Not implemented\");\n";
- print FILE "\treturn -1;\n";
- print FILE "}\n\n";
-}
+ print FILE "\n\treturn NT_STATUS_OK;\n";
+ print FILE "}\n\n";
-# Module initialisation
+ print FILE "NTSTATUS ejs_push_$s->{TYPE_DEFINED}(struct ejs_rpc *ejs, struct MprVar *v, const struct $s->{TYPE_DEFINED} *r)\n";
+ print FILE "{\n";
-print FILE "static int ejs_${basename}_init(int eid, int argc, struct MprVar **argv)\n";
-print FILE "{\n";
-print FILE "\tstruct MprVar *obj = mprInitObject(eid, \"${basename}\", argc, argtv);\n\n";
+ transfer_struct("push", $s);
-foreach my $x (@{$header}) {
- next, if $x->{STRUCT_NAME} eq "";
- print FILE "\tmprSetCFunction(obj, \"$x->{STRUCT_NAME}\", ejs_$x->{STRUCT_NAME});\n";
-}
+ print FILE "\n\treturn NT_STATUS_OK;\n";
+ print FILE "}\n\n";
+
+ # Function call
+
+ print FILE "static int ejs_$s->{TYPE_DEFINED}(int eid, int argc, struct MprVar **argv)\n";
+ print FILE "{\n";
+ print FILE "\treturn ejs_raw_call(eid, argc, argv, (ejs_pull_function_t)ejs_pull_$s->{TYPE_DEFINED}, (ejs_push_function_t)ejs_push_$s->{TYPE_DEFINED});\n";
+ print FILE "}\n\n";
+
+ } else {
-print FILE "}\n\n";
+ # Top level union
-print FILE "NTSTATUS ejs_init_${basename}(void)\n";
-print FILE "{\n";
-print FILE "\treturn smbcalls_register_ejs(\"${basename}_init\", ejs_${basename}_init);\n";
-print FILE "}\n";
+ foreach my $arm (@{$s->{FIELDS}}) {
+
+ # Push/pull union arm
+
+ print FILE "NTSTATUS ejs_pull_$s->{TYPE_DEFINED}_$arm->{NAME}(struct ejs_rpc *ejs, struct MprVar *v, union $s->{TYPE_DEFINED} *r)\n";
+ print FILE "{\n";
+
+ transfer_struct("pull", $arm);
+
+ print FILE "\n\treturn NT_STATUS_OK;\n";
+ print FILE "}\n\n";
+
+ print FILE "NTSTATUS ejs_push_$s->{TYPE_DEFINED}_$arm->{NAME}(struct ejs_rpc *ejs, struct MprVar *v, const union $s->{TYPE_DEFINED} *r)\n";
+ print FILE "{\n";
+
+ transfer_struct("push", $arm);
+
+ print FILE "\n\treturn NT_STATUS_OK;\n";
+ print FILE "}\n\n";
+
+ }
+ }
+}
close(FILE);
git.samba.org / samba.git / blobdiff