#include "type.h"
#include <mach/message.h>
#include <mach/kern_return.h>
#include "error.h"
#include "alloc.h"
#include "global.h"
#include "routine.h"
#include "write.h"
u_int rtNumber = 0;
routine_t *
rtAlloc()
{
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()
{
rtNumber++;
}
argument_t *
argAlloc()
{
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(name, args)
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(name, args)
identifier_t name;
argument_t *args;
{
register routine_t *rt = rtAlloc();
rt->rtName = name;
rt->rtKind = rkSimpleRoutine;
rt->rtArgs = args;
return rt;
}
char *
rtRoutineKindToStr(rk)
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(arg)
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 akeMsgOption:
printf("MsgOption");
break;
case akeMsgSeqno:
printf("MsgSeqno\t");
break;
case akeSecToken:
printf("SecToken\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(rt)
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(args, mask, simple)
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 u_int
rtFindSize(args, mask, ismax, simple)
argument_t *args;
u_int mask;
boolean_t ismax, simple;
{
register argument_t *arg;
u_int size = sizeof(mach_msg_header_t);
if (!simple)
size += sizeof(mach_msg_body_t);
for (arg = args; arg != argNULL; arg = arg->argNext)
if (akCheck(arg->argKind, mask)) {
register ipc_type_t *it = arg->argType;
size += it->itMinTypeSize;
if (ismax && IS_VARIABLE_SIZED_UNTYPED(it))
size += it->itTypeSize + it->itPadSize;
}
return size;
}
static void
rtFindHowMany(rt)
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(args, mask)
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(args, mask, func)
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(args, mask)
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(args, flag)
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(args, argcount)
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(args, mask)
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(rt, arg)
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(it, flags, kind, what, name)
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(arg)
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(it, flags, kind, what, name)
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(it, flags, kind, name)
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(thisarg)
register argument_t *thisarg;
{
register ipc_type_t *it = thisarg->argType;
register ipc_flags_t flags = thisarg->argFlags;
register arg_kind_t kind = thisarg->argKind;
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(it, flags, kind, name)
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(arg)
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->itUserKPDType = string;
it->itServerKPDType = string;
}
}
static void
rtAugmentArgKind(arg)
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 (((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(args)
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(rt, arg)
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 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(rt, arg)
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 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(arg)
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->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(arg)
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(arg)
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(arg)
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(rt)
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);
}
}
}
static void
rtCheckTrailerType(arg)
register argument_t *arg;
{
if (akIdent(arg->argKind) == akeSecToken)
itCheckSecTokenType(arg->argVarName, arg->argType);
if (akIdent(arg->argKind) == akeMsgSeqno)
itCheckIntType(arg->argVarName, arg->argType);
}
static void
rtCheckArgTypes(rt)
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(rt)
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(rt)
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(rt)
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(rt)
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(rt)
routine_t *rt;
{
register argument_t *ndr = rt->rtNdrCode;
#define ndr_send akbRequest|akbSend|akbSendSnd|akbSendBody
#define ndr_rcv akbReply|akbReplyInit|akbReturn|akbReturnBody
ndr->argKind = akAddFeature(ndr->argKind, ndr_send|ndr_rcv);
if (!rtCheckMask(ndr->argNext, akbRequest|akbSendBody))
ndr->argKind = akRemFeature(ndr->argKind, ndr_send);
if (!rt->rtOneWay &&
!akCheck(rt->rtRetCode->argKind, akbReply) &&
!rtCheckMask(ndr->argNext, akbReply|akbReturnBody))
ndr->argKind = akRemFeature(ndr->argKind, ndr_rcv);
}
static void
rtAddWaitTime(rt, name)
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(rt, name)
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(rt)
routine_t *rt;
{
register argument_t *msgop = rt->rtMsgOption;
register argument_t *arg;
for (arg = rt->rtArgs; arg != argNULL; arg = arg->argNext)
if (akCheckAll(arg->argKind, akbReturn|akbUserImplicit)) {
if (akIdent(arg->argKind) == akeSecToken)
msgop->argVarName = strconcat(msgop->argVarName,
"|MACH_RCV_TRAILER_ELEMENTS(MACH_RCV_TRAILER_SENDER)");
}
}
static void
rtAddDummyReplyPort(rt, type)
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(rt)
register routine_t *rt;
{
register argument_t *arg;
register 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(rt)
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(rt)
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(rt)
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) && it->itStruct) {
arg->argByReferenceServer = TRUE;
}
}
}
void
rtAddSameCount(rt)
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(rt)
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 &&
(rtCheckMask(rt->rtArgs, akbReturn) || rt->rtUserImpl))
error("%s %s has OUT argument",
rtRoutineKindToStr(rt->rtKind), rt->rtName);
if (errors > 0)
fatal("%d errors found. Abort.\n", 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");
}
rt->rtMessOnStack =
(MaxMessSizeOnStack < 0 ||
(rtFindSize(rt->rtArgs, akbRequest, TRUE, rt->rtSimpleRequest)
<= MaxMessSizeOnStack &&
rtFindSize(rt->rtArgs, akbReply, TRUE, rt->rtSimpleRequest)
<= MaxMessSizeOnStack));
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);
rt->rtRequestMinSize = rtFindSize(rt->rtArgs, akbRequest, FALSE, rt->rtSimpleRequest);
rt->rtRequestMaxSize = rtFindSize(rt->rtArgs, akbRequest, TRUE, rt->rtSimpleRequest);
rt->rtReplyMinSize = rtFindSize(rt->rtArgs, akbReply, FALSE, rt->rtSimpleReply);
rt->rtReplyMaxSize = rtFindSize(rt->rtArgs, akbReply, TRUE, rt->rtSimpleReply);
if (UseEventLogger)
rtFindHowMany(rt);
}