|  | # Reply server mig-output massager | 
|  | # | 
|  | #   Copyright (C) 1995-2023 Free Software Foundation, Inc. | 
|  | # | 
|  | #   Written by Miles Bader <miles@gnu.ai.mit.edu> | 
|  | # | 
|  | #   This program is free software; you can redistribute it and/or | 
|  | #   modify it under the terms of the GNU General Public License as | 
|  | #   published by the Free Software Foundation; either version 3, or (at | 
|  | #   your option) any later version. | 
|  | # | 
|  | #   This program is distributed in the hope that it will be useful, but | 
|  | #   WITHOUT ANY WARRANTY; without even the implied warranty of | 
|  | #   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU | 
|  | #   General Public License for more details. | 
|  | # | 
|  | #   You should have received a copy of the GNU General Public License | 
|  | #   along with this program.  If not, see <http://www.gnu.org/licenses/>. | 
|  | # | 
|  | # This awk script hacks the output of mig-generated reply server code | 
|  | # so that it allows replies with just the error-code in them (as this is | 
|  | # how mig returns errors). | 
|  | # | 
|  | # It is highly, highly, dependent on the exact format of mig output.  Ick. | 
|  | # | 
|  |  | 
|  | BEGIN { parse_phase = 0; } | 
|  |  | 
|  | /^}/ { parse_phase = 0; } | 
|  |  | 
|  | parse_phase == 0 && /^mig_internal void _X[a-zA-Z0-9_]*_reply/ { | 
|  | # The start of a mig server routine.  Reset everything.  Note that we only | 
|  | # mess with rpcs that have the suffix `_reply'. | 
|  | num_args = 0; | 
|  | num_checks = 0; | 
|  | parse_phase = 1; | 
|  | print; next; | 
|  | } | 
|  |  | 
|  | parse_phase == 1 && /^[\t ]*typedef struct/ { | 
|  | # The first structure in the server routine should describe the arguments | 
|  | parse_phase = 2; | 
|  | print; next; | 
|  | } | 
|  |  | 
|  | parse_phase == 2 { | 
|  | # The message header field in the args structure, which skip. | 
|  | parse_phase = 3; | 
|  | print; next; | 
|  | } | 
|  |  | 
|  | parse_phase == 3 && /} Request/ { | 
|  | # The args structure is over. | 
|  | if (num_args > 1) | 
|  | parse_phase = 5; | 
|  | else | 
|  | # There's no extra args that could screw up the normal mechanism for | 
|  | # error returns, so we don't have to insert any new code. | 
|  | parse_phase = 0; | 
|  | print; next; | 
|  | } | 
|  |  | 
|  | parse_phase == 3 && num_args == 0 { | 
|  | # The type field for an argument. | 
|  | # This won't be accurate in case of unions being used in the Request struct, | 
|  | # but that doesn't matter, as we'll only be looking at arg_type_code_name[0], | 
|  | # which will not be a union type. | 
|  | arg_type_code_name[num_args] = $2; | 
|  | sub (/;$/, "", arg_type_code_name[num_args]) # Get rid of the semi-colon | 
|  | parse_phase = 4; | 
|  | print; next; | 
|  | } | 
|  |  | 
|  | parse_phase == 3 && num_args == 1 { | 
|  | # We've got more than one argument (but we don't care what it is). | 
|  | num_args++; | 
|  | print; next; | 
|  | } | 
|  |  | 
|  | parse_phase == 3 { | 
|  | # We've know everything we need; now just wait for the end of the Request | 
|  | # struct. | 
|  | print; next; | 
|  | } | 
|  |  | 
|  | parse_phase == 4 { | 
|  | # The value field for an argument. | 
|  | # This won't be accurate in case of unions being used in the Request struct, | 
|  | # but that doesn't matter, as we'll only be looking at arg_name[0], which | 
|  | # will not be a union type. | 
|  | arg_name[num_args] = $2; | 
|  | sub (/;$/, "", arg_name[num_args]) # Get rid of the semi-colon | 
|  | num_args++; | 
|  | parse_phase = 3; | 
|  | print; next; | 
|  | } | 
|  |  | 
|  | parse_phase == 5 && /^[ \t]*(auto |static )?const mach_msg_type_t/ { | 
|  | # The type check structure for an argument. | 
|  | arg_check_name[num_checks] = $(NF - 2); | 
|  | num_checks++; | 
|  | print; next; | 
|  | } | 
|  |  | 
|  | parse_phase == 5 && /^[ \t]*mig_external kern_return_t/ { | 
|  | # The declaration of the user server function for this rpc. | 
|  | user_function_name = $3; | 
|  | print; next; | 
|  | } | 
|  |  | 
|  | parse_phase == 5 && /^#if[ \t]TypeCheck/ { | 
|  | # Keep going if we have not yet collected the type check structures. | 
|  | if (num_checks == 0) | 
|  | { | 
|  | print; next; | 
|  | } | 
|  |  | 
|  | # The first args type checking statement; we need to insert our chunk of | 
|  | # code that bypasses all the type checks if this is an error return, after | 
|  | # which we're done until we get to the next function.  Handily, the size | 
|  | # of mig's Reply structure is also the size of the alternate Request | 
|  | # structure that we want to check for. | 
|  | print "\tif (In0P->Head.msgh_size == sizeof (Reply)"; | 
|  | print "\t    && ! (In0P->Head.msgh_bits & MACH_MSGH_BITS_COMPLEX)"; | 
|  | print "\t    && ! BAD_TYPECHECK(&In0P->" arg_type_code_name[0] ", &" arg_check_name[0] ")"; | 
|  | print "\t    && In0P->" arg_name[0] " != 0)"; | 
|  | print "\t  /* Error return, only the error code argument is passed.  */"; | 
|  | print "\t  {"; | 
|  | # Force the function user_function_name into a type that only takes the first | 
|  | # two arguments. | 
|  | # This is possibly bogus, but easier than supplying bogus values for all | 
|  | # the other args (we can't just pass 0 for them, as they might not be scalar). | 
|  | print "\t    void * __error_call = " user_function_name ";"; | 
|  | print "\t    OutP->RetCode = (*(kern_return_t (*)(mach_port_t, kern_return_t)) __error_call) (In0P->Head.msgh_request_port, In0P->" arg_name[0] ");"; | 
|  | print "\t    return;"; | 
|  | print "\t  }"; | 
|  | print ""; | 
|  | parse_phase = 0; | 
|  | print; next; | 
|  | } | 
|  |  | 
|  | { print; } |