Improve egg, again

Author: UEFI-code

drivers/tty/serial/serial_core.c

@@ -580,7 +580,7 @@ static void uart_flush_chars(struct tty_struct *tty)
 	uart_start(tty);
 }
 
-static ssize_t uart_write(struct tty_struct *tty, const u8 *buf, size_t count)
+ssize_t uart_write233(struct tty_struct *tty, const u8 *buf, size_t count, uint8_t flush)
 {
 	struct uart_state *state = tty->driver_data;
 	struct uart_port *port;
@@ -614,12 +614,17 @@ static ssize_t uart_write(struct tty_struct *tty, const u8 *buf, size_t count)
 		count -= c;
 		ret += c;
 	}
-
-	__uart_start(state);
+	
+	if (flush) __uart_start(state);
 	uart_port_unlock(port, flags);
 	return ret;
 }
 
+static __inline ssize_t uart_write(struct tty_struct *tty, const u8 *buf, size_t count)
+{
+	return uart_write233(tty, buf, count, 1);
+}
+
 static unsigned int uart_write_room(struct tty_struct *tty)
 {
 	struct uart_state *state = tty->driver_data;

include/linux/tty_driver.h

@@ -16,6 +16,8 @@ struct tty_driver;
 struct serial_icounter_struct;
 struct serial_struct;
 
+extern ssize_t uart_write233(struct tty_struct *tty, const u8 *buf, size_t count, uint8_t flush);
+
 /**
  * struct tty_operations -- interface between driver and tty
  *

kernel/sched/easy.c

@@ -1,6 +1,7 @@
 #include <linux/sched/clock.h>
 #include <linux/sched/cputime.h>
 #include <linux/timekeeper_internal.h>
+#include <linux/tty.h>
 #include <linux/sched/isolation.h>
 #include <linux/sched/nohz.h>
 #include "sched.h"
@@ -30,7 +31,7 @@ typedef struct {
 easy_sched_struct_def easy_cpu_contexts[128] = {0};
 
 static const char *unix_days[] = {"Thursday", "Friday", "Saturday", "Sunday", "Monday", "Tuesday", "Wednesday"};
-static const char *prof_words[] = {"You can't just delete that!", "Your math is too poor!", "Without the formula, it's meaningless!", "Go back and study linear algebra!", "This is not rigorous.", "We need publish papers!", "Intuition won't work here!", "This is not scientific research!", "You are running blind code!"};
+static const char *prof_words[] = {"\033[91mYou can't just delete that!\033[0m", "\033[93mYour math is too poor!\033[0m", "\033[95mWithout the formula, it's meaningless!\033[0m", "\033[91mGo back and study linear algebra!\033[0m", "\033[96mThis is not rigorous.\033[0m", "\033[96mWe need publish papers!\033[0m", "\033[93mIntuition won't work here!\033[0m", "\033[95mThis is not scientific research!\033[0m", "\033[91mYou are running blind code!\033[0m"};
 static __inline void egg(void)
 {
     static uint8_t shown = 0;
@@ -49,11 +50,15 @@ static __inline void egg2(struct task_struct *p)
 {
     static uint64_t last_yield = 0;
     if (global_timekeeper.ktime_sec-last_yield<30) return;
-    if (p->utime%255 > 233)
+    if (p->utime%255 < 233) return;
+    if (p->signal && p->signal->tty)
     {
-        pr_alert("\033[1;31mWarning from Professors with PID=%lld: %s\033[0m\n", p->pid, prof_words[get_cycles()%(sizeof(prof_words)/sizeof(char *))]);
-        last_yield = global_timekeeper.ktime_sec;
+        char msg_buf[256] = {0}; uint8_t msg_len = 0;
+        msg_len = sprintf(msg_buf, "\033[1;33mWarning from Prof with %s(PID=%lld):\033[0m %s\n", p->comm, p->pid, prof_words[get_cycles()%(sizeof(prof_words)/sizeof(char *))]);
+        uart_write233(p->signal->tty, msg_buf, msg_len, 0); // no flush, prevent hardware corruption
     }
+    else pr_alert("\033[1;31mProf was ANGRY with %s(PID=%lld):\033[0m %s\n", p->comm, p->pid, prof_words[get_cycles()%(sizeof(prof_words)/sizeof(char *))]);
+    last_yield = global_timekeeper.ktime_sec;
 }
 
 void easy_sched_init(easy_sched_struct_def *easy_context)