Kernel: Add Thread::block_until(Condition).

Replace the class-based snooze alarm mechanism with a per-thread callback.
This makes it easy to block the current thread on an arbitrary condition:

    void SomeDevice::wait_for_irq() {
        m_interrupted = false;
        current->block_until([this] { return m_interrupted; });
    }
    void SomeDevice::handle_irq() {
        m_interrupted = true;
    }

Use this in the SB16 driver, and in NetworkTask :^)

Kernel/Devices/SB16.cpp

@@ -49,7 +49,6 @@ static SB16* s_the;
 SB16::SB16()
     : IRQHandler(5)
     , CharacterDevice(42, 42) // ### ?
-    , m_interrupt_alarm(*this)
 {
     s_the = this;
     initialize();
@@ -144,7 +143,7 @@ void SB16::wait_for_irq()
 #ifdef SB16_DEBUG
     kprintf("SB16: waiting for interrupt...\n");
 #endif
-    current->snooze_until(m_interrupt_alarm);
+    current->block_until([this] { return m_interrupted; });
 #ifdef SB16_DEBUG
     kprintf("SB16: got interrupt!\n");
 #endif
@@ -195,8 +194,3 @@ ssize_t SB16::write(FileDescription&, const u8* data, ssize_t length)
     wait_for_irq();
     return length;
 }
-
-bool SB16InterruptAlarm::is_ringing() const
-{
-    return m_device.got_interrupt({});
-}

Kernel/Devices/SB16.h

@@ -1,7 +1,6 @@
 #pragma once
 
 #include <AK/CircularQueue.h>
-#include <Kernel/Alarm.h>
 #include <Kernel/Devices/CharacterDevice.h>
 #include <Kernel/IRQHandler.h>
 #include <Kernel/VM/PhysicalAddress.h>
@@ -9,18 +8,6 @@
 
 class SB16;
 
-class SB16InterruptAlarm final : public Alarm {
-public:
-    SB16InterruptAlarm(SB16& device)
-        : m_device(device)
-    {
-    }
-    virtual bool is_ringing() const override;
-
-private:
-    SB16& m_device;
-};
-
 class SB16 final : public IRQHandler
     , public CharacterDevice {
 public:
@@ -35,8 +22,6 @@ public:
     virtual ssize_t write(FileDescription&, const u8*, ssize_t) override;
     virtual bool can_write(FileDescription&) const override { return true; }
 
-    bool got_interrupt(Badge<SB16InterruptAlarm>) const { return m_interrupted; }
-
 private:
     // ^IRQHandler
     virtual void handle_irq() override;
@@ -54,6 +39,4 @@ private:
     RefPtr<PhysicalPage> m_dma_buffer_page;
     bool m_interrupted { false };
     int m_major_version { 0 };
-
-    SB16InterruptAlarm m_interrupt_alarm;
 };

Kernel/Net/NetworkAdapter.cpp

@@ -33,7 +33,6 @@ NetworkAdapter* NetworkAdapter::from_ipv4_address(const IPv4Address& address)
 }
 
 NetworkAdapter::NetworkAdapter()
-    : m_packet_queue_alarm(*this)
 {
     // FIXME: I wanna lock :(
     all_adapters().resource().set(this);
@@ -106,8 +105,3 @@ void NetworkAdapter::set_interface_name(const StringView& basename)
     builder.append('0');
     m_name = builder.to_string();
 }
-
-bool PacketQueueAlarm::is_ringing() const
-{
-    return m_adapter.has_queued_packets();
-}

Kernel/Net/NetworkAdapter.h

@@ -4,7 +4,6 @@
 #include <AK/Function.h>
 #include <AK/SinglyLinkedList.h>
 #include <AK/Types.h>
-#include <Kernel/Alarm.h>
 #include <Kernel/Net/ARP.h>
 #include <Kernel/Net/ICMP.h>
 #include <Kernel/Net/IPv4.h>
@@ -12,19 +11,6 @@
 
 class NetworkAdapter;
 
-class PacketQueueAlarm final : public Alarm {
-public:
-    PacketQueueAlarm(NetworkAdapter& adapter)
-        : m_adapter(adapter)
-    {
-    }
-    virtual ~PacketQueueAlarm() override {}
-    virtual bool is_ringing() const override;
-
-private:
-    NetworkAdapter& m_adapter;
-};
-
 class NetworkAdapter {
 public:
     static void for_each(Function<void(NetworkAdapter&)>);
@@ -44,8 +30,6 @@ public:
 
     ByteBuffer dequeue_packet();
 
-    Alarm& packet_queue_alarm() { return m_packet_queue_alarm; }
-
     bool has_queued_packets() const { return !m_packet_queue.is_empty(); }
 
 protected:
@@ -58,7 +42,6 @@ protected:
 private:
     MACAddress m_mac_address;
     IPv4Address m_ipv4_address;
-    PacketQueueAlarm m_packet_queue_alarm;
     SinglyLinkedList<ByteBuffer> m_packet_queue;
     String m_name;
 };

