Return to wtssendmsg.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / smartmontools / os_win32

Annotation of embedaddon/smartmontools/os_win32/wtssendmsg.c, revision 1.1

1.1     ! misho       1: /*
        !             2:  * WTSSendMessage() command line tool
        !             3:  *
        !             4:  * Home page of code is: http://smartmontools.sourceforge.net
        !             5:  *
        !             6:  * Copyright (C) 2012 Christian Franke <smartmontools-support@lists.sourceforge.net>
        !             7:  *
        !             8:  * This program is free software; you can redistribute it and/or modify
        !             9:  * it under the terms of the GNU General Public License as published by
        !            10:  * the Free Software Foundation; either version 2, or (at your option)
        !            11:  * any later version.
        !            12:  *
        !            13:  * You should have received a copy of the GNU General Public License
        !            14:  * (for example COPYING); If not, see <http://www.gnu.org/licenses/>.
        !            15:  *
        !            16:  */
        !            17: 
        !            18: #define WINVER 0x0500
        !            19: #define _WIN32_WINNT WINVER
        !            20: 
        !            21: char svnid[] = "$Id: wtssendmsg.c 3714 2012-11-24 16:34:47Z chrfranke $";
        !            22: 
        !            23: #include <stdio.h>
        !            24: #include <string.h>
        !            25: 
        !            26: #define WIN32_LEAN_AND_MEAN
        !            27: #include <windows.h>
        !            28: #include <wtsapi32.h>
        !            29: 
        !            30: 
        !            31: static int usage()
        !            32: {
        !            33:   printf("wtssendmsg $Revision: 3714 $ - Display a message box on client desktops\n"
        !            34:          "Copyright (C) 2012 Christian Franke, smartmontools.org\n\n"
        !            35:          "Usage: wtssendmsg [-cas] [-v] [\"Caption\"] \"Message\"|-\n"
        !            36:          "       wtssendmsg -v\n\n"
        !            37:          "  -c    Console session [default]\n"
        !            38:          "  -a    Active sessions\n"
        !            39:          "  -s    Connected sessions\n"
        !            40:          "  -v    List sessions\n"
        !            41:   );
        !            42:   return 1;
        !            43: }
        !            44: 
        !            45: int main(int argc, const char **argv)
        !            46: {
        !            47:   int mode = 0, verbose = 0, status = 0, i;
        !            48:   const char * message = 0, * caption = "";
        !            49:   char msgbuf[1024];
        !            50:   WTS_SESSION_INFOA * sessions; DWORD count;
        !            51: 
        !            52:   for (i = 1; i < argc && argv[i][0] == '-' && argv[i][1]; i++) {
        !            53:     int j;
        !            54:     for (j = 1; argv[i][j]; j++)
        !            55:       switch (argv[i][j]) {
        !            56:       case 'c': mode = 0; break;
        !            57:       case 'a': mode = 1; break;
        !            58:       case 's': mode = 2; break;
        !            59:       case 'v': verbose = 1; break;
        !            60:       default: return usage();
        !            61:     }
        !            62:   }
        !            63: 
        !            64:   if (i < argc) {
        !            65:     if (i+1 < argc)
        !            66:       caption = argv[i++];
        !            67: 
        !            68:     message = argv[i++];
        !            69:     if (i < argc)
        !            70:       return usage();
        !            71: 
        !            72:     if (!strcmp(message, "-")) {
        !            73:       // Read message from stdin
        !            74:       i = fread(msgbuf, 1, sizeof(msgbuf)-1, stdin);
        !            75:       if (i < 0) {
        !            76:         perror("stdin");
        !            77:         return 1;
        !            78:       }
        !            79:       msgbuf[i] = 0;
        !            80:       message = msgbuf;
        !            81:     }
        !            82:   }
        !            83:   else {
        !            84:       if (!verbose)
        !            85:         return usage();
        !            86:   }
        !            87: 
        !            88:   // Get session list
        !            89:   if (!WTSEnumerateSessionsA(WTS_CURRENT_SERVER_HANDLE, 0, 1, &sessions, &count)) {
        !            90:     fprintf(stderr, "WTSEnumerateSessions() failed\n");
        !            91:     return 1;
        !            92:   }
        !            93: 
        !            94:   for (i = 0; i < (int)count; i++) {
        !            95: 
        !            96:     if (verbose) {
        !            97:       printf("Session %d (\"%s\", State=%d)%s",
        !            98:              i, sessions[i].pWinStationName, sessions[i].State,
        !            99:              (!message ? "\n" : ": "));
        !           100:       if (!message)
        !           101:         continue; // List sessions only
        !           102:       fflush(stdout);
        !           103:     }
        !           104: 
        !           105:     if (   !strcmpi(sessions[i].pWinStationName, "Console")
        !           106:         || (mode >= 1 && sessions[i].State == WTSActive)
        !           107:         || (mode >= 2 && sessions[i].State == WTSConnected)) {
        !           108: 
        !           109:       // Send Message, don't wait for OK button
        !           110:       DWORD result;
        !           111:       if (WTSSendMessageA(WTS_CURRENT_SERVER_HANDLE, sessions[i].SessionId,
        !           112:           (char *)caption, strlen(caption),
        !           113:           (char *)message, strlen(message),
        !           114:           MB_OK|MB_ICONEXCLAMATION, 0 /*Timeout*/,
        !           115:           &result, FALSE /*!Wait*/)) {
        !           116:         if (verbose)
        !           117:           printf("message sent\n");
        !           118:       }
        !           119:       else {
        !           120:         status = 1;
        !           121:         if (verbose)
        !           122:           printf("WTSSendMessage() failed with error=%d\n", (int)GetLastError());
        !           123:         else
        !           124:           fprintf(stderr, "Session %d (\"%s\", State=%d): WTSSendMessage() failed with error=%d\n",
        !           125:                   i, sessions[i].pWinStationName, sessions[i].State, (int)GetLastError());
        !           126:       }
        !           127:     }
        !           128:     else {
        !           129:       if (verbose)
        !           130:         printf("ignored\n");
        !           131:     }
        !           132:   }
        !           133: 
        !           134:   WTSFreeMemory(sessions);
        !           135: 
        !           136:   return status;
        !           137: }