k-space/nyancat-server
10
0
Fork 0
Code Issues Pull Requests Packages Projects Activity

Merge pull request #11 from Peetz0r/master

Browse Source 
Apple ][ support (please oh please don't break things)
This commit is contained in:
Kevin Lange 2012-02-20 11:19:53 -08:00
commit 4fd7ab2ce4
1 changed files with 58 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

81
src/nyancat.c
View File

@ -1,8 +1,13 @@
/* /*
* Copyright (c) 2011 Kevin Lange. All rights reserved. * Copyright (c) 2011 Kevin Lange. All rights reserved.
* *
* Developed by: Kevin Lange * Developed by: Kevin Lange
* http://github.com/klange/nyancat * http://github.com/klange/nyancat
* http://miku.acm.uiuc.edu
*
* 40-column support by: Peter Hazenberg
* http://github.com/Peetz0r/nyancat
* http://peter.haas-en-berg.nl
* *
* This is a simple telnet server / standalone application which renders the * This is a simple telnet server / standalone application which renders the
* classic Nyan Cat (or "poptart cat") to your terminal. * classic Nyan Cat (or "poptart cat") to your terminal.
@ -47,6 +52,7 @@
#include <signal.h> #include <signal.h>
#include <time.h> #include <time.h>
#include <setjmp.h> #include <setjmp.h>
#include <sys/ioctl.h>
/* /*
* telnet.h contains some #defines for the various * telnet.h contains some #defines for the various
@ -112,10 +118,10 @@ int digits(int val) {
/* /*
* In the standalone mode, we want to handle an interrupt signal * In the standalone mode, we want to handle an interrupt signal
* (^C) so that we can restore the cursor. * (^C) so that we can restore the cursor and clear the terminal.
*/ */
void SIGINT_handler(int sig){ void SIGINT_handler(int sig){
printf("\033[?25h\033[0m"); printf("\033[?25h\033[0m\033[H\033[2J");
exit(0); exit(0);
} }
@ -179,8 +185,8 @@ void set_options() {
telnet_willack[ECHO] = DO; telnet_willack[ECHO] = DO;
/* The client can set a graphics mode */ /* The client can set a graphics mode */
telnet_willack[SGA] = DO; telnet_willack[SGA] = DO;
/* The client should not change the window size */ /* The client should not change, but it should tell us its window size */
telnet_willack[NAWS] = DONT; telnet_willack[NAWS] = DO;
/* The client should tell us its terminal type (very important) */ /* The client should tell us its terminal type (very important) */
telnet_willack[TTYPE] = DO; telnet_willack[TTYPE] = DO;
/* No linemode */ /* No linemode */
@ -223,6 +229,7 @@ int main(int argc, char ** argv) {
/* I have a bad habit of being very C99, so this may not be everything */ /* I have a bad habit of being very C99, so this may not be everything */
/* The default terminal is ANSI */ /* The default terminal is ANSI */
char term[1024] = {'a','n','s','i', 0}; char term[1024] = {'a','n','s','i', 0};
int terminal_width = 80;
int k, ttype; int k, ttype;
uint32_t option = 0, done = 0, sb_mode = 0, do_echo = 0; uint32_t option = 0, done = 0, sb_mode = 0, do_echo = 0;
/* Various pieces for the telnet communication */ /* Various pieces for the telnet communication */
@ -260,7 +267,7 @@ int main(int argc, char ** argv) {
alarm(1); alarm(1);
/* Let's do this */ /* Let's do this */
while (!feof(stdin) && !done) { while (!feof(stdin) && done < 2) {
/* Get either IAC (start command) or a regular character (break, unless in SB mode) */ /* Get either IAC (start command) or a regular character (break, unless in SB mode) */
unsigned char i = getchar(); unsigned char i = getchar();
unsigned char opt = 0; unsigned char opt = 0;
@ -276,7 +283,14 @@ int main(int argc, char ** argv) {
* that this should be a terminal type */ * that this should be a terminal type */
alarm(0); alarm(0);
strcpy(term, &sb[2]); strcpy(term, &sb[2]);
goto ready; done++;
}
else if (sb[0] == NAWS) {
/* This was a response to the NAWS command, meaning
* that this should be a window size */
alarm(0);
terminal_width = sb[2];
done++;
} }
break; break;
case NOP: case NOP:
@ -324,7 +338,7 @@ int main(int argc, char ** argv) {
break; break;
case IAC: case IAC:
/* IAC IAC? That's probably not right. */ /* IAC IAC? That's probably not right. */
done = 1; done = 2;
break; break;
default: default:
break; break;
@ -336,7 +350,7 @@ int main(int argc, char ** argv) {
* but only if it doesn't put us over * but only if it doesn't put us over
* our limit; honestly, we shouldn't hit * our limit; honestly, we shouldn't hit
* the limit, as we're only collecting characters * the limit, as we're only collecting characters
* for a terminal type, but better safe than * for a terminal type or window size, but better safe than
* sorry (and vulernable). * sorry (and vulernable).
*/ */
sb[sb_len] = i; sb[sb_len] = i;
@ -351,23 +365,21 @@ int main(int argc, char ** argv) {
* terminal type from the environment. */ * terminal type from the environment. */
char * nterm = getenv("TERM"); char * nterm = getenv("TERM");
strcpy(term, nterm); strcpy(term, nterm);
}
/* /* Also get the number of columns */
* Labels. Yes, and I used a goto. struct winsize w;
* If you're going to complain about this, you ioctl(0, TIOCGWINSZ, &w);
* really need to reevaluate your approach to getting terminal_width = w.ws_col;
* things done. This works, it works well, and there }
* is absolutely, positively nothing wrong with using
* goto in C, so I'm going to do it.
*/
ready:
/* Convert the entire terminal string to lower case */ /* Convert the entire terminal string to lower case */
for (k = 0; k < strlen(term); ++k) { for (k = 0; k < strlen(term); ++k) {
term[k] = tolower(term[k]); term[k] = tolower(term[k]);
} }
/* We don't want terminals wider than 80 columns */
if(terminal_width > 80) terminal_width = 80;
/* Do our terminal detection */ /* Do our terminal detection */
if (strstr(term, "xterm")) { if (strstr(term, "xterm")) {
ttype = 1; /* 256-color, spaces */ ttype = 1; /* 256-color, spaces */
@ -383,6 +395,8 @@ ready:
ttype = 4; /* Unicode fallback */ ttype = 4; /* Unicode fallback */
} else if (strstr(term, "rxvt")) { } else if (strstr(term, "rxvt")) {
ttype = 3; /* Accepts LINUX mode */ ttype = 3; /* Accepts LINUX mode */
} else if (strstr(term, "vt100") && terminal_width == 40) {
ttype = 7; /* No color support, only 40 columns */
} else { } else {
ttype = 2; /* Everything else */ ttype = 2; /* Everything else */
} }
@ -490,6 +504,24 @@ ready:
colors['%'] = "()"; /* Pink cheeks */ colors['%'] = "()"; /* Pink cheeks */
always_escape = 1; always_escape = 1;
break; break;
case 7:
colors[','] = "."; /* Blue background */
colors['.'] = "@"; /* White stars */
colors['\''] = " "; /* Black border */
colors['@'] = "#"; /* Tan poptart */
colors['$'] = "?"; /* Pink poptart */
colors['-'] = "O"; /* Red poptart */
colors['>'] = "#"; /* Red rainbow */
colors['&'] = "="; /* Orange rainbow */
colors['+'] = "-"; /* Yellow Rainbow */
colors['#'] = "+"; /* Green rainbow */
colors['='] = "~"; /* Light blue rainbow */
colors[';'] = "$"; /* Dark blue rainbow */
colors['*'] = ";"; /* Gray cat face */
colors['%'] = "o"; /* Pink cheeks */
always_escape = 1;
terminal_width = 40;
break;
default: default:
break; break;
} }
@ -570,7 +602,7 @@ ready:
double diff = difftime(current, start); double diff = difftime(current, start);
/* Now count the length of the time difference so we can center */ /* Now count the length of the time difference so we can center */
int nLen = digits((int)diff); int nLen = digits((int)diff);
int width = (80 - 29 - nLen) / 2; int width = (terminal_width - 29 - nLen) / 2;
/* Spit out some spaces so that we're actually centered */ /* Spit out some spaces so that we're actually centered */
while (width > 0) { while (width > 0) {
printf(" "); printf(" ");
@ -578,9 +610,12 @@ ready:
} }
/* You have nyaned for [n] seconds! /* You have nyaned for [n] seconds!
* The \033[J ensures that the rest of the line has the dark blue * The \033[J ensures that the rest of the line has the dark blue
* background, and the \033[1;37m ensures that our text is bright white * background, and the \033[1;37m ensures that our text is bright white.
* The \033[0m prevents the Apple ][ from flipping everything, but
* makes the whole nyancat less bright on the vt220
*/ */
printf("\033[1;37mYou have nyaned for %0.0f seconds!\033[J", diff); printf("\033[1;37mYou have nyaned for %0.0f seconds!\033[J\033[0m", diff);
/* Reset the last color so that the escape sequences rewrite */ /* Reset the last color so that the escape sequences rewrite */
last = 0; last = 0;
/* Update frame crount */ /* Update frame crount */