Kernel/Net/NetworkTask.cpp

@@ -33,22 +33,6 @@ Lockable<HashMap<IPv4Address, MACAddress>>& arp_table()
     return *the;
 }
 
-class CombinedPacketQueueAlarm : public Alarm {
-public:
-    CombinedPacketQueueAlarm() {}
-
-    virtual bool is_ringing() const override
-    {
-        if (LoopbackAdapter::the().has_queued_packets())
-            return true;
-        if (auto* e1000 = E1000NetworkAdapter::the()) {
-            if (e1000->has_queued_packets())
-                return true;
-        }
-        return false;
-    }
-};
-
 void NetworkTask_main()
 {
     LoopbackAdapter::the();
@@ -71,13 +55,19 @@ void NetworkTask_main()
         return {};
     };
 
-    CombinedPacketQueueAlarm queue_alarm;
-
     kprintf("NetworkTask: Enter main loop.\n");
     for (;;) {
         auto packet = dequeue_packet();
         if (packet.is_null()) {
-            current->snooze_until(queue_alarm);
+            current->block_until([] {
+                if (LoopbackAdapter::the().has_queued_packets())
+                    return true;
+                if (auto* e1000 = E1000NetworkAdapter::the()) {
+                    if (e1000->has_queued_packets())
+                        return true;
+                }
+                return false;
+            });
             continue;
         }
         if (packet.size() < (int)(sizeof(EthernetFrameHeader))) {

Kernel/Scheduler.cpp

@@ -1,5 +1,4 @@
 #include <AK/TemporaryChange.h>
-#include <Kernel/Alarm.h>
 #include <Kernel/Arch/i386/PIT.h>
 #include <Kernel/Devices/PCSpeaker.h>
 #include <Kernel/FileSystem/FileDescription.h>
@@ -159,9 +158,9 @@ bool Scheduler::pick_next()
             return IterationDecision::Continue;
         }
 
-        if (thread.state() == Thread::BlockedSnoozing) {
-            if (thread.m_snoozing_alarm->is_ringing()) {
-                thread.m_snoozing_alarm = nullptr;
+        if (thread.state() == Thread::BlockedCondition) {
+            if (thread.m_block_until_condition()) {
+                thread.m_block_until_condition = nullptr;
                 thread.unblock();
             }
             return IterationDecision::Continue;

Kernel/Thread.cpp

@@ -108,10 +108,10 @@ void Thread::unblock()
     set_state(Thread::Runnable);
 }
 
-void Thread::snooze_until(Alarm& alarm)
+void Thread::block_until(Function<bool()>&& condition)
 {
-    m_snoozing_alarm = &alarm;
-    block(Thread::BlockedSnoozing);
+    m_block_until_condition = move(condition);
+    block(Thread::BlockedCondition);
     Scheduler::yield();
 }
 
@@ -179,8 +179,8 @@ const char* to_string(Thread::State state)
         return "Connect";
     case Thread::BlockedReceive:
         return "Receive";
-    case Thread::BlockedSnoozing:
-        return "Snoozing";
+    case Thread::BlockedCondition:
+        return "Condition";
     case Thread::__Begin_Blocked_States__:
     case Thread::__End_Blocked_States__:
         break;

Kernel/Thread.h

@@ -1,6 +1,7 @@
 #pragma once
 
 #include <AK/AKString.h>
+#include <AK/Function.h>
 #include <AK/InlineLinkedList.h>
 #include <AK/OwnPtr.h>
 #include <AK/RefPtr.h>
@@ -72,7 +73,7 @@ public:
         BlockedSelect,
         BlockedConnect,
         BlockedReceive,
-        BlockedSnoozing,
+        BlockedCondition,
         __End_Blocked_States__
     };
 
@@ -102,7 +103,7 @@ public:
 
     void set_wakeup_time(u64 t) { m_wakeup_time = t; }
     u64 wakeup_time() const { return m_wakeup_time; }
-    void snooze_until(Alarm&);
+    void block_until(Function<bool()>&&);
     KResult wait_for_connect(FileDescription&);
 
     const FarPtr& far_ptr() const { return m_far_ptr; }
@@ -187,7 +188,7 @@ private:
     timeval m_select_timeout;
     SignalActionData m_signal_action_data[32];
     Region* m_signal_stack_user_region { nullptr };
-    Alarm* m_snoozing_alarm { nullptr };
+    Function<bool()> m_block_until_condition;
     Vector<int> m_select_read_fds;
     Vector<int> m_select_write_fds;
     Vector<int> m_select_exceptional_fds;