PsmPartition sptrace_start(int smkey, int smsize, char *sm,
PsmPartition trace, char *name)
{
int nameLen;
int smid;
PsmMgtOutcome outcome;
TraceHeader *trh;
PsmAddress traceHeaderAddress;
CHKNULL(trace);
CHKNULL(smsize > 0);
CHKNULL(name);
if ((nameLen = strlen(name)) < 1 || nameLen > 31)
{
sptracePrint("start: name must be 1-31 characters.");
return NULL;
}
/* Attach to shared memory used for trace operations. */
if (sm_ShmAttach(smkey, smsize, &sm, &smid) < 0)
{
sptracePrint("start: can't attach shared memory for trace.");
return NULL;
}
/* Manage the shared memory region. "Trace" argument
* is normally NULL. */
if (psm_manage(sm, smsize, name, &trace, &outcome) < 0)
{
sptracePrint("start: shared memory mgt failed.");
return NULL;
}
switch (outcome)
{
case Refused:
sptracePrint("start: can't psm_manage shared memory.");
return NULL;
case Redundant:
trh = (TraceHeader *) psp(trace, psm_get_root(trace));
if (trh == NULL || strcmp(name, trh->name) != 0)
{
sptracePrint("start: shared memory used otherwise.");
return NULL;
}
return trace; /* Trace already started. */
default:
break;
}
/* Initialize the shared memory region for tracing. */
traceHeaderAddress = psm_zalloc(trace, sizeof(TraceHeader));
if (traceHeaderAddress == 0)
{
sptracePrint("start: not enough memory for header.");
sm_ShmDetach(sm);
sm_ShmDestroy(smid);
return NULL;
}
oK(psm_set_root(trace, traceHeaderAddress));
trh = (TraceHeader *) psp(trace, traceHeaderAddress);
CHKNULL(trh);
trh->traceSmId = smid;
memset(trh->name, 0, sizeof trh->name);
istrcpy(trh->name, name, sizeof trh->name);
trh->opCount = 0;
trh->files = sm_list_create(trace);
if (trh->files == 0)
{
sptracePrint("start: not enough memory for files list.");
sm_ShmDetach(sm);
sm_ShmDestroy(smid);
return NULL;
}
trh->log = sm_list_create(trace);
if (trh->log == 0)
{
sptracePrint("start: not enough memory for log.");
sm_ShmDetach(sm);
sm_ShmDestroy(smid);
return NULL;
}
return trace;
}
int main(int argc, char **argv)
#endif
{
char *wmspace;
int wmid;
PsmPartition wm = NULL;
PsmMgtOutcome outcome;
PsmAddress testlist;
sm_SemId semaphore;
int cycleNbr = 1;
char fileName[256];
int outputFile;
PsmAddress lineListElt;
PsmAddress lineAddress;
char *line;
if (sm_ipc_init() < 0)
{
return 0;
}
if (sm_ShmAttach(0x1108, 10000000, &wmspace, &wmid) < 0)
{
PERROR("can't attach to shared memory");
return 0;
}
if (psm_manage(wmspace, 10000000, "file2sm", &wm, &outcome) < 0
|| outcome == Refused)
{
PUTS("can't manage shared memory");
return 0;
}
testlist = psm_get_root(wm);
if (testlist == 0)
{
testlist = sm_list_create(wm);
if (testlist == 0)
{
PUTS("can't create shared memory list");
return 0;
}
psm_set_root(wm, testlist);
}
semaphore = sm_SemCreate(0x1101, SM_SEM_FIFO);
if (semaphore < 0)
{
PUTS("can't create semaphore");
return 0;
}
PUTMEMO("Working on cycle", utoa(cycleNbr));
isprintf(fileName, sizeof fileName, "file_copy_%d", cycleNbr);
outputFile = iopen(fileName, O_WRONLY | O_APPEND, 0666);
if (outputFile < 0)
{
PERROR("can't open output file");
return 0;
}
while (1)
{
while (sm_list_length(wm, testlist) == 0)
{
sm_SemTake(semaphore); /* Wait for line. */
}
lineListElt = sm_list_first(wm, testlist);
lineAddress = sm_list_data(wm, lineListElt);
line = psp(wm, lineAddress);
/* Process text of line. */
if (strcmp(line, "*** End of the file ***\n") == 0)
{
/* Close file, open next one. */
close(outputFile);
cycleNbr++;
PUTMEMO("Working on cycle", utoa(cycleNbr));
isprintf(fileName, sizeof fileName, "file_copy_%d",
cycleNbr);
outputFile = iopen(fileName, O_WRONLY | O_APPEND, 0666);
if (outputFile < 0)
{
PERROR("Can't open output file");
return 0;
}
}
else /* Just write line to output file. */
{
if (iputs(outputFile, line) < 0)
{
close(outputFile);
PERROR("Can't write to output file");
return 0;
}
}
/* Delete line from shared memory list. */
psm_free(wm, lineAddress);
CHKZERO(sm_list_delete(wm, lineListElt, (SmListDeleteFn) NULL,
NULL) == 0);
}
}
