lock_library.cpp

Go to the documentation of this file.

/*****************************************************************************\
*                                                                             *
*  Name   : lock_library                                                      *
*  Author : Chris Koeritz                                                     *
*                                                                             *
*  Purpose:                                                                   *
*                                                                             *
*    Puts a stranglehold on dlls passed on the command line.  This forces     *
*  the dlls to be treated as locked and in use.                               *
*                                                                             *
*******************************************************************************
* Copyright (c) 2007-$now By Author.  This program is free software; you can  *
* redistribute it and/or modify it under the terms of the GNU General Public  *
* License as published by the Free Software Foundation; either version 2 of   *
* the License or (at your option) any later version.  This is online at:      *
*     http://www.fsf.org/copyleft/gpl.html                                    *
* Please send any updates to: fred@gruntose.com                               *
\*****************************************************************************/

#include <basis/common_templates.h>
#include <data_struct/amorph.cpp>
#include <opsystem/byte_filer.h>
#include <opsystem/console_logger.h>
#include <opsystem/dynamic_loader.h>
#include <opsystem/filename.h>
#include <opsystem/startup_code.h>

HOOPLE_STARTUP_CODE;

#define LOG(to_print) program_wide_logger().log(to_print)

int main(int argc, char *argv[])
{
  SET_DEFAULT_CONSOLE_LOGGER;
  if (argc < 2) {
    LOG("This program requires a list of DLLs on the command line.  Each of those");
    LOG("dlls will be loaded and locked into memory.  This will force them to");
    LOG("be locked while the application is still running, and hence upgrades can");
    LOG("be tested for how they behave when DLLs that are to be upgraded are");
    LOG("inaccessible.");
    return 37;  // error.
  }

  string_array dll_list;
  for (int i = 1; i < argc; i++) {
    dll_list += argv[i];
  }

  library_plugins::dynamic_library_loader loaded("garbaloy.dll", false);
    // create the dll loader but don't actually try to load the bogus name
    // above.

  amorph<byte_filer> simplistic_locks;
  for (int i = 0; i < dll_list.length(); i++) {
    LOG(istring("loading dll ") + dll_list[i]);
/* this stuff is cool but doesn't lock the dll.
    void *handle = loaded.load_library(dll_list[i].s());
    if (handle) {
      void *mainfunc = loaded.load_method(handle, MAKEINTRESOURCEA(1));
      if (!mainfunc) {
        LOG(istring("failed to find first exported method in dll ")
            + dll_list[i]);
      }
*/
    filename check_it(dll_list[i]);
    if (!check_it.good()) {
      LOG(istring("failed to find the dll ") + dll_list[i]);
      continue;
    }
    byte_filer *gubba = new byte_filer(dll_list[i], "a+b");
    if (gubba->good()) {
      simplistic_locks.append(gubba);
    } else {
      LOG(istring("failed to load dll ") + dll_list[i]);
    }
  }

  if (dll_list.length()) {
    LOG("now pausing forever.  break the program to release the dlls.");
    while (true) { portable::sleep_ms(100); }
  } else {
    LOG("didn't find any dlls to load and lock.  exiting.");
  }

  return 0;
}

---