123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126 |
- #include <u.h>
- #include <libc.h>
- #include <ip.h>
- #include <thread.h>
- #include "netbios.h"
- static struct {
- int thread;
- int die;
- QLock;
- NbnsAlarm *head;
- } alarmlist = { -1 };
- #define MaxLong ((1 << (sizeof(long) * 8 - 1)) - 1)
- void
- alarmist(void *)
- {
- for (;;) {
- vlong now;
- long snooze;
- //print("running\n");
- qlock(&alarmlist);
- if (alarmlist.die) {
- qunlock(&alarmlist);
- break;
- }
- now = nsec() / 1000000;
- while (alarmlist.head && alarmlist.head->expirems <= now) {
- //print("expiring because %lld > %lld\n", alarmlist.head->expirems, now);
- sendul(alarmlist.head->c, 1);
- alarmlist.head = alarmlist.head->next;
- }
- if (alarmlist.head) {
- vlong vsnooze = alarmlist.head->expirems - now;
- if (vsnooze > MaxLong)
- snooze = MaxLong;
- else
- snooze = vsnooze;
- qunlock(&alarmlist);
- }
- else
- snooze = 60 * 1000;
- //print("snoozing for %ld\n", snooze);
- qunlock(&alarmlist);
- sleep(snooze);
- }
- }
- NbnsAlarm *
- nbnsalarmnew(void)
- {
- NbnsAlarm *a;
- a = mallocz(sizeof(*a), 1);
- if (a == nil)
- return nil;
- a->c = chancreate(sizeof(ulong), 1);
- if (a->c == nil) {
- free(a);
- return nil;
- }
- return a;
- }
- void
- nbnsalarmcancel(NbnsAlarm *a)
- {
- NbnsAlarm **ap;
- int rv;
- qlock(&alarmlist);
- for (ap = &alarmlist.head; *ap && *ap != a; ap = &(*ap)->next)
- ;
- if (*ap) {
- *ap = a->next;
- if (ap == &alarmlist.head)
- threadint(alarmlist.thread);
- }
- qunlock(&alarmlist);
- do {
- ulong v;
- rv = nbrecv(a->c, &v);
- } while (rv != 0);
- }
- void
- nbnsalarmend(void)
- {
- qlock(&alarmlist);
- if (alarmlist.thread >= 0) {
- alarmlist.die = 1;
- threadint(alarmlist.thread);
- }
- qunlock(&alarmlist);
- }
- void
- nbnsalarmset(NbnsAlarm *a, ulong millisec)
- {
- NbnsAlarm **ap;
- nbnsalarmcancel(a);
- a->expirems = nsec() / 1000000 + millisec;
- qlock(&alarmlist);
- for (ap = &alarmlist.head; *ap; ap = &(*ap)->next)
- if (a->expirems < (*ap)->expirems)
- break;
- a->next = (*ap);
- *ap = a;
- if (alarmlist.thread < 0)
- alarmlist.thread = proccreate(alarmist, nil, 16384);
- else
- threadint(alarmlist.thread);
- qunlock(&alarmlist);
- }
- void
- nbnsalarmfree(NbnsAlarm **ap)
- {
- NbnsAlarm *a;
- a = *ap;
- if (a) {
- nbnsalarmcancel(a);
- chanfree(a->c);
- free(a);
- *ap = nil;
- }
- }
|