shutdown/main.c at master · superfury/shutdown · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
/*
compile with:
tcc -mt -tDc main.c
*/
#include <stdio.h>

// APM installation check
char chkAPM() {
  asm {
    mov ax, 5300h
    xor bx, bx
    int 15h
    jc APM_error
  }
  return 0;
APM_error:
  return -1;
}

void usage() {
  printf("shutdown for DOS\r\n"
         "Cesar Gimenes, @crgimenes\r\n"
         "https://github.com/crgimenes/shutdown\r\n\r\n"
         "Usage:\r\n"
         "\t-h halt require APM\r\n"
         "\t-s suspend require APM\r\n"
         "\t-S stand-by require APM\r\n"
         "\t-R ready require APM\r\n"
      "\t-r reboot (call BIOS int 19h)\r\n");
}

int main(int argc, char *argv[]) {
  unsigned int haltstate;
  if (argc == 1) {
    usage();
    return 1;
  }

  if (strcmp(argv[1], "-r") == 0) {
    asm int 19h
  }

  if (chkAPM()) {
    printf("APM error or not available");
    return 1;
  }

  if (strcmp(argv[1], "-h") == 0) { //Halt?
      haltstate = 3; //State #3: off!
  }
  else if (strcmp(argv[1], "-s") == 0) { //Suspend?
      haltstate = 2; //State #2: suspend!
  }
  else if (strcmp(argv[1], "-S") == 0) { //Stand-by?
      haltstate = 1; //State #1: stand-by!
  }
  else if (strcmp(argv[1], "-R") == 0) { //Ready?
      haltstate = 0; //State #0: ready!
  }
  else {
      haltstate = 4; //Default: not used!
  }
  if (haltstate < 4) { //Valid to set?
      asm{
      /* connect to APM */
      mov     ax, 5301h
      xor     bx, bx
      int     15h

      /* set APM version */
      mov     ax, 530Eh
      mov     cx, 0102h
      xor     bx, bx
      int     15h

      /* shutdown */
      mov     ax, 5307h
      mov     bx, 0001h
      mov     cx,haltstate /* Halt state to set */
      mov     ch, 0
      int     15h
      hlt
    }
  }

  return 0;
}