source/script/build_smb_interfaces.pl

   1 #!/usr/bin/perl
   2 #
   3 # Create ejs interfaces for structures in a C header file
   4 #
   5
   6 use File::Basename;
   7 use Data::Dumper;
   8
   9 #
  10 # Generate parse tree for header file
  11 #
  12
  13 my $file = shift;
  14 require smb_interfaces;
  15 my $parser = new smb_interfaces;
  16 $header = $parser->parse($file);
  17
  18 #
  19 # Make second pass over tree to make it easier to process.
  20 #
  21
  22 sub flatten_structs($) {
  23   my $obj = shift;
  24   my $s = { %$obj };
  25
  26   # Map NAME, STRUCT_NAME and UNION_NAME elements into a more likeable
  27   # property.
  28
  29   if (defined($obj->{STRUCT_NAME}) or defined($obj->{UNION_NAME})) {
  30
  31     $s->{TYPE_DEFINED} = defined($obj->{STRUCT_NAME}) ? $obj->{STRUCT_NAME}
  32       : $obj->{UNION_NAME};
  33
  34     delete $s->{STRUCT_NAME};
  35     delete $s->{UNION_NAME};
  36   }
  37
  38   # Create a new list of structure fields with flattened names
  39
  40   foreach my $elt (@{$obj->{DATA}}) {
  41     foreach my $name (@{$elt->{NAME}}) {
  42       my $new_elt = { %$elt };
  43       $new_elt->{NAME} = $name;
  44 #      $new_elt->{PARENT} = $s;
  45       push(@{$s->{FIELDS}}, flatten_structs($new_elt));
  46     }
  47   }
  48
  49   delete $s->{DATA};
  50
  51   return $s;
  52 }
  53
  54 @newheader = map { flatten_structs($_) } @{$header};
  55
  56 #
  57 # Generate implementation
  58 #
  59
  60 my $basename = basename($file, ".h");
  61 stat "libcli/gen_raw" || mkdir("libcli/gen_raw") || die("mkdir");
  62
  63 open(FILE, ">libcli/gen_raw/ejs_${basename}.c");
  64
  65 print FILE "/* EJS wrapper functions auto-generated by build_smb_interfaces.pl */\n\n";
  66
  67 print FILE "#include \"includes.h\"\n";
  68 print FILE "#include \"scripting/ejs/smbcalls.h\"\n";
  69 print FILE "#include \"lib/appweb/ejs/ejs.h\"\n";
  70 print FILE "#include \"scripting/ejs/ejsrpc.h\"\n"; # TODO: remove this
  71 print FILE "\n";
  72
  73 sub transfer_element($$$) {
  74   my $dir = shift;
  75   my $prefix = shift;
  76   my $elt = shift;
  77
  78   $type = $elt->{TYPE};
  79   $type =~ s/_t$//;
  80
  81   print FILE "\tNDR_CHECK(ejs_${dir}_$type(ejs, v, \"$prefix.$elt->{NAME}\"));\n";
  82 }
  83
  84 sub transfer_struct($$) {
  85   my $dir = shift;
  86   my $struct = shift;
  87
  88   foreach my $field (@{$struct->{FIELDS}}) {
  89     next if $dir eq "pull" and $field->{NAME} eq "out";
  90     next if $dir eq "push" and $field->{NAME} eq "in";
  91
  92     if ($field->{TYPE} eq "struct") {
  93       foreach $subfield (@{$field->{FIELDS}}) {
  94         transfer_element($dir, $field->{NAME}, $subfield);
  95       }
  96     } else {
  97       transfer_element($dir, $struct->{NAME}, $field);
  98     }
  99   }
 100 }
 101
 102 # Top level call functions
 103
 104 foreach my $s (@newheader) {
 105
 106   if ($s->{TYPE} eq "struct") {
 107
 108     # Push/pull top level struct
 109
 110     print FILE "NTSTATUS ejs_pull_$s->{TYPE_DEFINED}(struct ejs_rpc *ejs, struct MprVar *v, struct $s->{TYPE_DEFINED} *r)\n";
 111     print FILE "{\n";
 112
 113     transfer_struct("pull", $s);
 114
 115     print FILE "\n\treturn NT_STATUS_OK;\n";
 116     print FILE "}\n\n";
 117
 118     print FILE "NTSTATUS ejs_push_$s->{TYPE_DEFINED}(struct ejs_rpc *ejs, struct MprVar *v, const struct $s->{TYPE_DEFINED} *r)\n";
 119     print FILE "{\n";
 120
 121     transfer_struct("push", $s);
 122
 123     print FILE "\n\treturn NT_STATUS_OK;\n";
 124     print FILE "}\n\n";
 125
 126     # Function call
 127
 128     print FILE "static int ejs_$s->{TYPE_DEFINED}(int eid, int argc, struct MprVar **argv)\n";
 129     print FILE "{\n";
 130     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";
 131     print FILE "}\n\n";
 132
 133   } else {
 134
 135     # Top level union
 136
 137     foreach my $arm (@{$s->{FIELDS}}) {
 138
 139       # Push/pull union arm
 140
 141       print FILE "NTSTATUS ejs_pull_$s->{TYPE_DEFINED}_$arm->{NAME}(struct ejs_rpc *ejs, struct MprVar *v, union $s->{TYPE_DEFINED} *r)\n";
 142       print FILE "{\n";
 143
 144       transfer_struct("pull", $arm);
 145
 146       print FILE "\n\treturn NT_STATUS_OK;\n";
 147       print FILE "}\n\n";
 148
 149       print FILE "NTSTATUS ejs_push_$s->{TYPE_DEFINED}_$arm->{NAME}(struct ejs_rpc *ejs, struct MprVar *v, const union $s->{TYPE_DEFINED} *r)\n";
 150       print FILE "{\n";
 151
 152       transfer_struct("push", $arm);
 153
 154       print FILE "\n\treturn NT_STATUS_OK;\n";
 155       print FILE "}\n\n";
 156
 157     }
 158   }
 159 }
 160
 161 close(FILE);