Źródło

msgtool

Następujący kod źródłowy to msgtool. Powinien skompilować się bez problemu na nowszych ( przyzwoitych ) wersjach jądra, które wspierają System V IPC. Pamiętaj o włączeniu IPC Systemu V podczas rekompilacji jądra!

Program utowrzy kolejkę wiadomości jeżeli jeszcze nie istnieje, bez względu na rodzaj żądanego działania.

UWAGA: Ponieważ narzędzie to używa funkcji ftok() do wygenerowania klucza IPC możesz zaobserwować konflikty katalogów. Jeżeli gdziekolwiek w swoim skrypcie zmienisz katalog prawdopodobnie program nie zadziała. Rozwiązaniem mogłoby być zakodowanie ścieżki wewnątrz programu ( np.: "/tmp/msgtool" ), lub przekazywanie ścieżki jako parametru wywołania.

  /*****************************************************************************
   Zaczerpnięto z "Linux Programmer's Guide - Rozdział 6"
   (C)opyright 1994-1995, Scott Burkett
   ***************************************************************************** 
   MODUŁ: msgtool.c
   *****************************************************************************
   Narzędzie linii poleceń do majstrowania z kolejkami wiadomości stylu SysV
   *****************************************************************************/
  
  #include <stdio.h>
  #include <stdlib.h>
  #include <ctype.h>
  #include <sys/types.h>
  #include <sys/ipc.h>
  #include <sys/msg.h>
  
  #define MAX_SEND_SIZE 80
  
  struct mymsgbuf {
          long mtype;
          char mtext[MAX_SEND_SIZE];
  };
  
  void send_message(int qid, struct mymsgbuf *qbuf, long type, char *text);
  void read_message(int qid, struct mymsgbuf *qbuf, long type);
  void remove_queue(int qid);
  void change_queue_mode(int qid, char *mode);
  void usage(void);
  
  
  int main(int argc, char *argv[])
  {
          key_t key;
          int   msgqueue_id;
          struct mymsgbuf qbuf;
  
          if(argc == 1)
                  usage();
  
          /* utwórz unikalny klucz poprzez wywołanie ftok() */
          key = ftok(".", 'm');
  
          /* otwórz kolejkę - utwórz jeżeli trzeba */
          if((msgqueue_id = msgget(key, IPC_CREAT|0660)) == -1) {
                  perror("msgget");
                  exit(1);
          }
          
          switch(tolower(argv[1][0]))
          {
                  case 'w': send_message(msgqueue_id, (struct mymsgbuf *)&qbuf,
                                         atol(argv[2]), argv[3]); 
                            break;
                  case 'o': read_message(msgqueue_id, &qbuf, atol(argv[2])); 
                            break;
                  case 'u': remove_queue(msgqueue_id); 
                            break;        
                  case 't': change_queue_mode(msgqueue_id, argv[2]); 
                            break;
  
                   default: usage();
  
          }
          
          return(0);
  }
  
  void send_message(int qid, struct mymsgbuf *qbuf, long type, char *text)
  {
          /* wyślij wiadomość do kolejki */
          printf("Wysyłanie wiadomości ...\n");
          qbuf->mtype = type;
          strcpy(qbuf->mtext, text);
  
          if((msgsnd(qid, (struct msgbuf *)qbuf,
                  strlen(qbuf->mtext)+1, 0)) ==-1)
          {
                  perror("msgsnd");
                  exit(1);
          }
  }
  
  void read_message(int qid, struct mymsgbuf *qbuf, long type)
  {
          /* odczytaj wiadomość z kolejki */
          printf("Odczytywanie wiadomości ...\n");
          qbuf->mtype = type;
          msgrcv(qid, (struct msgbuf *)qbuf, MAX_SEND_SIZE, type, 0);
          
          printf("Typ: %ld Tekst: %s\n", qbuf->mtype, qbuf->mtext);
  }
  
  void remove_queue(int qid)
  {
          /* Remove the queue */
          msgctl(qid, IPC_RMID, 0);
  }
  
  void change_queue_mode(int qid, char *mode)
  {
          struct msqid_ds myqueue_ds;
  
          /* pobierz informacje */
          msgctl(qid, IPC_STAT, &myqueue_ds);
  
          /* konwertuj i załaduj tryb */
          sscanf(mode, "%ho", &myqueue_ds.msg_perm.mode);
  
          /* uaktualnij tryb */
          msgctl(qid, IPC_SET, &myqueue_ds);
  }
  
  void usage(void)
  {
          fprintf(stderr, "msgtool - Narzędzie do majstrowania przy kolejkach msg\n");
          fprintf(stderr, "\nUŻYCIE: msgtool (w)yślij <typ> <tekst>\n");
          fprintf(stderr, "                (o)dbierz <typ>\n");
          fprintf(stderr, "                (u)suń\n");
          fprintf(stderr, "                (t)ryb <ósemkowy tryb>\n");
          exit(1);
  }