Working on the new UDP protocol
This commit is contained in:
jaseg
parent
5a77d3a0e0
commit
809f8ac2e0
5 changed files with 206 additions and 247 deletions
220
host/main.c
220
host/main.c
|
|
@ -1,8 +1,9 @@
|
|||
|
||||
#include "config.h"
|
||||
#include "main.h"
|
||||
#include "color.h"
|
||||
#include "font.h"
|
||||
#include "gif.h"
|
||||
#include "net.h"
|
||||
#include <stdint.h>
|
||||
#include <stdlib.h>
|
||||
#include <stdio.h>
|
||||
|
|
@ -11,8 +12,13 @@
|
|||
#include <wchar.h>
|
||||
#include <locale.h>
|
||||
#include <sys/timeb.h>
|
||||
#include <sys/queue.h>
|
||||
#include <netdb.h>
|
||||
#include <sys/socket.h>
|
||||
#include <netinet/in.h>
|
||||
#include <unistd.h>
|
||||
|
||||
|
||||
/* CAUTION: REQUIRES INPUT TO BE \0-TERMINATED
|
||||
* ...also, it does a hardcodes setlocale of LC_CTYPE to en_US.utf8 for... reasons. */
|
||||
framebuffer_t *framebuffer_render_text(char *s, glyph_t **glyph_table, unsigned int glyph_table_size){
|
||||
|
|
@ -347,12 +353,18 @@ void console_render_buffer(framebuffer_t *fb){
|
|||
|
||||
int main(int argc, char **argv){
|
||||
FILE *f = NULL;
|
||||
int udpfd = 0;
|
||||
int udp6fd = 0;
|
||||
uint8_t *udpbuf = NULL;
|
||||
uint8_t *fbdata = NULL;
|
||||
framebuffer_t *fb = NULL;
|
||||
|
||||
if(argc != 2){
|
||||
fprintf(stderr, "No or too much input text given\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* Read font file */
|
||||
FILE *fontfile = fopen("unifont.bdf", "r");
|
||||
if(!fontfile){
|
||||
fprintf(stderr, "Error opening font file: %s\n", strerror(errno));
|
||||
|
|
@ -366,61 +378,181 @@ int main(int argc, char **argv){
|
|||
}
|
||||
fclose(fontfile);
|
||||
|
||||
int delay = 20;
|
||||
framebuffer_t *fb;
|
||||
gifAnimationState_t *gifstate = NULL;
|
||||
char *p = strstr(argv[1], ".gif");
|
||||
if(p && p[4] == '\0'){ /* Argument ends with ".gif", try to read it as a gif file */
|
||||
f = fopen(argv[1], "r");
|
||||
if(!f){
|
||||
fprintf(stderr, "Error opening gif file from argument (\"%s\"): %s\n", argv[1], strerror(errno));
|
||||
goto error;
|
||||
}
|
||||
uint8_t *buf = NULL;
|
||||
size_t size = 0;
|
||||
size_t read = 0;
|
||||
const size_t READ_INC = 1024;
|
||||
do{
|
||||
size_t newsize = size+READ_INC;
|
||||
uint8_t *oldbuf = buf;
|
||||
buf = realloc(buf, newsize);
|
||||
if(!buf){
|
||||
free(oldbuf);
|
||||
fclose(f);
|
||||
fprintf(stderr, "Error opening gif file from argument (\"%s\"): Cannot allocate %lu bytes.\n", argv[1], newsize);
|
||||
goto error;
|
||||
}
|
||||
read += fread(buf+size, 1, READ_INC, f);
|
||||
size = newsize;
|
||||
}while(read == size);
|
||||
fb = framebuffer_render_gif(buf, read, &gifstate, &delay);
|
||||
free(buf);
|
||||
fclose(f);
|
||||
}else{
|
||||
fb = framebuffer_render_text(argv[1], glyph_table, BLP_SIZE);
|
||||
/* Set up framebuffer */
|
||||
fbdata = malloc(DISPLAY_WIDTH*DISPLAY_HEIGHT*sizeof(color_t));
|
||||
if(!fbdata){
|
||||
fprintf(stderr, "Cannot alloccate framebuffer\n");
|
||||
goto error;
|
||||
}
|
||||
fb = malloc(sizeof(*fb));
|
||||
if(!fb){
|
||||
fprintf(stderr, "Cannot alloccate framebuffer\n");
|
||||
goto error;
|
||||
}
|
||||
fb->w = DISPLAY_WIDTH;
|
||||
fb->h = DISPLAY_HEIGHT;
|
||||
fb->data = fbdata;
|
||||
|
||||
/* Set up UDP server */
|
||||
udp4fd = socket(AF_INET, SOCK_DGRAM | SOCK_NONBLOCK, IPPROTO_UDP);
|
||||
udp6fd = socket(AF_INET6, SOCK_DGRAM | SOCK_NONBLOCK, IPPROTO_UDP);
|
||||
if(udp4fd < 0 || udp6fd < 0){
|
||||
fprintf(stderr, "Cannot open UDP sockets: %s\n", strerror(errno));
|
||||
goto error;
|
||||
}
|
||||
struct sockaddr_in udp_addr = {
|
||||
AF_INET,
|
||||
INADDR_ANY,
|
||||
htons(UDP_PORT)
|
||||
};
|
||||
bind(udp4fd, &udp_addr, sizeof(udp_addr));
|
||||
struct sockaddr_in udp6_addr = {
|
||||
AF_INET6,
|
||||
IN6ADDR_ANY_INIT,
|
||||
htons(UDP_PORT)
|
||||
};
|
||||
bind(udp6fd, &udp_addr, sizeof(udp_addr));
|
||||
/* Set up UDP receive buffer */
|
||||
struct msghdr msg;
|
||||
memset(&msg, 0, sizeof(msg));
|
||||
udpbuf = malloc(UDP_BUF_SIZE);
|
||||
if(!buf){
|
||||
fprintf(stderr, "Cannot allocate UDP buffer");
|
||||
goto error;
|
||||
}
|
||||
struct iovec iov[1] = {{udpbuf, sizeof(UDP_BUF_SIZE)}};
|
||||
msg.msg_iov = iov;
|
||||
msg.msg_iovlen = 1;
|
||||
|
||||
/* List of currently active UDP streams */
|
||||
LIST_HEAD(udp_client_entry, udp_client) udp_client_head;
|
||||
struct udp_client {
|
||||
uint64_t last_timestamp;
|
||||
uint64_t last_sequence_number;
|
||||
struct sockaddr remote_addr;
|
||||
int active:1;
|
||||
LIST_ENTRY(udp_client) entries;
|
||||
};
|
||||
LIST_INIT(&udp_client_head);
|
||||
|
||||
LIST_HEAD(text_queue_head_t, text_queue_entry) text_queue_head;
|
||||
struct text_queue_entry {
|
||||
char *text;
|
||||
int current_position;
|
||||
unsigned int loop_count;
|
||||
LIST_ENTRY(text_queue_entry) entries;
|
||||
};
|
||||
LIST_INIT(&text_queue_head);
|
||||
|
||||
for(;;){ /* Never gonna give you up, never gonna let you down! */
|
||||
if(!fb){
|
||||
fprintf(stderr, "Error rendering.\n");
|
||||
struct timeb time = {0};
|
||||
ftime(&time);
|
||||
uint64_t now = time.time*1000 + time.millitm;
|
||||
|
||||
int made_active = 0;
|
||||
/* Choose next active client */
|
||||
for(struct udp_client *p = udp_client_head.lh_first; p != NULL; p = p->entries.le_next){
|
||||
if(now - p->last_frame_received > FRAME_TIMEOUT){ /* Connection timeout */
|
||||
LIST_REMOVE(p, entries);
|
||||
continue;
|
||||
}
|
||||
if(!made_active){
|
||||
p->active = 1;
|
||||
made_active = 1;
|
||||
}else{
|
||||
p->active = 0;
|
||||
}
|
||||
}
|
||||
|
||||
if(!made_active){ /* No active streams, render marquee */
|
||||
text_queue_entry *entry = text_queue_head.lh_first;
|
||||
framebuffer_t *text = framebuffer_render_text(entry, glyph_table, BLP_SIZE);
|
||||
/* COPY TEXT DATA */
|
||||
for(size_t line = 0; line < DISPLAY_HEIGHT; line++){
|
||||
color_t *datarow = fb->data+(line*DISPLAY_WIDTH);
|
||||
color_t *textrow = text->data+(line*text->w);
|
||||
if(entry->current_position > 0)
|
||||
memset(datarow, 0, sizeof(color_t)*entry->current_position);
|
||||
if(entry->current_position+text->w < DISPLAY_WIDTH)
|
||||
memset(datarow+(entry->curent_position+text->w), 0, DISPLAY_WIDTH-(entry->curent_position+text->w));
|
||||
memcpy(datarow+(entry->current_position>0 ? entry->current_position : 0),
|
||||
textrow+(entry->current_position>0 ? 0 : -entry->current_position),
|
||||
//DISPLAY_WIDTH, text->w-current_position FIXME
|
||||
}
|
||||
entry->current_position++;
|
||||
if(entry->current_position <= -text->w){
|
||||
entry->loop_count++;
|
||||
if(entry->loop_count > TEXT_LOOP_COUNT){
|
||||
LIST_REMOVE(entry, entries);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/* Receive pending IPv4 UDP packets */
|
||||
ssize_t received = 0;
|
||||
while((received = recvmsg(udp4fd, &msg, 0)) > 0){
|
||||
if(received < UDP_BUF_SIZE) /* Packet too short */
|
||||
continue;
|
||||
ml_packet_t *pkt = (ml_packet_t*)udpbuf;
|
||||
if(ntohl(pkt->magic) != 0xDEADBEEF)
|
||||
continue;
|
||||
if(ntohs(pkt->width) != DISPLAY_WIDTH || ntohs(pkt->height) != DISPLAY_HEIGHT)
|
||||
continue;
|
||||
struct sockaddr_in *src = msg.msg_name;
|
||||
struct udp_client *entry = NULL;
|
||||
for(struct udp_client *p = udp_client_head.lh_first; p != NULL; p = p->entries.le_next){
|
||||
if(!memcmp(&p->remote_addr, src, sizeof(sockaddr))){ /* found */
|
||||
entry = p;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if(!entry){
|
||||
entry = malloc(sizeof(struct udp_client));
|
||||
if(!entry){
|
||||
fprintf(stderr, "Cannot allocate UDP connection entry\n");
|
||||
goto error;
|
||||
}
|
||||
memset(entry, 0, sizeof(entry));
|
||||
LIST_INSERT_HEAD(&udp_client_head, entry, entries);
|
||||
memcpy(entry->remote_addr, src, sizeof(sockaddr));
|
||||
}
|
||||
if(entry->last_sequence_number > pkt->seq) /* Delayed packet containing old data */
|
||||
continue;
|
||||
entry->last_timestamp = now;
|
||||
entry->last_sequence_number = pkt->seq;
|
||||
if(entry->active){
|
||||
/* Copy frame, extending RGB UDP packet data to RGBA framebuffer format */
|
||||
color_t *p = fb->data;
|
||||
rgb_t *q = pkt->data;
|
||||
while(p < fb->data+fb->w*fb->h){
|
||||
p->a = 0;
|
||||
*(rgb_t *)p++ = *q++;
|
||||
}
|
||||
}
|
||||
}
|
||||
if(received < 0){
|
||||
fprintf(stderr, "Error receiving UDP datagram: %s\n", strerror(errno));
|
||||
goto error;
|
||||
}
|
||||
|
||||
printf("\033[2J");
|
||||
/* Receive pending IPv6 UDP packets */
|
||||
/* FIXME */
|
||||
|
||||
/* Render frame buffer */
|
||||
printf("\033[2J");
|
||||
console_render_buffer(fb);
|
||||
printf("\n");
|
||||
usleep(delay*1000);
|
||||
|
||||
if(gifstate){
|
||||
fb = framebuffer_render_gif(NULL, 0, &gifstate, &delay);
|
||||
if(delay == -1)
|
||||
delay = 20;
|
||||
}else{
|
||||
fb = framebuffer_render_text(argv[1], glyph_table, BLP_SIZE);
|
||||
}
|
||||
usb_send_buffer(fb);
|
||||
usleep(1000000/FRAMERATE);
|
||||
}
|
||||
return 0;
|
||||
error:
|
||||
fclose(f);
|
||||
close(udpfd);
|
||||
close(udp6fd);
|
||||
free(udpbuf);
|
||||
free(fbdata);
|
||||
free(fb);
|
||||
return 1;
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue