my_scheduler.c

#include "vcpu.h"
#include "schedule.h"
#include "timer.h"
#include "logger.h"
#include "debug.h"
#include "port.h"
#include "gic.h"
#include "my_scheduler.h"
#include "timer_event.h"
#include "vcpu.h"

typedef struct t_queue {
    QUEUE que;
    T_VCPU *vcpu;
} queue_t;

static queue_t _que[16];
static QUEUE_LIST _wque;
static QUEUE_LIST _fque;



static void my_init(void) {
    int i;
    queue_list_init(&_wque);
    queue_list_init(&_fque);
    for(i=0;i<16;i++){
        queue_list_enque(&_fque, &(_que[i].que));
    }
}

static void *my_allocate(T_VCPU *vcpu){
    if(queue_list_is_empty(&_fque)){
        tv_abort("# canot allocate !");
        return NULL;
    }
    queue_t *ret = (queue_t *)queue_list_pop(&_fque);
    ret->vcpu = vcpu;
    return ret;
}

static void my_enque(T_VCPU *vcpu){
    queue_list_enque(&_wque, (QUEUE *)vcpu->sched_state);
    if(vcpu_get_executing()==NULL) scheduler_set_flag();
}


static T_VCPU *my_schedule(void) {
    if(queue_list_is_empty(&_wque)){
        return vcpu_get_executing();
    }
    T_VCPU *ret = ((queue_t *)queue_list_pop(&_wque))->vcpu;
    return ret;
}

static void my_yield(void){
    scheduler_set_flag();
}

static void my_block(T_VCPU *vcpu){
    my_enque(vcpu);
}

static void my_unblock(T_VCPU *vcpu){
    if(vcpu->state==VCPU_STATE_SLEEP){
        my_enque(vcpu);
        if(vcpu_get_executing()==NULL) scheduler_set_flag();
    }
}

schedule_operation_t my_scheduler = {
    my_init,
    my_schedule,
    my_yield,
    my_block,
    my_unblock,
    my_allocate,
    my_enque,
};