artiq/soc/runtime/main.c

215 lines
4.9 KiB
C
Raw Normal View History

2014-07-04 23:49:08 +08:00
#include <stdio.h>
#include <string.h>
2014-07-04 23:49:08 +08:00
#include <irq.h>
#include <uart.h>
#include <console.h>
2014-07-04 23:49:08 +08:00
#include <system.h>
2014-09-15 22:56:23 +08:00
#include <time.h>
#include <generated/csr.h>
2014-07-04 23:49:08 +08:00
#include "test_mode.h"
2014-10-19 23:51:49 +08:00
#include "comm.h"
#include "kernelcpu.h"
2014-07-04 23:49:08 +08:00
#include "elf_loader.h"
2014-09-21 23:36:10 +08:00
#include "exceptions.h"
#include "services.h"
2014-08-28 16:56:48 +08:00
#include "rtio.h"
#include "dds.h"
2014-07-24 07:10:49 +08:00
#ifdef ARTIQ_AMP
#include "mailbox.h"
#include "messages.h"
#endif
static struct symbol symtab[128];
static int _symtab_count;
static char _symtab_strings[128*16];
static char *_symtab_strptr;
static void symtab_init(void)
{
memset(symtab, 0, sizeof(symtab));
_symtab_count = 0;
_symtab_strptr = _symtab_strings;
}
static int symtab_add(const char *name, void *target)
{
if(_symtab_count >= sizeof(symtab)/sizeof(symtab[0])) {
2014-10-19 23:51:49 +08:00
comm_log("Too many provided symbols in object");
symtab_init();
return 0;
}
symtab[_symtab_count].name = _symtab_strptr;
symtab[_symtab_count].target = target;
_symtab_count++;
while(1) {
if(_symtab_strptr >= &_symtab_strings[sizeof(_symtab_strings)]) {
2014-10-19 23:51:49 +08:00
comm_log("Provided symbol string table overflow");
symtab_init();
return 0;
}
*_symtab_strptr = *name;
_symtab_strptr++;
if(*name == 0)
break;
name++;
}
return 1;
}
static int load_object(void *buffer, int length)
{
symtab_init();
return load_elf(
resolve_service_symbol, symtab_add,
buffer, length, (void *)KERNELCPU_PAYLOAD_ADDRESS, 4*1024*1024);
}
#ifdef ARTIQ_AMP
2015-04-16 21:47:05 +08:00
static int process_msg(struct msg_base *umsg, int *eid, long long int *eparams)
{
int i;
switch(umsg->type) {
case MESSAGE_TYPE_FINISHED:
return KERNEL_RUN_FINISHED;
case MESSAGE_TYPE_EXCEPTION: {
struct msg_exception *msg = (struct msg_exception *)umsg;
*eid = msg->eid;
for(i=0;i<3;i++)
eparams[i] = msg->eparams[i];
return KERNEL_RUN_EXCEPTION;
}
2015-04-06 19:40:12 +08:00
case MESSAGE_TYPE_RPC_REQUEST: {
struct msg_rpc_request *msg = (struct msg_rpc_request *)umsg;
struct msg_rpc_reply reply;
reply.type = MESSAGE_TYPE_RPC_REPLY;
2015-04-06 22:28:10 +08:00
comm_rpc_va(msg->rpc_num, msg->args, &reply.eid, &reply.retval);
2015-04-06 19:40:12 +08:00
mailbox_send_and_wait(&reply);
return KERNEL_RUN_INVALID_STATUS;
}
case MESSAGE_TYPE_LOG: {
struct msg_log *msg = (struct msg_log *)umsg;
comm_log(msg->fmt, msg->args);
return KERNEL_RUN_INVALID_STATUS;
}
default:
*eid = EID_INTERNAL_ERROR;
for(i=0;i<3;i++)
eparams[i] = 0;
return KERNEL_RUN_EXCEPTION;
}
}
#endif
2014-07-06 04:47:23 +08:00
typedef void (*kernel_function)(void);
2014-07-04 23:49:08 +08:00
static int run_kernel(const char *kernel_name, int *eid, long long int *eparams)
2014-07-04 23:49:08 +08:00
{
kernel_function k;
#ifdef ARTIQ_AMP
int r;
#else
2014-09-22 13:18:48 +08:00
void *jb;
#endif
2014-07-04 23:49:08 +08:00
k = find_symbol(symtab, kernel_name);
if(k == NULL) {
2014-10-19 23:51:49 +08:00
comm_log("Failed to find kernel entry point '%s' in object", kernel_name);
2014-09-21 23:36:10 +08:00
return KERNEL_RUN_STARTUP_FAILED;
}
2015-04-05 17:55:05 +08:00
#ifdef ARTIQ_AMP
kernelcpu_start(k);
while(1) {
2015-04-16 21:47:05 +08:00
struct msg_base *umsg;
umsg = mailbox_receive();
r = KERNEL_RUN_INVALID_STATUS;
if(umsg)
r = process_msg(umsg, eid, eparams);
if(r != KERNEL_RUN_INVALID_STATUS)
break;
}
kernelcpu_stop();
return r;
#else
2014-09-22 13:18:48 +08:00
jb = exception_push();
if(exception_setjmp(jb)) {
*eid = exception_getid(eparams);
2014-09-21 23:36:10 +08:00
return KERNEL_RUN_EXCEPTION;
2014-09-22 13:18:48 +08:00
} else {
dds_init();
2014-09-21 23:36:10 +08:00
rtio_init();
flush_cpu_icache();
k();
exception_pop(1);
2014-09-21 23:36:10 +08:00
return KERNEL_RUN_FINISHED;
}
#endif
}
2014-09-15 22:56:23 +08:00
static void blink_led(void)
{
int i, ev, p;
p = identifier_frequency_read()/10;
time_init();
for(i=0;i<3;i++) {
leds_out_write(1);
while(!elapsed(&ev, p));
leds_out_write(0);
while(!elapsed(&ev, p));
}
}
static int check_test_mode(void)
{
char c;
timer0_en_write(0);
timer0_reload_write(0);
timer0_load_write(identifier_frequency_read() >> 2);
timer0_en_write(1);
timer0_update_value_write(1);
while(timer0_value_read()) {
if(readchar_nonblock()) {
c = readchar();
if((c == 't')||(c == 'T'))
return 1;
}
timer0_update_value_write(1);
}
return 0;
}
int main(void)
{
2014-09-05 12:03:22 +08:00
irq_setmask(0);
irq_setie(1);
uart_init();
2014-09-22 13:18:48 +08:00
2015-04-05 17:55:05 +08:00
#ifdef ARTIQ_AMP
puts("ARTIQ runtime built "__DATE__" "__TIME__" for AMP systems\n");
#else
2015-04-05 17:55:05 +08:00
puts("ARTIQ runtime built "__DATE__" "__TIME__" for UP systems\n");
#endif
2014-09-15 22:56:23 +08:00
blink_led();
if(check_test_mode()) {
puts("Entering test mode.");
test_main();
} else {
puts("Entering regular mode.");
comm_serve(load_object, run_kernel);
}
2014-09-05 12:03:22 +08:00
return 0;
2014-07-04 23:49:08 +08:00
}