Check callback memory is allocated before trying to use it.

This commit is contained in:
Chris Young 2019-06-09 20:30:46 +01:00
parent 93494790f3
commit 24590a1145
1 changed files with 2 additions and 1 deletions

View File

@ -345,12 +345,13 @@ nserror ami_schedule(int t, void (*callback)(void *p), void *p)
nscb = AllocSysObjectTags(ASOT_IOREQUEST,
ASOIOR_Duplicate, tioreq,
TAG_DONE);
if(nscb == NULL) return NSERROR_NOMEM;
#else
if(schedule_msgport == NULL) return NSERROR_NOMEM;
nscb = AllocVec(sizeof(struct nscallback), MEMF_PUBLIC | MEMF_CLEAR);
if(nscb == NULL) return NSERROR_NOMEM;
*nscb = *tioreq;
#endif
if(!nscb) return NSERROR_NOMEM;
if (ami_schedule_add_timer_event(nscb, t) != NSERROR_OK)
return NSERROR_NOMEM;