#include "type.h"
#include <mach/message.h>
#include <mach/kern_return.h>
#include "mig_machine.h"
#include "error.h"
#include "alloc.h"
#include "global.h"
#include "routine.h"
#include "write.h"
u_int rtNumber = 0;
static void rtSizeDelta();
routine_t *
rtAlloc(void)
{
register routine_t *new;
new = (routine_t *) calloc(1, sizeof *new);
if (new == rtNULL)
fatal("rtAlloc(): %s", strerror(errno));
new->rtNumber = rtNumber++;
new->rtName = strNULL;
new->rtErrorName = strNULL;
new->rtUserName = strNULL;
new->rtServerName = strNULL;
return new;
}
void
rtSkip(void)
{
rtNumber++;
}
argument_t *
argAlloc(void)
{
extern void KPD_error();
static argument_t prototype =
{
strNULL,
argNULL,
akNone,
itNULL,
argKPD_NULL,
KPD_error,
KPD_error,
KPD_error,
KPD_error,
KPD_error,
strNULL,
strNULL,
strNULL,
strNULL,
strNULL,
flNone,
d_NO,
FALSE,
rtNULL,
argNULL,
argNULL,
argNULL,
argNULL,
argNULL,
argNULL,
argNULL,
1,
0,
0,
FALSE,
FALSE
};
register argument_t *new;
new = (argument_t *) malloc(sizeof *new);
if (new == argNULL)
fatal("argAlloc(): %s", strerror(errno));
*new = prototype;
return new;
}
routine_t *
rtMakeRoutine(identifier_t name, argument_t *args)
{
register routine_t *rt = rtAlloc();
rt->rtName = name;
rt->rtKind = rkRoutine;
rt->rtArgs = args;
return rt;
}
routine_t *
rtMakeSimpleRoutine(identifier_t name, argument_t *args)
{
register routine_t *rt = rtAlloc();
rt->rtName = name;
rt->rtKind = rkSimpleRoutine;
rt->rtArgs = args;
return rt;
}
char *
rtRoutineKindToStr(routine_kind_t rk)
{
switch (rk) {
case rkRoutine:
return "Routine";
case rkSimpleRoutine:
return "SimpleRoutine";
default:
fatal("rtRoutineKindToStr(%d): not a routine_kind_t", rk);
return strNULL;
}
}
static void
rtPrintArg(register argument_t *arg)
{
register ipc_type_t *it = arg->argType;
if (!akCheck(arg->argKind, akbUserArg|akbServerArg) ||
(akIdent(arg->argKind) == akeCount) ||
(akIdent(arg->argKind) == akeDealloc) ||
(akIdent(arg->argKind) == akeNdrCode) ||
(akIdent(arg->argKind) == akePoly))
return;
printf("\n\t");
switch (akIdent(arg->argKind)) {
case akeRequestPort:
printf("RequestPort");
break;
case akeReplyPort:
printf("ReplyPort");
break;
case akeWaitTime:
printf("WaitTime");
break;
case akeSendTime:
printf("SendTime");
break;
case akeMsgOption:
printf("MsgOption");
break;
case akeMsgSeqno:
printf("MsgSeqno\t");
break;
case akeSecToken:
printf("SecToken\t");
break;
case akeAuditToken:
printf("AuditToken\t");
break;
case akeContextToken:
printf("ContextToken\t");
break;
case akeImplicit:
printf("Implicit\t");
break;
default:
if (akCheck(arg->argKind, akbRequest)) {
if (akCheck(arg->argKind, akbSend))
printf("In");
else
printf("(In)");
}
if (akCheck(arg->argKind, akbReply)) {
if (akCheck(arg->argKind, akbReturn))
printf("Out");
else
printf("(Out)");
}
printf("\t");
}
printf("\t%s: %s", arg->argName, it->itName);
if (arg->argDeallocate == d_YES)
printf(", Dealloc");
else if (arg->argDeallocate == d_MAYBE)
printf(", Dealloc[]");
if (arg->argCountInOut)
printf(", CountInOut");
if (arg->argFlags & flSameCount)
printf(", SameCount");
if (arg->argFlags & flPhysicalCopy)
printf(", PhysicalCopy");
if (arg->argFlags & flRetCode)
printf(", PhysicalCopy");
if (arg->argFlags & flOverwrite)
printf(", Overwrite");
if (arg->argFlags & flAuto)
printf(", Auto");
if (arg->argFlags & flConst)
printf(", Const");
}
void
rtPrintRoutine(register routine_t *rt)
{
register argument_t *arg;
printf("%s (%d) %s(", rtRoutineKindToStr(rt->rtKind), rt->rtNumber, rt->rtName);
for (arg = rt->rtArgs; arg != argNULL; arg = arg->argNext)
rtPrintArg(arg);
printf(")\n");
printf("\n");
}
static void
rtCheckSimple(argument_t *args, u_int mask, boolean_t *simple)
{
register argument_t *arg;
boolean_t MustBeComplex = FALSE;
for (arg = args; arg != argNULL; arg = arg->argNext)
if (akCheck(arg->argKind, mask)) {
register ipc_type_t *it = arg->argType;
if (IS_KERN_PROC_DATA(it))
MustBeComplex = TRUE;
}
*simple = !MustBeComplex;
}
static void
rtCheckFit(routine_t *rt, u_int mask, boolean_t *fitp, boolean_t *uselimp, u_int *knownp)
{
boolean_t uselim = FALSE;
argument_t *arg;
u_int size = sizeof(mach_msg_header_t);
if (!rt->rtSimpleRequest)
machine_alignment(size,sizeof(mach_msg_body_t));
for (arg = rt->rtArgs; arg != argNULL; arg = arg->argNext)
if (akCheck(arg->argKind, mask)) {
register ipc_type_t *it = arg->argType;
machine_alignment(size, it->itMinTypeSize);
if (it->itNative)
uselim = TRUE;
else if (IS_VARIABLE_SIZED_UNTYPED(it)) {
machine_alignment(size, it->itTypeSize);
size += it->itPadSize;
}
}
*knownp = size;
if (MaxMessSizeOnStack == -1) {
*fitp = TRUE;
*uselimp = FALSE;
}
else if (size > MaxMessSizeOnStack) {
*fitp = FALSE;
*uselimp = FALSE;
}
else if (!uselim) {
*fitp = TRUE;
*uselimp = FALSE;
}
else if (UserTypeLimit == -1) {
*fitp = FALSE;
*uselimp = FALSE;
}
else if (size + UserTypeLimit > MaxMessSizeOnStack) {
*fitp = FALSE;
*uselimp = TRUE;
}
else {
*fitp = TRUE;
*uselimp = TRUE;
}
}
static void
rtFindHowMany(routine_t *rt)
{
register argument_t *arg;
int multiplier = 1;
boolean_t test;
for (arg = rt->rtArgs; arg != argNULL; arg = arg->argNext) {
register ipc_type_t *it = arg->argType;
if (IS_MULTIPLE_KPD(it)) {
if (!it->itVarArray)
multiplier = it->itKPD_Number;
test = !it->itVarArray && !it->itElement->itVarArray;
it = it->itElement;
}
else
test = !it->itVarArray;
if (akCheck(arg->argKind, akbSendKPD)) {
if (it->itInLine)
rt->rtCountPortsIn += it->itNumber * multiplier;
else if (it->itPortType) {
if (test)
rt->rtCountOolPortsIn += it->itNumber * multiplier;
}
else {
if (test)
rt->rtCountOolIn += (it->itNumber * it->itSize + 7)/8 * multiplier;
}
}
if (akCheckAll(arg->argKind, akbReturnKPD)) {
if (it->itInLine)
rt->rtCountPortsOut += it->itNumber * multiplier;
else if (it->itPortType) {
if (test)
rt->rtCountOolPortsOut += it->itNumber * multiplier;
}
else {
if (test)
rt->rtCountOolOut += ((it->itNumber * it->itSize + 7)/8) * multiplier;
}
}
}
}
boolean_t
rtCheckMask(argument_t *args, u_int mask)
{
register argument_t *arg;
for (arg = args; arg != argNULL; arg = arg->argNext)
if (akCheckAll(arg->argKind, mask))
return TRUE;
return FALSE;
}
boolean_t
rtCheckMaskFunction(argument_t *args, u_int mask, boolean_t (*func)())
{
register argument_t *arg;
for (arg = args; arg != argNULL; arg = arg->argNext)
if (akCheckAll(arg->argKind, mask))
if ((*func)(arg))
return TRUE;
return FALSE;
}
int
rtCountKPDs(argument_t *args, u_int mask)
{
register argument_t *arg;
int count = 0;
for (arg = args; arg != argNULL; arg = arg->argNext)
if (akCheckAll(arg->argKind, mask))
count += arg->argType->itKPD_Number;
return count;
}
int
rtCountFlags(argument_t *args, u_int flag)
{
register argument_t *arg;
int count = 0;
for (arg = args; arg != argNULL; arg = arg->argNext)
if (arg->argFlags & flag)
count++;
return count;
}
int
rtCountArgDescriptors(argument_t *args, int *argcount)
{
register argument_t *arg;
int count = 0;
if (argcount)
*argcount = 0;
for (arg = args; arg != argNULL; arg = arg->argNext)
if (akCheck(arg->argKind, akbServerArg)) {
if (RPCFixedArray(arg) ||
RPCPort(arg) ||
RPCVariableArray(arg) ||
RPCPortArray(arg)) {
count++;
if (argcount)
(*argcount)++;
}
else {
if (argcount) {
if (arg->argType->itStruct && arg->argType->itNumber &&
(arg->argType->itSize >= 32))
*argcount += arg->argType->itNumber * (arg->argType->itSize / 32);
else
(*argcount)++;
}
}
}
return count;
}
int
rtCountMask(argument_t *args, u_int mask)
{
register argument_t *arg;
int count = 0;
for (arg = args; arg != argNULL; arg = arg->argNext)
if (akCheckAll(arg->argKind, mask))
count++;
return count;
}
static void
rtDefaultArgKind(routine_t *rt, argument_t *arg)
{
if ((arg->argKind == akNone) && (rt->rtRequestPort == argNULL))
arg->argKind = akRequestPort;
if (arg->argKind == akNone)
arg->argKind = akIn;
}
static ipc_flags_t
rtProcessDeallocFlag(register ipc_type_t *it, register ipc_flags_t flags, register arg_kind_t kind, dealloc_t *what, string_t name)
{
if (flags & flMaybeDealloc) {
if (flags & (flDealloc|flNotDealloc)) {
warn("%s: Dealloc and NotDealloc ignored with Dealloc[]", name);
flags &= ~(flDealloc|flNotDealloc);
}
}
if ((flags&(flDealloc|flNotDealloc)) == (flDealloc|flNotDealloc)) {
warn("%s: Dealloc and NotDealloc cancel out", name);
flags &= ~(flDealloc|flNotDealloc);
}
if (((IsKernelServer && akCheck(kind, akbReturn)) ||
(IsKernelUser && akCheck(kind, akbSend))) &&
(flags & flDealloc)) {
*what= d_YES;
}
else if (flags & (flMaybeDealloc|flDealloc)) {
if (it->itInLine && !it->itPortType) {
warn("%s: Dealloc is ignored: it is meaningless for that type of argument", name);
flags &= ~(flMaybeDealloc|flDealloc);
}
else
*what = (flags & flMaybeDealloc) ? d_MAYBE : d_YES;
}
return flags;
}
static void
rtProcessSameCountFlag(register argument_t *arg)
{
register ipc_type_t *it = arg->argType;
register ipc_flags_t flags = arg->argFlags;
string_t name = arg->argVarName;
static argument_t *old_arg;
if (flags & flSameCount) {
if (!it->itVarArray) {
warn("%s: SameCount is ignored - the argument is not variable", name);
flags &= ~flSameCount;
}
if (old_arg) {
if (old_arg->argParent)
old_arg = old_arg->argParent;
if (old_arg->argSameCount)
old_arg = old_arg->argSameCount;
if (!old_arg->argType->itVarArray) {
warn("%s: SameCount is ignored - adjacent argument is not variable", name);
flags &= ~flSameCount;
}
#define SAMECOUNT_MASK akeBITS|akbSend|akbReturn|akbRequest|akbReply|akbUserArg|akbServerArg
if (akCheck(old_arg->argKind, SAMECOUNT_MASK) !=
akCheck(arg->argKind, SAMECOUNT_MASK) ||
old_arg->argCountInOut != arg->argCountInOut) {
warn("%s: SameCount is ignored - inconsistencies with the adjacent argument\n", name);
flags &= ~flSameCount;
}
arg->argSameCount = old_arg;
}
arg->argFlags = flags;
}
old_arg = arg;
}
static ipc_flags_t
rtProcessCountInOutFlag(register ipc_type_t *it, register ipc_flags_t flags, register arg_kind_t kind, boolean_t *what, string_t name)
{
if (flags & flCountInOut) {
if (!akCheck(kind, akbReply)) {
warn("%s: CountInOut is ignored: argument must be Out\n", name);
flags &= ~flCountInOut;
}
else if (!it->itVarArray || !it->itInLine) {
warn("%s: CountInOut is ignored: argument isn't variable or in-line\n", name);
flags &= ~flCountInOut;
}
else
*what = TRUE;
}
return flags;
}
static ipc_flags_t
rtProcessPhysicalCopyFlag(register ipc_type_t *it, register ipc_flags_t flags, register arg_kind_t kind, string_t name)
{
if (flags & flPhysicalCopy) {
if (it->itInLine) {
warn("%s: PhysicalCopy is ignored, argument copied inline anyway", name);
flags &= ~flPhysicalCopy;
}
if (it->itPortType) {
warn("%s: PhysicalCopy is ignored, it does not apply to ports and array of ports", name);
flags &= ~flPhysicalCopy;
}
}
return flags;
}
static void
rtProcessRetCodeFlag(register argument_t *thisarg)
{
register ipc_type_t *it = thisarg->argType;
register ipc_flags_t flags = thisarg->argFlags;
string_t name = thisarg->argVarName;
routine_t *thisrout = thisarg->argRoutine;
if (flags & flRetCode) {
if (!it->itInLine || !it->itStruct ||
it->itSize != 32 || it->itNumber != 1) {
warn("%s: RetCode is ignored - the type doesn't match a MIG RetCode", name);
flags &= ~flRetCode;
}
else if (thisrout->rtKind != rkSimpleRoutine) {
fatal("%s: RetCode is allowed only for SimpleRoutines", name);
}
else if (thisrout->rtRetCArg != argNULL) {
warn("%s: RetCode is ignored - only one argument can be flagged as RetCode", name);
flags &= ~flRetCode;
}
else {
thisrout->rtRetCArg = thisarg;
}
thisarg->argFlags = flags;
}
}
static ipc_flags_t
rtProcessOverwriteFlag(register ipc_type_t *it, register ipc_flags_t flags, register arg_kind_t kind, string_t name)
{
if (flags & flOverwrite)
if (it->itInLine || it->itMigInLine ||
!akCheck(kind, akbReturn) || akCheck(kind, akbSend)) {
warn("%s: Overwrite is ignored - it must be Out AND Ool!", name);
flags &= ~flOverwrite;
}
return flags;
}
static void
rtDetectKPDArg(argument_t *arg)
{
register ipc_type_t *it = arg->argType;
char *string;
if (IS_KERN_PROC_DATA(it)) {
if (akCheck(arg->argKind, akbSendBody)) {
arg->argKind = akRemFeature(arg->argKind, akbSendBody);
arg->argKind = akAddFeature(arg->argKind, akbSendKPD);
}
if (akCheck(arg->argKind, akbReturnBody)) {
arg->argKind = akRemFeature(arg->argKind, akbReturnBody);
arg->argKind = akAddFeature(arg->argKind, akbReturnKPD);
}
if (it->itInLine) {
string = "mach_msg_port_descriptor_t";
arg->argKPD_Type = MACH_MSG_PORT_DESCRIPTOR;
}
else if (it->itPortType) {
string = "mach_msg_ool_ports_descriptor_t";
arg->argKPD_Type = MACH_MSG_OOL_PORTS_DESCRIPTOR;
}
else {
string = "mach_msg_ool_descriptor_t";
arg->argKPD_Type = MACH_MSG_OOL_DESCRIPTOR;
}
it->itKPDType = string;
}
}
static void
rtAugmentArgKind(argument_t *arg)
{
register ipc_type_t *it = arg->argType;
if (IS_VARIABLE_SIZED_UNTYPED(it)) {
if (akCheckAll(arg->argKind, akbRequest|akbReply))
error("%s: Inline variable-sized arguments can't be InOut", arg->argName);
arg->argKind = akAddFeature(arg->argKind, akbVariable);
}
if (IS_OPTIONAL_NATIVE(it))
arg->argKind = akAddFeature(arg->argKind, akbVariable);
if (
((it->itOutTrans != strNULL) && akCheck(arg->argKind, akbReturnSnd)) ||
((it->itInTrans != strNULL) && akCheckAll(arg->argKind, akbSendRcv|akbReturnSnd)) ||
((it->itDestructor != strNULL) && akCheck(arg->argKind, akbSendRcv) && !akCheck(arg->argKind, akbReturnSnd) && (it->itInTrans != strNULL)) ||
(IS_MULTIPLE_KPD(it)) ||
((akIdent(arg->argKind) == akePoly) && akCheck(arg->argKind, akbReturnSnd)) ||
((akIdent(arg->argKind) == akeDealloc) && akCheck(arg->argKind, akbReturnSnd))
) {
arg->argKind = akRemFeature(arg->argKind, akbReplyCopy);
arg->argKind = akAddFeature(arg->argKind, akbVarNeeded);
}
}
static void
rtSuffixExtArg(register argument_t *args)
{
register argument_t *arg;
register char *subindex;
char string[MAX_STR_LEN];
for (arg = args; arg != argNULL; arg = arg->argNext) {
if (akCheck(arg->argKind, akbSendKPD | akbReturnKPD)) {
if (IS_MULTIPLE_KPD(arg->argType))
subindex = "[0]";
else
subindex = "";
switch (arg->argKPD_Type) {
case MACH_MSG_PORT_DESCRIPTOR:
(void)sprintf(string, "%s.name", subindex);
break;
case MACH_MSG_OOL_DESCRIPTOR:
case MACH_MSG_OOL_PORTS_DESCRIPTOR:
(void)sprintf(string, "%s.address", subindex);
break;
default:
error("Type of kernel processed data unknown\n");
}
arg->argSuffix = strconcat(arg->argMsgField, string);
}
else if (akIdent(arg->argKind) == akePoly &&
akCheck(arg->argParent->argKind, akbSendKPD | akbReturnKPD)) {
register argument_t *par_arg = arg->argParent;
if (IS_MULTIPLE_KPD(par_arg->argType))
subindex = "[0]";
else
subindex = "";
switch (par_arg->argKPD_Type) {
case MACH_MSG_PORT_DESCRIPTOR:
case MACH_MSG_OOL_PORTS_DESCRIPTOR:
(void)sprintf(string, "%s.disposition", subindex);
arg->argSuffix = strconcat(par_arg->argMsgField, string);
break;
default:
error("Type of kernel processed data inconsistent\n");
}
}
else if (akIdent(arg->argKind) == akeDealloc &&
akCheck(arg->argParent->argKind, akbSendKPD | akbReturnKPD)) {
register argument_t *par_arg = arg->argParent;
if (IS_MULTIPLE_KPD(par_arg->argType))
subindex = "[0]";
else
subindex = "";
switch (par_arg->argKPD_Type) {
case MACH_MSG_OOL_DESCRIPTOR:
case MACH_MSG_OOL_PORTS_DESCRIPTOR:
(void)sprintf(string, "%s.deallocate", subindex);
arg->argSuffix = strconcat(par_arg->argMsgField, string);
break;
default:
error("Type of kernel processed data inconsistent\n");
}
}
}
}
static void
rtCheckRoutineArg(routine_t *rt, argument_t *arg)
{
switch (akIdent(arg->argKind)) {
case akeRequestPort:
if (rt->rtRequestPort != argNULL)
warn("multiple RequestPort args in %s; %s won't be used", rt->rtName, rt->rtRequestPort->argName);
rt->rtRequestPort = arg;
break;
case akeReplyPort:
if (rt->rtReplyPort != argNULL)
warn("multiple ReplyPort args in %s; %s won't be used", rt->rtName, rt->rtReplyPort->argName);
rt->rtReplyPort = arg;
break;
case akeWaitTime:
if (rt->rtWaitTime != argNULL)
warn("multiple WaitTime/SendTime type args in %s; %s won't be used", rt->rtName, rt->rtWaitTime->argName);
rt->rtWaitTime = arg;
break;
case akeSendTime:
if (rt->rtWaitTime != argNULL)
warn("multiple WaitTime/SendTime type args in %s; %s won't be used", rt->rtName, rt->rtWaitTime->argName);
rt->rtWaitTime = arg;
break;
case akeMsgOption:
if (rt->rtMsgOption != argNULL)
warn("multiple MsgOption args in %s; %s won't be used", rt->rtName, rt->rtMsgOption->argName);
rt->rtMsgOption = arg;
break;
default:
break;
}
}
static void
rtSetArgDefaults(routine_t *rt, register argument_t *arg)
{
arg->argRoutine = rt;
if (arg->argVarName == strNULL)
arg->argVarName = arg->argName;
if (arg->argMsgField == strNULL)
switch(akIdent(arg->argKind)) {
case akeRequestPort:
arg->argMsgField = "Head.msgh_request_port";
break;
case akeReplyPort:
arg->argMsgField = "Head.msgh_reply_port";
break;
case akeNdrCode:
arg->argMsgField = "NDR";
break;
case akeSecToken:
arg->argMsgField = "msgh_sender";
break;
case akeAuditToken:
arg->argMsgField = "msgh_audit";
break;
case akeContextToken:
arg->argMsgField = "msgh_context";
break;
case akeMsgSeqno:
arg->argMsgField = "msgh_seqno";
break;
case akeImplicit:
break;
default:
arg->argMsgField = arg->argName;
break;
}
if (arg->argTTName == strNULL)
arg->argTTName = strconcat(arg->argName, "Template");
if (arg->argPadName == strNULL)
arg->argPadName = strconcat(arg->argName, "Pad");
if ((rt->rtRequestPort != argNULL) &&
(rt->rtRequestPort->argPoly == arg) &&
(arg->argType != itNULL)) {
arg->argMsgField = "Head.msgh_bits";
arg->argType->itInTrans = "MACH_MSGH_BITS_REQUEST";
}
if ((rt->rtReplyPort != argNULL) &&
(rt->rtReplyPort->argPoly == arg) &&
(arg->argType != itNULL)) {
arg->argMsgField = "Head.msgh_bits";
arg->argType->itInTrans = "MACH_MSGH_BITS_REPLY";
}
}
static void
rtAddCountArg(register argument_t *arg)
{
register argument_t *count, *master;
register ipc_type_t *it = arg->argType;
count = argAlloc();
if (IS_MULTIPLE_KPD(it) && it->itElement->itVarArray) {
count->argName = strconcat(arg->argName, "Subs");
count->argType = itMakeSubCountType(it->itKPD_Number, it->itVarArray, arg->argVarName);
count->argKind = akeSubCount;
arg->argSubCount = count;
}
else {
count->argName = strconcat(arg->argName, "Cnt");
count->argType = itMakeCountType();
count->argKind = akeCount;
arg->argCount = count;
if (arg->argParent != argNULL) {
arg->argParent->argCount = count;
}
}
master = (arg->argParent != argNULL) ? arg->argParent : arg;
if (IS_MULTIPLE_KPD(master->argType))
count->argMultiplier = 1;
else
count->argMultiplier = it->itElement->itNumber;
count->argParent = arg;
count->argNext = arg->argNext;
arg->argNext = count;
if (arg->argType->itString) {
count->argKind = akAddFeature(count->argKind, akCheck(arg->argKind, akbSend) ? akbSendRcv : akbReturnRcv);
count->argVarName = (char *)0;
}
else {
count->argKind |= akAddFeature(akbUserArg|akbServerArg, (arg->argKind) & ~akeBITS);
count->argKind = akRemFeature(count->argKind, akbVariable|akbVarNeeded);
if (IS_VARIABLE_SIZED_UNTYPED(arg->argType))
count->argKind = akRemFeature(count->argKind, akbRequest|akbReply);
}
}
static void
rtAddCountInOutArg(register argument_t *arg)
{
register argument_t *count;
count = argAlloc();
count->argName = strconcat(arg->argName, "Cnt");
count->argType = itMakeCountType();
count->argParent = argNULL;
count->argNext = arg->argNext;
arg->argNext = count;
(count->argCInOut = arg->argCount)->argCInOut = count;
count->argKind = akCountInOut;
}
static void
rtAddPolyArg(register argument_t *arg)
{
register ipc_type_t *it = arg->argType;
register argument_t *poly;
arg_kind_t akbsend, akbreturn;
poly = argAlloc();
poly->argName = strconcat(arg->argName, "Poly");
poly->argType = itMakePolyType();
poly->argParent = arg;
poly->argNext = arg->argNext;
arg->argNext = poly;
arg->argPoly = poly;
akbsend = akbSend;
akbreturn = akbReturn;
if (it->itInName == MACH_MSG_TYPE_POLYMORPHIC) {
akbsend |= akbUserArg|akbSendSnd;
if (!IsKernelServer)
akbreturn |= akbServerArg|akbReturnSnd;
}
if (it->itOutName == MACH_MSG_TYPE_POLYMORPHIC) {
akbsend |= akbServerArg|akbSendRcv;
akbreturn |= akbUserArg|akbReturnRcv;
if (IsKernelServer)
akbreturn |= akbServerArg|akbReturnSnd;
}
poly->argKind = akPoly;
if (akCheck(arg->argKind, akbSend))
poly->argKind = akAddFeature(poly->argKind, akCheck(arg->argKind, akbsend));
if (akCheck(arg->argKind, akbReturn))
poly->argKind = akAddFeature(poly->argKind, akCheck(arg->argKind, akbreturn));
}
static void
rtAddDeallocArg(register argument_t *arg)
{
register argument_t *dealloc;
dealloc = argAlloc();
dealloc->argName = strconcat(arg->argName, "Dealloc");
dealloc->argType = itMakeDeallocType();
dealloc->argParent = arg;
dealloc->argNext = arg->argNext;
arg->argNext = dealloc;
arg->argDealloc = dealloc;
dealloc->argKind = akeDealloc;
if (akCheck(arg->argKind, akbSend))
dealloc->argKind = akAddFeature(dealloc->argKind, akCheck(arg->argKind, akbUserArg|akbSend|akbSendSnd));
if (akCheck(arg->argKind, akbReturn)) {
dealloc->argKind = akAddFeature(dealloc->argKind, akCheck(arg->argKind, akbServerArg|akbReturn|akbReturnSnd));
dealloc->argByReferenceServer = TRUE;
}
}
static void
rtCheckRoutineArgs(routine_t *rt)
{
register argument_t *arg;
for (arg = rt->rtArgs; arg != argNULL; arg = arg->argNext) {
register ipc_type_t *it = arg->argType;
rtDefaultArgKind(rt, arg);
rtCheckRoutineArg(rt, arg);
rtSetArgDefaults(rt, arg);
if (it != itNULL) {
arg->argFlags = rtProcessDeallocFlag(it, arg->argFlags, arg->argKind, &arg->argDeallocate, arg->argVarName);
arg->argFlags = rtProcessCountInOutFlag(it, arg->argFlags, arg->argKind, &arg->argCountInOut, arg->argVarName);
rtProcessSameCountFlag(arg);
arg->argFlags = rtProcessPhysicalCopyFlag(it, arg->argFlags, arg->argKind, arg->argVarName);
rtProcessRetCodeFlag(arg);
arg->argFlags = rtProcessOverwriteFlag(it, arg->argFlags, arg->argKind, arg->argVarName);
rtAugmentArgKind(arg);
if (arg->argDeallocate == d_MAYBE)
rtAddDeallocArg(arg);
if (it->itVarArray || (IS_MULTIPLE_KPD(it) && it->itElement->itVarArray))
rtAddCountArg(arg);
if (arg->argCountInOut)
rtAddCountInOutArg(arg);
if ((it->itInName == MACH_MSG_TYPE_POLYMORPHIC) || (it->itOutName == MACH_MSG_TYPE_POLYMORPHIC))
rtAddPolyArg(arg);
rtDetectKPDArg(arg);
}
}
}
boolean_t
rtCheckTrailerType(register argument_t *arg)
{
if (akIdent(arg->argKind) == akeSecToken ||
akIdent(arg->argKind) == akeAuditToken ||
akIdent(arg->argKind) == akeContextToken )
itCheckTokenType(arg->argVarName, arg->argType);
if (akIdent(arg->argKind) == akeMsgSeqno)
itCheckIntType(arg->argVarName, arg->argType);
return TRUE;
}
static void
rtCheckArgTypes(routine_t *rt)
{
if (rt->rtRequestPort == argNULL)
error("%s %s doesn't have a server port argument", rtRoutineKindToStr(rt->rtKind), rt->rtName);
if ((rt->rtRequestPort != argNULL) &&
(rt->rtRequestPort->argType != itNULL))
itCheckRequestPortType(rt->rtRequestPort->argName, rt->rtRequestPort->argType);
if ((rt->rtReplyPort != argNULL) &&
(rt->rtReplyPort->argType != itNULL))
itCheckReplyPortType(rt->rtReplyPort->argName, rt->rtReplyPort->argType);
if ((rt->rtWaitTime != argNULL) &&
(rt->rtWaitTime->argType != itNULL))
itCheckIntType(rt->rtWaitTime->argName, rt->rtWaitTime->argType);
if ((rt->rtMsgOption != argNULL) &&
(rt->rtMsgOption->argType != itNULL))
itCheckIntType(rt->rtMsgOption->argName, rt->rtMsgOption->argType);
if ((IsKernelServer && rt->rtServerImpl) ||
(IsKernelUser && rt->rtUserImpl))
fatal("Implicit data is not supported in the KernelUser and KernelServer modes");
if (rt->rtServerImpl)
rtCheckMaskFunction(rt->rtArgs, akbServerImplicit, rtCheckTrailerType);
if (rt->rtUserImpl)
rtCheckMaskFunction(rt->rtArgs, akbUserImplicit, rtCheckTrailerType);
}
static void
rtCheckArgTrans(routine_t *rt)
{
register argument_t *arg;
for (arg = rt->rtArgs; arg != argNULL; arg = arg->argNext) {
register ipc_type_t *it = arg->argType;
if ((it != itNULL) && !streql(it->itServerType, it->itTransType)) {
if (akCheck(arg->argKind, akbSendRcv) && (it->itInTrans == strNULL))
warn("%s: argument has no in-translation function", arg->argName);
if (akCheck(arg->argKind, akbReturnSnd) && (it->itOutTrans == strNULL))
warn("%s: argument has no out-translation function", arg->argName);
}
}
}
static void
rtAddRetCode(routine_t *rt)
{
register argument_t *arg = argAlloc();
arg->argName = "RetCode";
arg->argType = itRetCodeType;
arg->argKind = akRetCode;
rt->rtRetCode = arg;
arg->argNext = rt->rtArgs;
rt->rtArgs = arg;
}
static void
rtProcessRetCode(routine_t *rt)
{
if (!rt->rtOneWay && !rt->rtSimpleReply) {
register argument_t *arg = rt->rtRetCode;
arg->argKind = akRemFeature(arg->argKind, akbReply);
arg->argKind = akAddFeature(arg->argKind, akbVarNeeded);
}
if (rt->rtRetCArg != argNULL && !rt->rtSimpleRequest) {
register argument_t *arg = rt->rtRetCArg;
arg->argKind = akeRetCode|akbUserArg|akbServerArg|akbSendRcv;
}
}
static void
rtAddNdrCode(routine_t *rt)
{
register argument_t *arg = argAlloc();
arg->argName = "NDR_record";
arg->argType = itNdrCodeType;
arg->argKind = akeNdrCode;
rt->rtNdrCode = arg;
arg->argNext = rt->rtArgs;
rt->rtArgs = arg;
}
static void
rtProcessNdrCode(routine_t *rt)
{
register argument_t *ndr = rt->rtNdrCode;
argument_t *arg;
boolean_t found;
#define ndr_send akbRequest|akbSend|akbSendSnd|akbSendBody
#define ndr_rcv akbReply|akbReplyInit|akbReturn|akbReturnBody
ndr->argKind = akAddFeature(ndr->argKind, ndr_send|ndr_rcv);
for (found = FALSE, arg = ndr->argNext; arg != argNULL; arg = arg->argNext)
if (akCheck(arg->argKind, akbSendRcv|akbSendBody) &&
!akCheck(arg->argKind, akbServerImplicit) && !arg->argType->itPortType &&
(!arg->argParent || akIdent(arg->argKind) == akeCount ||
akIdent(arg->argKind) == akeCountInOut)) {
arg->argKind = akAddFeature(arg->argKind, akbSendNdr);
found = TRUE;
}
if (!found)
ndr->argKind = akRemFeature(ndr->argKind, ndr_send);
found = FALSE;
if (!rt->rtOneWay)
for (arg = ndr->argNext; arg != argNULL; arg = arg->argNext)
if ((arg == rt->rtRetCode && akCheck(arg->argKind, akbReply)) ||
(arg != rt->rtRetCode &&
akCheck(arg->argKind, akbReturnRcv|akbReturnBody) &&
!akCheck(arg->argKind, akbUserImplicit) && !arg->argType->itPortType &&
(!arg->argParent || akIdent(arg->argKind) == akeCount ||
akIdent(arg->argKind) == akeCountInOut))) {
arg->argKind = akAddFeature(arg->argKind, akbReturnNdr);
found = TRUE;
}
if (!found && !akCheck(rt->rtRetCode->argKind, akbReply))
ndr->argKind = akRemFeature(ndr->argKind, ndr_rcv);
}
static void
rtAddWaitTime(routine_t *rt, identifier_t name)
{
register argument_t *arg = argAlloc();
argument_t **loc;
arg->argName = "dummy WaitTime arg";
arg->argVarName = name;
arg->argType = itWaitTimeType;
arg->argKind = akeWaitTime;
rt->rtWaitTime = arg;
if (rt->rtMsgOption != argNULL)
loc = &rt->rtMsgOption->argNext;
else
loc = &rt->rtArgs;
arg->argNext = *loc;
*loc = arg;
rtSetArgDefaults(rt, arg);
}
static void
rtAddMsgOption(routine_t *rt, identifier_t name)
{
register argument_t *arg = argAlloc();
argument_t **loc;
arg->argName = "dummy MsgOption arg";
arg->argVarName = name;
arg->argType = itMsgOptionType;
arg->argKind = akeMsgOption;
rt->rtMsgOption = arg;
loc = &rt->rtArgs;
arg->argNext = *loc;
*loc = arg;
rtSetArgDefaults(rt, arg);
}
static void
rtProcessMsgOption(routine_t *rt)
{
register argument_t *msgop = rt->rtMsgOption;
register argument_t *arg;
boolean_t sectoken = FALSE;
boolean_t audittoken = FALSE;
boolean_t contexttoken = FALSE;
for (arg = rt->rtArgs; arg != argNULL; arg = arg->argNext)
if (akCheckAll(arg->argKind, akbReturn|akbUserImplicit)) {
if (akIdent(arg->argKind) == akeSecToken)
sectoken = TRUE;
else if (akIdent(arg->argKind) == akeAuditToken)
audittoken = TRUE;
else if (akIdent(arg->argKind) == akeContextToken)
contexttoken = TRUE;
}
if (contexttoken == TRUE)
msgop->argVarName = strconcat(msgop->argVarName, "|MACH_RCV_TRAILER_ELEMENTS(MACH_RCV_TRAILER_CTX)");
else if (audittoken == TRUE)
msgop->argVarName = strconcat(msgop->argVarName, "|MACH_RCV_TRAILER_ELEMENTS(MACH_RCV_TRAILER_AUDIT)");
else if (sectoken == TRUE)
msgop->argVarName = strconcat(msgop->argVarName, "|MACH_RCV_TRAILER_ELEMENTS(MACH_RCV_TRAILER_SENDER)");
}
static void
rtAddDummyReplyPort(routine_t *rt, ipc_type_t *type)
{
register argument_t *arg = argAlloc();
argument_t **loc;
arg->argName = "dummy ReplyPort arg";
arg->argVarName = "dummy ReplyPort arg";
arg->argType = type;
arg->argKind = akeReplyPort;
rt->rtReplyPort = arg;
if (rt->rtRequestPort != argNULL)
loc = &rt->rtRequestPort->argNext;
else
loc = &rt->rtArgs;
arg->argNext = *loc;
*loc = arg;
rtSetArgDefaults(rt, arg);
}
static void
rtCheckOverwrite(register routine_t *rt)
{
register argument_t *arg;
register int howmany = rt->rtOverwrite;
for (arg = rt->rtArgs; arg != argNULL; arg = arg->argNext) {
register ipc_type_t *it = arg->argType;
if (akCheck(arg->argKind, akbReturnKPD) && !it->itInLine) {
arg->argKind = akAddFeature(arg->argKind, akbOverwrite);
if (arg->argFlags & flOverwrite)
howmany--;
if (!howmany)
return;
}
}
}
static void
rtCheckVariable(register routine_t *rt)
{
register argument_t *arg;
int NumRequestVar = 0;
int NumReplyVar = 0;
int MaxRequestPos = 0;
int MaxReplyPos = 0;
for (arg = rt->rtArgs; arg != argNULL; arg = arg->argNext) {
register argument_t *parent = arg->argParent;
if (parent == argNULL || akCheck(parent->argKind, akbSendKPD|akbReturnKPD)) {
if (akCheckAll(arg->argKind, akbSend|akbSendBody)) {
arg->argRequestPos = NumRequestVar;
MaxRequestPos = NumRequestVar;
if (akCheck(arg->argKind, akbVariable))
NumRequestVar++;
}
if (akCheckAll(arg->argKind, akbReturn|akbReturnBody)) {
arg->argReplyPos = NumReplyVar;
MaxReplyPos = NumReplyVar;
if (akCheck(arg->argKind, akbVariable))
NumReplyVar++;
}
}
else {
arg->argRequestPos = parent->argRequestPos;
arg->argReplyPos = parent->argReplyPos;
}
if (akCheckAll(arg->argKind, akbReturnSnd|akbReturnBody) &&
!akCheck(arg->argKind, akbReplyCopy|akbVarNeeded) &&
(arg->argReplyPos > 0))
arg->argKind = akAddFeature(arg->argKind, akbVarNeeded);
}
rt->rtNumRequestVar = NumRequestVar;
rt->rtNumReplyVar = NumReplyVar;
rt->rtMaxRequestPos = MaxRequestPos;
rt->rtMaxReplyPos = MaxReplyPos;
}
static void
rtCheckDestroy(register routine_t *rt)
{
register argument_t *arg;
for (arg = rt->rtArgs; arg != argNULL; arg = arg->argNext) {
register ipc_type_t *it = arg->argType;
if(akCheck(arg->argKind, akbSendRcv) &&
!akCheck(arg->argKind, akbReturnSnd) &&
(it->itDestructor != strNULL || IS_MIG_INLINE_EMUL(it))) {
arg->argKind = akAddFeature(arg->argKind, akbDestroy);
}
if (argIsIn(arg) && akCheck(arg->argKind, akbSendKPD|akbReturnKPD) &&
arg->argKPD_Type == MACH_MSG_OOL_DESCRIPTOR &&
(arg->argFlags & flAuto))
arg->argKind = akAddFeature(arg->argKind, akbDestroy);
}
}
static void
rtAddByReference(register routine_t *rt)
{
register argument_t *arg;
for (arg = rt->rtArgs; arg != argNULL; arg = arg->argNext) {
register ipc_type_t *it = arg->argType;
if (akCheck(arg->argKind, akbReturnRcv) && it->itStruct) {
arg->argByReferenceUser = TRUE;
if (arg->argCInOut != argNULL)
arg->argCInOut->argByReferenceUser = TRUE;
}
if ((akCheck(arg->argKind, akbReturnSnd) ||
(akCheck(arg->argKind, akbServerImplicit) &&
akCheck(arg->argKind, akbReturnRcv) &&
akCheck(arg->argKind, akbSendRcv)))
&& it->itStruct) {
arg->argByReferenceServer = TRUE;
}
}
}
void
rtAddSameCount(register routine_t *rt)
{
register argument_t *arg;
for (arg = rt->rtArgs; arg != argNULL; arg = arg->argNext)
if (arg->argFlags & flSameCount) {
register ipc_type_t *it = arg->argType;
register argument_t *tmp_count;
register argument_t *my_count = arg->argCount;
register argument_t *ref_count = arg->argSameCount->argCount;
tmp_count = argAlloc();
*tmp_count = *ref_count;
tmp_count->argKind = akeSameCount;
ref_count->argKind = akAddFeature(ref_count->argKind, akCheck(my_count->argKind, akbVarNeeded));
tmp_count->argKind = akAddFeature(tmp_count->argKind, akCheck(my_count->argKind, akbVarNeeded));
tmp_count->argNext = my_count->argNext;
tmp_count->argMultiplier = my_count->argMultiplier;
tmp_count->argType = my_count->argType;
tmp_count->argParent = arg;
arg->argCount = tmp_count;
arg->argNext = tmp_count;
if (IS_VARIABLE_SIZED_UNTYPED(it))
it->itMinTypeSize = 0;
tmp_count->argType->itMinTypeSize = 0;
tmp_count->argType->itTypeSize = 0;
}
}
void
rtCheckRoutine(register routine_t *rt)
{
rt->rtErrorName = ErrorProc;
rt->rtOneWay = (rt->rtKind == rkSimpleRoutine);
rt->rtServerName = strconcat(ServerPrefix, rt->rtName);
rt->rtUserName = strconcat(UserPrefix, rt->rtName);
rtAddRetCode(rt);
rtAddNdrCode(rt);
rtCheckRoutineArgs(rt);
if (rt->rtReplyPort == argNULL) {
if (rt->rtOneWay)
rtAddDummyReplyPort(rt, itZeroReplyPortType);
else
rtAddDummyReplyPort(rt, itRealReplyPortType);
}
if (rt->rtMsgOption == argNULL) {
if (MsgOption == strNULL)
rtAddMsgOption(rt, "MACH_MSG_OPTION_NONE");
else
rtAddMsgOption(rt, MsgOption);
}
if ((rt->rtWaitTime == argNULL) &&
(WaitTime != strNULL))
rtAddWaitTime(rt, WaitTime);
rtCheckArgTypes(rt);
rtCheckArgTrans(rt);
if (rt->rtOneWay) {
if (rtCheckMask(rt->rtArgs, akbReturn) || rt->rtUserImpl)
error("%s %s has OUT argument", rtRoutineKindToStr(rt->rtKind), rt->rtName);
}
else {
if ((rt->rtWaitTime != argNULL) && (rt->rtWaitTime->argKind == akeSendTime))
error("%s %s SendTime not supported on routines, use WaitTime", rtRoutineKindToStr(rt->rtKind), rt->rtName);
}
if (mig_errors > 0)
fatal("%d errors found. Abort.\n", mig_errors);
rt->rtServerImpl = rtCountMask(rt->rtArgs, akbServerImplicit);
rt->rtUserImpl = rtCountMask(rt->rtArgs, akbUserImplicit);
rtCheckSimple(rt->rtArgs, akbRequest, &rt->rtSimpleRequest);
rtCheckSimple(rt->rtArgs, akbReply, &rt->rtSimpleReply);
rt->rtRequestKPDs = rtCountKPDs(rt->rtArgs, akbSendKPD);
rt->rtReplyKPDs = rtCountKPDs(rt->rtArgs, akbReturnKPD);
if ((rt->rtOverwrite = rtCountFlags(rt->rtArgs, flOverwrite))) {
rtCheckOverwrite(rt);
rt->rtOverwriteKPDs = rtCountKPDs(rt->rtArgs, akbReturnKPD|akbOverwrite);
if (IsKernelUser)
fatal("Overwrite option(s) do not match with the KernelUser personality\n");
}
rtCheckFit(rt, akbRequest, &rt->rtRequestFits, &rt->rtRequestUsedLimit, &rt->rtRequestSizeKnown);
rtCheckFit(rt, akbReply, &rt->rtReplyFits, &rt->rtReplyUsedLimit, &rt->rtReplySizeKnown);
rtCheckVariable(rt);
rtCheckDestroy(rt);
rtAddByReference(rt);
rtSuffixExtArg(rt->rtArgs);
rtAddSameCount(rt);
rtProcessRetCode(rt);
rtProcessNdrCode(rt);
if (rt->rtUserImpl)
rtProcessMsgOption(rt);
rt->rtNoReplyArgs = !rtCheckMask(rt->rtArgs, akbReturnSnd);
if (UseEventLogger)
rtFindHowMany(rt);
}
void
rtMinRequestSize(FILE *file, routine_t *rt, char *str)
{
fprintf(file, "(mach_msg_size_t)(sizeof(%s)", str);
rtSizeDelta(file, akbRequest, rt);
fprintf(file, ")");
}
void
rtMinReplySize(FILE *file, routine_t *rt, char *str)
{
fprintf(file, "(mach_msg_size_t)(sizeof(%s)", str);
rtSizeDelta(file, akbReply, rt);
fprintf(file, ")");
}
static void
rtSizeDelta(FILE *file, u_int mask, routine_t *rt)
{
argument_t *arg;
u_int min_size = sizeof(mach_msg_header_t);
u_int max_size;
boolean_t output = FALSE;
if (!rt->rtSimpleRequest)
machine_alignment(min_size, sizeof(mach_msg_body_t));
max_size = min_size;
for (arg = rt->rtArgs; arg != argNULL; arg = arg->argNext)
if (akCheck(arg->argKind, mask)) {
register ipc_type_t *it = arg->argType;
machine_alignment(min_size, it->itMinTypeSize);
machine_alignment(max_size, it->itMinTypeSize);
if (IS_VARIABLE_SIZED_UNTYPED(it)) {
machine_alignment(max_size, it->itTypeSize);
max_size += it->itPadSize;
}
if (IS_OPTIONAL_NATIVE(it)) {
if (output)
fprintf(file, " + ");
else {
output = TRUE;
fprintf(file, " - (");
}
fprintf(file, "_WALIGNSZ_(%s)", it->itUserType);
}
}
if (min_size != max_size) {
if (output)
fprintf(file, " + ");
else
fprintf(file, " - ");
fprintf(file, "%d", max_size - min_size);
}
if (output)
fprintf(file, ")");
}