mstask: Implement ITask::GetNextRunTime() for WEEKLY trigger type.

Signed-off-by: Dmitry Timoshkov <dmitry@baikal.ru>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Dmitry Timoshkov 2018-05-22 16:18:44 +08:00 committed by Alexandre Julliard
parent 2a56402c56
commit 5a9ed96aec
2 changed files with 60 additions and 1 deletions

View file

@ -466,6 +466,11 @@ static void filetime_add_days(FILETIME *ft, ULONG days)
filetime_add_hours(ft, (ULONGLONG)days * 24);
}
static void filetime_add_weeks(FILETIME *ft, ULONG weeks)
{
filetime_add_days(ft, (ULONGLONG)weeks * 7);
}
static HRESULT WINAPI MSTASK_ITask_GetNextRunTime(ITask *iface, SYSTEMTIME *rt)
{
TaskImpl *This = impl_from_ITask(iface);
@ -540,6 +545,40 @@ static HRESULT WINAPI MSTASK_ITask_GetNextRunTime(ITask *iface, SYSTEMTIME *rt)
}
break;
case TASK_TIME_TRIGGER_WEEKLY:
if (!This->trigger[i].Type.Weekly.rgfDaysOfTheWeek)
break; /* avoid infinite loop */
st = current_st;
st.wHour = This->trigger[i].wStartHour;
st.wMinute = This->trigger[i].wStartMinute;
st.wSecond = 0;
st.wMilliseconds = 0;
SystemTimeToFileTime(&st, &current_ft);
while (CompareFileTime(&current_ft, &end_ft) < 0)
{
FileTimeToSystemTime(&current_ft, &st);
if (CompareFileTime(&current_ft, &begin_ft) >= 0)
{
if (This->trigger[i].Type.Weekly.rgfDaysOfTheWeek & (1 << st.wDayOfWeek))
{
if (!have_best_time || CompareFileTime(&current_ft, &best_ft) < 0)
{
best_ft = current_ft;
have_best_time = TRUE;
}
break;
}
}
if (st.wDayOfWeek == 0) /* Sunday, goto next week */
filetime_add_weeks(&current_ft, This->trigger[i].Type.Weekly.WeeksInterval - 1);
else /* check next weekday */
filetime_add_days(&current_ft, 1);
}
break;
default:
FIXME("trigger type %u is not handled\n", This->trigger[i].TriggerType);
break;

View file

@ -552,7 +552,27 @@ static void test_GetNextRunTime(void)
st.wDay, st.wMonth, st.wYear, st.wDayOfWeek,
st.wHour, st.wMinute, st.wSecond);
/* FIXME: TASK_TIME_TRIGGER_WEEKLY */
/* TASK_TIME_TRIGGER_WEEKLY */
hr = ITaskTrigger_GetTrigger(trigger, &data);
ok(hr == S_OK, "got %#x\n", hr);
data.rgFlags &= ~TASK_TRIGGER_FLAG_DISABLED;
data.TriggerType = TASK_TIME_TRIGGER_WEEKLY;
data.Type.Weekly.WeeksInterval = 1;
/* add 3 days */
time_add_ms(&cmp, 3 * 24 * 60 * 60 * 1000);
/* bits: TASK_SUNDAY = 1, TASK_MONDAY = 2, TASK_TUESDAY = 4, etc. */
data.Type.Weekly.rgfDaysOfTheWeek = 1 << cmp.wDayOfWeek; /* wDayOfWeek is 0 based */
hr = ITaskTrigger_SetTrigger(trigger, &data);
ok(hr == S_OK, "got %#x\n", hr);
memset(&st, 0xff, sizeof(st));
hr = ITask_GetNextRunTime(task, &st);
ok(hr == S_OK, "got %#x\n", hr);
ok(!memcmp(&st, &cmp, sizeof(st)), "got %u/%u/%u wday %u %u:%02u:%02u\n",
st.wDay, st.wMonth, st.wYear, st.wDayOfWeek,
st.wHour, st.wMinute, st.wSecond);
/* FIXME: TASK_TIME_TRIGGER_MONTHLYDATE */
/* FIXME: TASK_TIME_TRIGGER_MONTHLYDOW */