123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999 |
- /*
- * Copyright (c) Contributors, http://opensimulator.org/
- * See CONTRIBUTORS.TXT for a full list of copyright holders.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions are met:
- * * Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- * * Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in the
- * documentation and/or other materials provided with the distribution.
- * * Neither the name of the OpenSimulator Project nor the
- * names of its contributors may be used to endorse or promote products
- * derived from this software without specific prior written permission.
- *
- * THIS SOFTWARE IS PROVIDED BY THE DEVELOPERS ``AS IS'' AND ANY
- * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
- * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
- * DISCLAIMED. IN NO EVENT SHALL THE CONTRIBUTORS BE LIABLE FOR ANY
- * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
- * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
- * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
- * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
- * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- */
- using System;
- using System.Collections.Generic;
- using System.Linq;
- using System.Reflection;
- using System.Threading;
- using OpenMetaverse;
- using log4net;
- using log4net.Appender;
- using log4net.Core;
- using log4net.Repository;
- using Nini.Config;
- using OpenSim.Framework;
- using OpenSim.Framework.Console;
- using OpenSim.Framework.Monitoring;
- using pCampBot.Interfaces;
- namespace pCampBot
- {
- public enum BotManagerBotConnectingState
- {
- Initializing,
- Ready,
- Connecting,
- Disconnecting
- }
- /// <summary>
- /// Thread/Bot manager for the application
- /// </summary>
- public class BotManager
- {
- private static readonly ILog m_log = LogManager.GetLogger(MethodBase.GetCurrentMethod().DeclaringType);
- public const int DefaultLoginDelay = 5000;
- /// <summary>
- /// Is pCampbot ready to connect or currently in the process of connecting or disconnecting bots?
- /// </summary>
- public BotManagerBotConnectingState BotConnectingState { get; private set; }
- /// <summary>
- /// Used to control locking as we can't lock an enum.
- /// </summary>
- private object BotConnectingStateChangeObject = new object();
- /// <summary>
- /// Delay between logins of multiple bots.
- /// </summary>
- /// <remarks>TODO: This value needs to be configurable by a command line argument.</remarks>
- public int LoginDelay { get; set; }
- /// <summary>
- /// Command console
- /// </summary>
- protected CommandConsole m_console;
- /// <summary>
- /// Controls whether bots start out sending agent updates on connection.
- /// </summary>
- public bool InitBotSendAgentUpdates { get; set; }
- /// <summary>
- /// Controls whether bots request textures for the object information they receive
- /// </summary>
- public bool InitBotRequestObjectTextures { get; set; }
- /// <summary>
- /// Created bots, whether active or inactive.
- /// </summary>
- protected List<Bot> m_bots;
- /// <summary>
- /// Random number generator.
- /// </summary>
- public Random Rng { get; private set; }
- /// <summary>
- /// Track the assets we have and have not received so we don't endlessly repeat requests.
- /// </summary>
- public Dictionary<UUID, bool> AssetsReceived { get; private set; }
- /// <summary>
- /// The regions that we know about.
- /// </summary>
- public Dictionary<ulong, GridRegion> RegionsKnown { get; private set; }
- /// <summary>
- /// First name for bots
- /// </summary>
- private string m_firstName;
- /// <summary>
- /// Last name stem for bots
- /// </summary>
- private string m_lastNameStem;
- /// <summary>
- /// Password for bots
- /// </summary>
- private string m_password;
- /// <summary>
- /// Login URI for bots.
- /// </summary>
- private string m_loginUri;
- /// <summary>
- /// Start location for bots.
- /// </summary>
- private string m_startUri;
- /// <summary>
- /// Postfix bot number at which bot sequence starts.
- /// </summary>
- private int m_fromBotNumber;
- /// <summary>
- /// Wear setting for bots.
- /// </summary>
- private string m_wearSetting;
- /// <summary>
- /// Behaviour switches for bots.
- /// </summary>
- private HashSet<string> m_defaultBehaviourSwitches = new HashSet<string>();
- /// <summary>
- /// Collects general information on this server (which reveals this to be a misnamed class).
- /// </summary>
- private ServerStatsCollector m_serverStatsCollector;
- /// <summary>
- /// Constructor Creates MainConsole.Instance to take commands and provide the place to write data
- /// </summary>
- public BotManager()
- {
- // We set this to avoid issues with bots running out of HTTP connections if many are run from a single machine
- // to multiple regions.
- Settings.MAX_HTTP_CONNECTIONS = int.MaxValue;
- // System.Threading.ThreadPool.SetMaxThreads(600, 240);
- //
- // int workerThreads, iocpThreads;
- // System.Threading.ThreadPool.GetMaxThreads(out workerThreads, out iocpThreads);
- // Console.WriteLine("ThreadPool.GetMaxThreads {0} {1}", workerThreads, iocpThreads);
- InitBotSendAgentUpdates = true;
- InitBotRequestObjectTextures = true;
- LoginDelay = DefaultLoginDelay;
- Rng = new Random(Environment.TickCount);
- AssetsReceived = new Dictionary<UUID, bool>();
- RegionsKnown = new Dictionary<ulong, GridRegion>();
- m_console = CreateConsole();
- MainConsole.Instance = m_console;
- // Make log4net see the console
- //
- ILoggerRepository repository = LogManager.GetRepository();
- IAppender[] appenders = repository.GetAppenders();
- OpenSimAppender consoleAppender = null;
- foreach (IAppender appender in appenders)
- {
- if (appender.Name == "Console")
- {
- consoleAppender = (OpenSimAppender)appender;
- consoleAppender.Console = m_console;
- break;
- }
- }
- m_console.Commands.AddCommand(
- "Bots", false, "shutdown", "shutdown", "Shutdown bots and exit", HandleShutdown);
- m_console.Commands.AddCommand(
- "Bots", false, "quit", "quit", "Shutdown bots and exit", HandleShutdown);
- m_console.Commands.AddCommand(
- "Bots", false, "connect", "connect [<n>]", "Connect bots",
- "If an <n> is given, then the first <n> disconnected bots by postfix number are connected.\n"
- + "If no <n> is given, then all currently disconnected bots are connected.",
- HandleConnect);
- m_console.Commands.AddCommand(
- "Bots", false, "disconnect", "disconnect [<n>]", "Disconnect bots",
- "Disconnecting bots will interupt any bot connection process, including connection on startup.\n"
- + "If an <n> is given, then the last <n> connected bots by postfix number are disconnected.\n"
- + "If no <n> is given, then all currently connected bots are disconnected.",
- HandleDisconnect);
- m_console.Commands.AddCommand(
- "Bots", false, "add behaviour", "add behaviour <abbreviated-name> [<bot-number>]",
- "Add a behaviour to a bot",
- "If no bot number is specified then behaviour is added to all bots.\n"
- + "Can be performed on connected or disconnected bots.",
- HandleAddBehaviour);
- m_console.Commands.AddCommand(
- "Bots", false, "remove behaviour", "remove behaviour <abbreviated-name> [<bot-number>]",
- "Remove a behaviour from a bot",
- "If no bot number is specified then behaviour is added to all bots.\n"
- + "Can be performed on connected or disconnected bots.",
- HandleRemoveBehaviour);
- m_console.Commands.AddCommand(
- "Bots", false, "sit", "sit", "Sit all bots on the ground.",
- HandleSit);
- m_console.Commands.AddCommand(
- "Bots", false, "stand", "stand", "Stand all bots.",
- HandleStand);
- m_console.Commands.AddCommand(
- "Bots", false, "set bots", "set bots <key> <value>", "Set a setting for all bots.", HandleSetBots);
- m_console.Commands.AddCommand(
- "Bots", false, "show regions", "show regions", "Show regions known to bots", HandleShowRegions);
- m_console.Commands.AddCommand(
- "Bots", false, "show bots", "show bots", "Shows the status of all bots.", HandleShowBotsStatus);
- m_console.Commands.AddCommand(
- "Bots", false, "show bot", "show bot <bot-number>",
- "Shows the detailed status and settings of a particular bot.", HandleShowBotStatus);
- m_console.Commands.AddCommand(
- "Debug",
- false,
- "debug lludp packet",
- "debug lludp packet <level> <avatar-first-name> <avatar-last-name>",
- "Turn on received packet logging.",
- "If level > 0 then all received packets that are not duplicates are logged.\n"
- + "If level <= 0 then no received packets are logged.",
- HandleDebugLludpPacketCommand);
- m_console.Commands.AddCommand(
- "Bots", false, "show status", "show status", "Shows pCampbot status.", HandleShowStatus);
- m_bots = new List<Bot>();
- Watchdog.Enabled = true;
- StatsManager.RegisterConsoleCommands(m_console);
- m_serverStatsCollector = new ServerStatsCollector();
- m_serverStatsCollector.Initialise(null);
- m_serverStatsCollector.Enabled = true;
- m_serverStatsCollector.Start();
- BotConnectingState = BotManagerBotConnectingState.Ready;
- }
- /// <summary>
- /// Startup number of bots specified in the starting arguments
- /// </summary>
- /// <param name="botcount">How many bots to start up</param>
- /// <param name="cs">The configuration for the bots to use</param>
- public void CreateBots(int botcount, IConfig startupConfig)
- {
- m_firstName = startupConfig.GetString("firstname");
- m_lastNameStem = startupConfig.GetString("lastname");
- m_password = startupConfig.GetString("password");
- m_loginUri = startupConfig.GetString("loginuri");
- m_fromBotNumber = startupConfig.GetInt("from", 0);
- m_wearSetting = startupConfig.GetString("wear", "no");
- m_startUri = ParseInputStartLocationToUri(startupConfig.GetString("start", "last"));
- Array.ForEach<string>(
- startupConfig.GetString("behaviours", "p").Split(new char[] { ',' }), b => m_defaultBehaviourSwitches.Add(b));
- if(botcount == 1)
- {
- lock (m_bots)
- {
- CreateBot(
- this,
- CreateBehavioursFromAbbreviatedNames(m_defaultBehaviourSwitches),
- m_firstName, m_lastNameStem, m_password, m_loginUri, m_startUri, m_wearSetting);
- }
- }
- else
- {
- for (int i = 0; i < botcount; i++)
- {
- lock (m_bots)
- {
- string lastName = string.Format("{0}{1}", m_lastNameStem, i + m_fromBotNumber);
- CreateBot(
- this,
- CreateBehavioursFromAbbreviatedNames(m_defaultBehaviourSwitches),
- m_firstName, lastName, m_password, m_loginUri, m_startUri, m_wearSetting);
- }
- }
- }
- }
- private List<IBehaviour> CreateBehavioursFromAbbreviatedNames(HashSet<string> abbreviatedNames)
- {
- // We must give each bot its own list of instantiated behaviours since they store state.
- List<IBehaviour> behaviours = new List<IBehaviour>();
- // Hard-coded for now
- foreach (string abName in abbreviatedNames)
- {
- IBehaviour newBehaviour = null;
- if (abName == "c")
- newBehaviour = new CrossBehaviour();
- if (abName == "g")
- newBehaviour = new GrabbingBehaviour();
- if (abName == "n")
- newBehaviour = new NoneBehaviour();
- if (abName == "p")
- newBehaviour = new PhysicsBehaviour();
- if (abName == "t")
- newBehaviour = new TeleportBehaviour();
- if (abName == "tw")
- newBehaviour = new TwitchyBehaviour();
- if (abName == "ph2")
- newBehaviour = new PhysicsBehaviour2();
- if (abName == "inv")
- newBehaviour = new InventoryDownloadBehaviour();
- if (newBehaviour != null)
- {
- behaviours.Add(newBehaviour);
- }
- else
- {
- MainConsole.Instance.Output("No behaviour with abbreviated name {0} found", null, abName);
- }
- }
- return behaviours;
- }
- public void ConnectBots(int botcount)
- {
- lock (BotConnectingStateChangeObject)
- {
- if (BotConnectingState != BotManagerBotConnectingState.Ready)
- {
- MainConsole.Instance.Output(
- "Bot connecting status is {0}. Please wait for previous process to complete.", null, BotConnectingState);
- return;
- }
- BotConnectingState = BotManagerBotConnectingState.Connecting;
- }
- Thread connectBotThread = new Thread(o => ConnectBotsInternal(botcount));
- connectBotThread.Name = "Bots connection thread";
- connectBotThread.Start();
- }
- private void ConnectBotsInternal(int botCount)
- {
- m_log.InfoFormat(
- "[BOT MANAGER]: Starting {0} bots connecting to {1}, location {2}, named {3} {4}_<n>",
- botCount,
- m_loginUri,
- m_startUri,
- m_firstName,
- m_lastNameStem);
- m_log.DebugFormat("[BOT MANAGER]: Delay between logins is {0}ms", LoginDelay);
- m_log.DebugFormat("[BOT MANAGER]: BotsSendAgentUpdates is {0}", InitBotSendAgentUpdates);
- m_log.DebugFormat("[BOT MANAGER]: InitBotRequestObjectTextures is {0}", InitBotRequestObjectTextures);
- List<Bot> botsToConnect = new List<Bot>();
- lock (m_bots)
- {
- foreach (Bot bot in m_bots)
- {
- if (bot.ConnectionState == ConnectionState.Disconnected)
- botsToConnect.Add(bot);
- if (botsToConnect.Count >= botCount)
- break;
- }
- }
- foreach (Bot bot in botsToConnect)
- {
- lock (BotConnectingStateChangeObject)
- {
- if (BotConnectingState != BotManagerBotConnectingState.Connecting)
- {
- MainConsole.Instance.Output(
- "[BOT MANAGER]: Aborting bot connection due to user-initiated disconnection");
- return;
- }
- }
- bot.Connect();
- // Stagger logins
- Thread.Sleep(LoginDelay);
- }
- lock (BotConnectingStateChangeObject)
- {
- if (BotConnectingState == BotManagerBotConnectingState.Connecting)
- BotConnectingState = BotManagerBotConnectingState.Ready;
- }
- }
- /// <summary>
- /// Parses the command line start location to a start string/uri that the login mechanism will recognize.
- /// </summary>
- /// <returns>
- /// The input start location to URI.
- /// </returns>
- /// <param name='startLocation'>
- /// Start location.
- /// </param>
- private string ParseInputStartLocationToUri(string startLocation)
- {
- if (startLocation == "home" || startLocation == "last")
- return startLocation;
- string regionName;
- // Just a region name or only one (!) extra component. Like a viewer, we will stick 128/128/0 on the end
- Vector3 startPos = new Vector3(128, 128, 0);
- string[] startLocationComponents = startLocation.Split('/');
- regionName = startLocationComponents[0];
- if (startLocationComponents.Length >= 2)
- {
- float.TryParse(startLocationComponents[1], out startPos.X);
- if (startLocationComponents.Length >= 3)
- {
- float.TryParse(startLocationComponents[2], out startPos.Y);
- if (startLocationComponents.Length >= 4)
- float.TryParse(startLocationComponents[3], out startPos.Z);
- }
- }
- return string.Format("uri:{0}&{1}&{2}&{3}", regionName, startPos.X, startPos.Y, startPos.Z);
- }
- /// <summary>
- /// This creates a bot but does not start it.
- /// </summary>
- /// <param name="bm"></param>
- /// <param name="behaviours">Behaviours for this bot to perform.</param>
- /// <param name="firstName">First name</param>
- /// <param name="lastName">Last name</param>
- /// <param name="password">Password</param>
- /// <param name="loginUri">Login URI</param>
- /// <param name="startLocation">Location to start the bot. Can be "last", "home" or a specific sim name.</param>
- /// <param name="wearSetting"></param>
- public void CreateBot(
- BotManager bm, List<IBehaviour> behaviours,
- string firstName, string lastName, string password, string loginUri, string startLocation, string wearSetting)
- {
- MainConsole.Instance.Output(
- "[BOT MANAGER]: Creating bot {0} {1}, behaviours are {2}",
- null,
- firstName, lastName, string.Join(",", behaviours.ConvertAll<string>(b => b.Name).ToArray()));
- Bot pb = new Bot(bm, behaviours, firstName, lastName, password, startLocation, loginUri);
- pb.wear = wearSetting;
- pb.Client.Settings.SEND_AGENT_UPDATES = InitBotSendAgentUpdates;
- pb.RequestObjectTextures = InitBotRequestObjectTextures;
- pb.OnConnected += handlebotEvent;
- pb.OnDisconnected += handlebotEvent;
- m_bots.Add(pb);
- }
- /// <summary>
- /// High level connnected/disconnected events so we can keep track of our threads by proxy
- /// </summary>
- /// <param name="callbot"></param>
- /// <param name="eventt"></param>
- private void handlebotEvent(Bot callbot, EventType eventt)
- {
- switch (eventt)
- {
- case EventType.CONNECTED:
- {
- m_log.Info("[" + callbot.FirstName + " " + callbot.LastName + "]: Connected");
- break;
- }
- case EventType.DISCONNECTED:
- {
- m_log.Info("[" + callbot.FirstName + " " + callbot.LastName + "]: Disconnected");
- break;
- }
- }
- }
- /// <summary>
- /// Standard CreateConsole routine
- /// </summary>
- /// <returns></returns>
- protected CommandConsole CreateConsole()
- {
- return new LocalConsole("pCampbot");
- }
- private void HandleConnect(string module, string[] cmd)
- {
- lock (m_bots)
- {
- int botsToConnect;
- int disconnectedBots = m_bots.Count(b => b.ConnectionState == ConnectionState.Disconnected);
- if (cmd.Length == 1)
- {
- botsToConnect = disconnectedBots;
- }
- else
- {
- if (!ConsoleUtil.TryParseConsoleNaturalInt(MainConsole.Instance, cmd[1], out botsToConnect))
- return;
- botsToConnect = Math.Min(botsToConnect, disconnectedBots);
- }
- MainConsole.Instance.Output("Connecting {0} bots", null, botsToConnect);
- ConnectBots(botsToConnect);
- }
- }
- private void HandleAddBehaviour(string module, string[] cmd)
- {
- if (cmd.Length < 3 || cmd.Length > 4)
- {
- MainConsole.Instance.Output("Usage: add behaviour <abbreviated-behaviour> [<bot-number>]");
- return;
- }
- string rawBehaviours = cmd[2];
- List<Bot> botsToEffect = new List<Bot>();
- if (cmd.Length == 3)
- {
- lock (m_bots)
- botsToEffect.AddRange(m_bots);
- }
- else
- {
- int botNumber;
- if (!ConsoleUtil.TryParseConsoleNaturalInt(MainConsole.Instance, cmd[3], out botNumber))
- return;
- Bot bot = GetBotFromNumber(botNumber);
- if (bot == null)
- {
- MainConsole.Instance.Output("Error: No bot found with number {0}", null, botNumber);
- return;
- }
- botsToEffect.Add(bot);
- }
- HashSet<string> rawAbbreviatedSwitchesToAdd = new HashSet<string>();
- Array.ForEach<string>(rawBehaviours.Split(new char[] { ',' }), b => rawAbbreviatedSwitchesToAdd.Add(b));
- foreach (Bot bot in botsToEffect)
- {
- List<IBehaviour> behavioursAdded = new List<IBehaviour>();
- foreach (IBehaviour behaviour in CreateBehavioursFromAbbreviatedNames(rawAbbreviatedSwitchesToAdd))
- {
- if (bot.AddBehaviour(behaviour))
- behavioursAdded.Add(behaviour);
- }
- MainConsole.Instance.Output(
- "Added behaviours {0} to bot {1}",
- null,
- string.Join(", ", behavioursAdded.ConvertAll<string>(b => b.Name).ToArray()), bot.Name);
- }
- }
- private void HandleRemoveBehaviour(string module, string[] cmd)
- {
- if (cmd.Length < 3 || cmd.Length > 4)
- {
- MainConsole.Instance.Output("Usage: remove behaviour <abbreviated-behaviour> [<bot-number>]");
- return;
- }
- string rawBehaviours = cmd[2];
- List<Bot> botsToEffect = new List<Bot>();
- if (cmd.Length == 3)
- {
- lock (m_bots)
- botsToEffect.AddRange(m_bots);
- }
- else
- {
- int botNumber;
- if (!ConsoleUtil.TryParseConsoleNaturalInt(MainConsole.Instance, cmd[3], out botNumber))
- return;
- Bot bot = GetBotFromNumber(botNumber);
- if (bot == null)
- {
- MainConsole.Instance.Output("Error: No bot found with number {0}", null, botNumber);
- return;
- }
- botsToEffect.Add(bot);
- }
- HashSet<string> abbreviatedBehavioursToRemove = new HashSet<string>();
- Array.ForEach<string>(rawBehaviours.Split(new char[] { ',' }), b => abbreviatedBehavioursToRemove.Add(b));
- foreach (Bot bot in botsToEffect)
- {
- List<IBehaviour> behavioursRemoved = new List<IBehaviour>();
- foreach (string b in abbreviatedBehavioursToRemove)
- {
- IBehaviour behaviour;
- if (bot.TryGetBehaviour(b, out behaviour))
- {
- bot.RemoveBehaviour(b);
- behavioursRemoved.Add(behaviour);
- }
- }
- MainConsole.Instance.Output(
- "Removed behaviours {0} from bot {1}",
- null,
- string.Join(", ", behavioursRemoved.ConvertAll<string>(b => b.Name).ToArray()), bot.Name);
- }
- }
- private void HandleDisconnect(string module, string[] cmd)
- {
- List<Bot> connectedBots;
- int botsToDisconnectCount;
- lock (m_bots)
- connectedBots = m_bots.FindAll(b => b.ConnectionState == ConnectionState.Connected);
- if (cmd.Length == 1)
- {
- botsToDisconnectCount = connectedBots.Count;
- }
- else
- {
- if (!ConsoleUtil.TryParseConsoleNaturalInt(MainConsole.Instance, cmd[1], out botsToDisconnectCount))
- return;
- botsToDisconnectCount = Math.Min(botsToDisconnectCount, connectedBots.Count);
- }
- lock (BotConnectingStateChangeObject)
- BotConnectingState = BotManagerBotConnectingState.Disconnecting;
- Thread disconnectBotThread = new Thread(o => DisconnectBotsInternal(connectedBots, botsToDisconnectCount));
- disconnectBotThread.Name = "Bots disconnection thread";
- disconnectBotThread.Start();
- }
- private void DisconnectBotsInternal(List<Bot> connectedBots, int disconnectCount)
- {
- MainConsole.Instance.Output("Disconnecting {0} bots", null, disconnectCount);
- int disconnectedBots = 0;
- for (int i = connectedBots.Count - 1; i >= 0; i--)
- {
- if (disconnectedBots >= disconnectCount)
- break;
- Bot thisBot = connectedBots[i];
- if (thisBot.ConnectionState == ConnectionState.Connected)
- {
- ThreadPool.QueueUserWorkItem(o => thisBot.Disconnect());
- disconnectedBots++;
- }
- }
- lock (BotConnectingStateChangeObject)
- BotConnectingState = BotManagerBotConnectingState.Ready;
- }
- private void HandleSit(string module, string[] cmd)
- {
- lock (m_bots)
- {
- foreach (Bot bot in m_bots)
- {
- if (bot.ConnectionState == ConnectionState.Connected)
- {
- MainConsole.Instance.Output("Sitting bot {0} on ground.", null, bot.Name);
- bot.SitOnGround();
- }
- }
- }
- }
- private void HandleStand(string module, string[] cmd)
- {
- lock (m_bots)
- {
- foreach (Bot bot in m_bots)
- {
- if (bot.ConnectionState == ConnectionState.Connected)
- {
- MainConsole.Instance.Output("Standing bot {0} from ground.", null, bot.Name);
- bot.Stand();
- }
- }
- }
- }
- private void HandleShutdown(string module, string[] cmd)
- {
- lock (m_bots)
- {
- int connectedBots = m_bots.Count(b => b.ConnectionState == ConnectionState.Connected);
- if (connectedBots > 0)
- {
- MainConsole.Instance.Output("Please disconnect {0} connected bots first", null, connectedBots);
- return;
- }
- }
- MainConsole.Instance.Output("Shutting down");
- m_serverStatsCollector.Close();
- Environment.Exit(0);
- }
- private void HandleSetBots(string module, string[] cmd)
- {
- string key = cmd[2];
- string rawValue = cmd[3];
- if (key == "SEND_AGENT_UPDATES")
- {
- bool newSendAgentUpdatesSetting;
- if (!ConsoleUtil.TryParseConsoleBool(MainConsole.Instance, rawValue, out newSendAgentUpdatesSetting))
- return;
- MainConsole.Instance.Output("Setting SEND_AGENT_UPDATES to {0} for all bots",
- null, newSendAgentUpdatesSetting);
- lock (m_bots)
- m_bots.ForEach(b => b.Client.Settings.SEND_AGENT_UPDATES = newSendAgentUpdatesSetting);
- }
- else
- {
- MainConsole.Instance.Output("Error: Only setting currently available is SEND_AGENT_UPDATES");
- }
- }
- private void HandleDebugLludpPacketCommand(string module, string[] args)
- {
- if (args.Length != 6)
- {
- MainConsole.Instance.Output("Usage: debug lludp packet <level> <bot-first-name> <bot-last-name>");
- return;
- }
- int level;
- if (!ConsoleUtil.TryParseConsoleInt(MainConsole.Instance, args[3], out level))
- return;
- string botFirstName = args[4];
- string botLastName = args[5];
- Bot bot;
- lock (m_bots)
- bot = m_bots.FirstOrDefault(b => b.FirstName == botFirstName && b.LastName == botLastName);
- if (bot == null)
- {
- MainConsole.Instance.Output("No bot named {0} {1}", null, botFirstName, botLastName);
- return;
- }
- bot.PacketDebugLevel = level;
- MainConsole.Instance.Output("Set debug level of {0} to {1}", null, bot.Name, bot.PacketDebugLevel);
- }
- private void HandleShowRegions(string module, string[] cmd)
- {
- string outputFormat = "{0,-30} {1, -20} {2, -5} {3, -5}";
- MainConsole.Instance.Output(outputFormat, null, "Name", "Handle", "X", "Y");
- lock (RegionsKnown)
- {
- foreach (GridRegion region in RegionsKnown.Values)
- {
- MainConsole.Instance.Output(
- outputFormat, null, region.Name, region.RegionHandle, region.X, region.Y);
- }
- }
- }
- private void HandleShowStatus(string module, string[] cmd)
- {
- ConsoleDisplayList cdl = new ConsoleDisplayList();
- cdl.AddRow("Bot connecting state", BotConnectingState);
- MainConsole.Instance.Output(cdl.ToString());
- }
- private void HandleShowBotsStatus(string module, string[] cmd)
- {
- ConsoleDisplayTable cdt = new ConsoleDisplayTable();
- cdt.AddColumn("Name", 24);
- cdt.AddColumn("Region", 24);
- cdt.AddColumn("Status", 13);
- cdt.AddColumn("Conns", 5);
- cdt.AddColumn("Behaviours", 20);
- Dictionary<ConnectionState, int> totals = new Dictionary<ConnectionState, int>();
- foreach (object o in Enum.GetValues(typeof(ConnectionState)))
- totals[(ConnectionState)o] = 0;
- lock (m_bots)
- {
- foreach (Bot bot in m_bots)
- {
- Simulator currentSim = bot.Client.Network.CurrentSim;
- totals[bot.ConnectionState]++;
- cdt.AddRow(
- bot.Name,
- currentSim != null ? currentSim.Name : "(none)",
- bot.ConnectionState,
- bot.SimulatorsCount,
- string.Join(",", bot.Behaviours.Keys.ToArray()));
- }
- }
- MainConsole.Instance.Output(cdt.ToString());
- ConsoleDisplayList cdl = new ConsoleDisplayList();
- foreach (KeyValuePair<ConnectionState, int> kvp in totals)
- cdl.AddRow(kvp.Key, kvp.Value);
- MainConsole.Instance.Output(cdl.ToString());
- }
- private void HandleShowBotStatus(string module, string[] cmd)
- {
- if (cmd.Length != 3)
- {
- MainConsole.Instance.Output("Usage: show bot <n>");
- return;
- }
- int botNumber;
- if (!ConsoleUtil.TryParseConsoleInt(MainConsole.Instance, cmd[2], out botNumber))
- return;
- Bot bot = GetBotFromNumber(botNumber);
- if (bot == null)
- {
- MainConsole.Instance.Output("Error: No bot found with number {0}", null, botNumber);
- return;
- }
- ConsoleDisplayList cdl = new ConsoleDisplayList();
- cdl.AddRow("Name", bot.Name);
- cdl.AddRow("Status", bot.ConnectionState);
- Simulator currentSim = bot.Client.Network.CurrentSim;
- cdl.AddRow("Region", currentSim != null ? currentSim.Name : "(none)");
- List<Simulator> connectedSimulators = bot.Simulators;
- List<string> simulatorNames = connectedSimulators.ConvertAll<string>(cs => cs.Name);
- cdl.AddRow("Connections", string.Join(", ", simulatorNames.ToArray()));
- MainConsole.Instance.Output(cdl.ToString());
- MainConsole.Instance.Output("Settings");
- ConsoleDisplayList statusCdl = new ConsoleDisplayList();
- statusCdl.AddRow(
- "Behaviours",
- string.Join(", ", bot.Behaviours.Values.ToList().ConvertAll<string>(b => b.Name).ToArray()));
- GridClient botClient = bot.Client;
- statusCdl.AddRow("SEND_AGENT_UPDATES", botClient.Settings.SEND_AGENT_UPDATES);
- MainConsole.Instance.Output(statusCdl.ToString());
- }
- /// <summary>
- /// Get a specific bot from its number.
- /// </summary>
- /// <returns>null if no bot was found</returns>
- /// <param name='botNumber'></param>
- private Bot GetBotFromNumber(int botNumber)
- {
- string name = GenerateBotNameFromNumber(botNumber);
- Bot bot;
- lock (m_bots)
- bot = m_bots.Find(b => b.Name == name);
- return bot;
- }
- private string GenerateBotNameFromNumber(int botNumber)
- {
- return string.Format("{0} {1}{2}", m_firstName, m_lastNameStem, botNumber);
- }
- internal void Grid_GridRegion(object o, GridRegionEventArgs args)
- {
- lock (RegionsKnown)
- {
- GridRegion newRegion = args.Region;
- if (RegionsKnown.ContainsKey(newRegion.RegionHandle))
- {
- return;
- }
- else
- {
- m_log.DebugFormat(
- "[BOT MANAGER]: Adding {0} {1} to known regions", newRegion.Name, newRegion.RegionHandle);
- RegionsKnown[newRegion.RegionHandle] = newRegion;
- }
- }
- }
- }
- }
|