summaryrefslogtreecommitdiff
path: root/src/VBox/Additions/os2/VBoxReplaceDll.cpp
blob: cb07bd83c7cf95011c91dec5ffb7474ae28c95d2 (plain)
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
/** $Id: VBoxReplaceDll.cpp $ */
/** @file
 * VBoxReplaceDll - helper for replacing a dll when it's in use by the system
 */

/*
 * Copyright (C) 2013 Oracle Corporation
 *
 * This file is part of VirtualBox Open Source Edition (OSE), as
 * available from http://www.virtualbox.org. This file is free software;
 * you can redistribute it and/or modify it under the terms of the GNU
 * General Public License (GPL) as published by the Free Software
 * Foundation, in version 2 as it comes in the "COPYING" file of the
 * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
 * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
 */


/*******************************************************************************
*   Header Files                                                               *
*******************************************************************************/
#define INCL_BASE
#include <os2.h>
#include <stdio.h>
#include <string.h>


static int usage(const char *argv0)
{
    char *psz1 = strrchr(argv0, '\\');
    if (psz1)
        argv0 = psz1 + 1;
    psz1 = strrchr(argv0, '/');
    if (psz1)
        argv0 = psz1 + 1;
    psz1 = strrchr(argv0, ':');
    if (psz1)
        argv0 = psz1 + 1;

    printf("Usage: %s <dll1> [dll2 ...[dllN]]\n"
           "\n"
           "Tells the kernel to cache the specified DLLs in memory and close the\n"
           "files on disk, allowing new DLL versions to be installed.\n"
           "\n"
           "Copyright (C) 2013 Oracle Corporation\n",
           argv0);
    return 0;
}

int main(int argc, char **argv)
{
    int fOptions   = 1;
    int cProcessed = 0;
    int i;
    for (i = 1; i < argc; i++)
    {
        if (   fOptions
            && argv[i][0] == '-')
        {
            if (!strcmp(argv[i], "--"))
                fOptions = 0;
            else if (   !strcmp(argv[i], "--help")
                     || !strcmp(argv[i], "-help")
                     || !strcmp(argv[i], "-h")
                     || !strcmp(argv[i], "-?") )
                return usage(argv[0]);
            else if (   !strcmp(argv[i], "--version")
                     || !strcmp(argv[i], "-V") )
            {
                printf("$Revision: 89639 $\n");
                return 0;
            }
            else
            {
                fprintf(stderr, "syntax error: Invalid option '%s'!\n", argv[i]);
                return 2;
            }
        }
        else
        {
            /*
             * Replace the specified DLL.
             */
            APIRET rc = DosReplaceModule((PCSZ)argv[i], NULL, NULL);
            if (rc == NO_ERROR)
                printf("info: Successfully cached '%s'.\n", argv[i]);
            else
            {
                fprintf(stderr, "error: DosReplaceModule failed with rc=%u on  '%s'.\n", rc, argv[i]);
                return 1;
            }
            cProcessed++;
        }
    }

    if (cProcessed == 0)
    {
        fprintf(stderr, "syntax error: No DLLs specified. (Consult --help for usage.)\n");
        return 1;
    }

    return 0;
}