Author: soliton
Date: Fri Oct  3 00:12:43 2008
New Revision: 29812

URL: http://svn.gna.org/viewcvs/wesnoth?rev=29812&view=rev
Log:
various server command improvements

* added the banned nick to a ban when available
* made the sample command socket only and have it output the current
  value when no parameter is given
* made the status command report for all nicks with the same IP when
  using a simple nick as parameter (default for ordinary users)
* removed a couple of extraneous newlines from command output
* fixed an improper name for a local variable

Modified:
    trunk/src/server/ban.cpp
    trunk/src/server/ban.hpp
    trunk/src/server/server.cpp

Modified: trunk/src/server/ban.cpp
URL: 
http://svn.gna.org/viewcvs/wesnoth/trunk/src/server/ban.cpp?rev=29812&r1=29811&r2=29812&view=diff
==============================================================================
--- trunk/src/server/ban.cpp (original)
+++ trunk/src/server/ban.cpp Fri Oct  3 00:12:43 2008
@@ -34,12 +34,12 @@
 
        std::ostream& operator<<(std::ostream& o, const banned& n)
        {
-          return o << "IP: " << n.get_ip() << 
-                                       " reason: '" << n.get_reason() << 
-                                       "' end_time: " << 
n.get_human_end_time() <<
-                                       " start_time: " << 
n.get_human_start_time() <<
-                                       " issuer: " <<  n.get_who_banned();
-
+          return o << "IP: " << n.get_ip() <<
+                                       (n.get_nick().empty() ? "" : "  nick: " 
+ n.get_nick()) <<
+                                       "  reason: '" << n.get_reason() << "'"
+                                       "\nstart_time: " << 
n.get_human_start_time() <<
+                                       "  end_time: " << 
n.get_human_end_time() <<
+                                       "  issuer: " <<  n.get_who_banned();
        }
 
        bool banned_compare::operator()(const banned_ptr& a, const banned_ptr& 
b) const
@@ -114,13 +114,15 @@
                                   const time_t end_time, 
                                   const std::string& reason, 
                                   const std::string& who_banned, 
-                                  const std::string& group) : 
+                                  const std::string& group,
+                                  const std::string& nick) :
                ip_text_(ip),
                end_time_(end_time), 
                start_time_(time(0)),
                reason_(reason), 
                who_banned_(who_banned), 
-               group_(group)
+               group_(group),
+               nick_(nick)
        {
                ip_mask pair = parse_ip(ip_text_);
                ip_ = pair.first;
@@ -190,7 +192,7 @@
                        ip_ = pair.first;
                        mask_ = pair.second;
                }
-       
+               nick_ = cfg["nick"];
                if (cfg.has_attribute("end_time"))
                        end_time_       = 
lexical_cast_default<time_t>(cfg["end_time"], 0);
                if (cfg.has_attribute("start_time"))
@@ -207,6 +209,7 @@
        void banned::write(config& cfg) const
        {
                cfg["ip"]               = get_ip();
+               cfg["nick"] = get_nick();
                if (end_time_ > 0)
                {
                        std::stringstream ss;
@@ -447,7 +450,8 @@
                                                                 const time_t& 
end_time, 
                                                                 const 
std::string& reason, 
                                                                 const 
std::string& who_banned, 
-                                                                const 
std::string& group)
+                                                                const 
std::string& group,
+                                                                const 
std::string& nick)
        {
                try {
                        ban_set::iterator ban;
@@ -463,7 +467,7 @@
                }
                std::ostringstream ret;
                try {
-                       banned_ptr new_ban(new banned(ip, end_time, 
reason,who_banned, group));
+                       banned_ptr new_ban(new banned(ip, end_time, 
reason,who_banned, group, nick));
                        bans_.insert(new_ban);
                        if (end_time != 0)
                                time_queue_.push(new_ban);
@@ -563,7 +567,7 @@
                        out << "No bans set.";
                        return;
                }
-
+               out << "BAN LIST\n";
                std::set<std::string> groups;
 
                for (ban_set::const_iterator i = bans_.begin();

Modified: trunk/src/server/ban.hpp
URL: 
http://svn.gna.org/viewcvs/wesnoth/trunk/src/server/ban.hpp?rev=29812&r1=29811&r2=29812&view=diff
==============================================================================
--- trunk/src/server/ban.hpp (original)
+++ trunk/src/server/ban.hpp Fri Oct  3 00:12:43 2008
@@ -71,6 +71,7 @@
                std::string reason_;
                std::string who_banned_;
                std::string group_;
+               std::string nick_;
                static const std::string who_banned_default_;
                typedef std::pair<unsigned int, unsigned int> ip_mask;
                
@@ -79,7 +80,7 @@
                banned(const std::string& ip);
 
        public:
-               banned(const std::string& ip, const time_t end_time, const 
std::string& reason, const std::string& who_banned=who_banned_default_, const 
std::string& group="");
+               banned(const std::string& ip, const time_t end_time, const 
std::string& reason, const std::string& who_banned=who_banned_default_, const 
std::string& group="", const std::string& nick="");
                banned(const config&);
 
                void read(const config&);
@@ -102,6 +103,9 @@
 
                std::string get_who_banned() const
                { return who_banned_; }
+
+               std::string get_nick() const
+               { return nick_; }
 
                bool match_group(const std::string& group) const
                { return group_ == group; }
@@ -148,7 +152,7 @@
 
                time_t parse_time(std::string time_in) const;
 
-               std::string ban(const std::string&, const time_t&, const 
std::string&, const std::string&, const std::string&);
+               std::string ban(const std::string&, const time_t&, const 
std::string&, const std::string&, const std::string&, const std::string& = "");
                void unban(std::ostringstream& os, const std::string& ip);
                void unban_group(std::ostringstream& os, const std::string& 
group);
 

Modified: trunk/src/server/server.cpp
URL: 
http://svn.gna.org/viewcvs/wesnoth/trunk/src/server/server.cpp?rev=29812&r1=29811&r2=29812&view=diff
==============================================================================
--- trunk/src/server/server.cpp (original)
+++ trunk/src/server/server.cpp Fri Oct  3 00:12:43 2008
@@ -1208,6 +1208,17 @@
                        out << "New server started.";
                }
 #endif
+       } else if (command == "sample" && issuer_name == "*socket*") {
+               if (parameters.empty()) {
+                       out << "Current sample frequency: " << 
request_sample_frequency;
+                       return out.str();
+               }
+               request_sample_frequency = atoi(parameters.c_str());
+               if (request_sample_frequency <= 0) {
+                       out << "Sampling turned off.";
+               } else {
+                       out << "Sampling every " << request_sample_frequency << 
" requests.";
+               }
        } else if (command == "help") {
                out << help_msg;
        } else if (command == "stats") {
@@ -1245,20 +1256,32 @@
                LOG_SERVER << "<server> " + parameters + "\n";
                out << "message '" << parameters << "' relayed to players\n";
        } else if (command == "status") {
-               out << "STATUS REPORT\n";
+               out << "STATUS REPORT";
+               // If a simple username is given we'll check for its IP instead.
+               if (utils::isvalid_username(parameters)) {
+                       bool found = false;
+                       for (wesnothd::player_map::const_iterator pl = 
players_.begin(); pl != players_.end(); ++pl) {
+                               if (parameters == pl->second.name().c_str()) {
+                                       parameters = 
network::ip_address(pl->first);
+                                       found = true;
+                                       break;
+                               }
+                       }
+                       if (!found) return "";
+               }
                for (wesnothd::player_map::const_iterator pl = 
players_.begin(); pl != players_.end(); ++pl) {
                        if (parameters == ""
-                               || 
utils::wildcard_string_match(pl->second.name(), parameters)
-                               || 
utils::wildcard_string_match(network::ip_address(pl->first), parameters)) {
+                       || 
utils::wildcard_string_match(network::ip_address(pl->first), parameters)
+                       || utils::wildcard_string_match(pl->second.name(), 
parameters)) {
                                const network::connection_stats& stats = 
network::get_connection_stats(pl->first);
                                const int time_connected = 
stats.time_connected/1000;
                                const int seconds = time_connected%60;
                                const int minutes = (time_connected/60)%60;
                                const int hours = time_connected/(60*60);
-                               out << "'" << pl->second.name() << "' @ " << 
network::ip_address(pl->first)
+                               out << "\n'" << pl->second.name() << "' @ " << 
network::ip_address(pl->first)
                                        << " connected for " << hours << ":" << 
minutes << ":" << seconds
                                        << " sent " << stats.bytes_sent << " 
bytes, received "
-                                       << stats.bytes_received << " bytes\n";
+                                       << stats.bytes_received << " bytes";
                        }
                }
        } else if (command == "bans") {
@@ -1268,7 +1291,7 @@
                        ban_manager_.list_bans(out);
                }
        } else if (command == "ban" || command == "kban" || command == 
"kickban" || command == "gban") {
-               bool banned_ = false;
+               bool banned = false;
                const bool kick = (command == "kban" || command == "kickban");
                const bool group_ban = command == "gban";
                std::string::iterator first_space = 
std::find(parameters.begin(), parameters.end(), ' ');
@@ -1299,7 +1322,7 @@
                // if we find a '.' consider it an ip mask
                /** @todo  FIXME: make a proper check for valid IPs. */
                if (std::count(target.begin(), target.end(), '.') >= 1) {
-                       banned_ = true;
+                       banned = true;
 
                        std::string err = ban_manager_.ban(target, parsed_time, 
reason, issuer_name, group);
                        out << err;
@@ -1316,36 +1339,39 @@
                                }
                        }
                } else {
+                       bool kicked = false;
                        for (wesnothd::player_map::const_iterator pl = 
players_.begin();
                                        pl != players_.end(); ++pl)
                        {
                                if 
(utils::wildcard_string_match(pl->second.name(), target)) {
-                                       banned_ = true;
+                                       banned = true;
                                        const std::string& ip = 
network::ip_address(pl->first);
                                        if (!is_ip_banned(ip)) {
-                                               std::string err = 
ban_manager_.ban(ip,parsed_time, reason, issuer_name, group);
+                                               std::string err = 
ban_manager_.ban(ip, parsed_time, reason, issuer_name, group, target);
                                                out << err;
                                        }
                                        if (kick) {
+                                               if (kicked) out << "\n";
+                                               else kicked = true;
                                                out << "\nKicked " << 
pl->second.name() << ".";
                                                send_error(pl->first, ("You 
have been banned. Reason: " + reason).c_str());
                                                
network::queue_disconnect(pl->first);
                                        }
                                }
                        }
-                       if (!banned_) {
+                       if (!banned) {
                                // If nobody was banned yet look in the logs
                                for (std::map<std::string, 
std::string>::const_iterator i = ip_log_.begin();
                                                i != ip_log_.end(); i++) {
                                        if 
(utils::wildcard_string_match(i->first, target)) {
-                                               banned_ = true;
+                                               banned = true;
                                                if (!is_ip_banned(i->second)) {
                                                        std::string err = 
ban_manager_.ban(i->second,parsed_time, reason, issuer_name, group);
                                                        out << err;
                                                }
                                        }
                                }
-                               if(!banned_) {
+                               if(!banned) {
                                        out << "Nickmask '" << target << "' did 
not match, no bans set.";
                                }
                        }
@@ -1357,7 +1383,7 @@
                ban_manager_.unban(out, parameters);
        } else if (command == "ungban") {
                if (parameters == "") {
-                       return "You must enter an ipmask to unban.";
+                       return "You must enter an ipmask to ungban.";
                }
                ban_manager_.unban_group(out, parameters);
        } else if (command == "kick") {
@@ -1371,8 +1397,10 @@
                                pl != players_.end(); ++pl)
                        {
                                if 
(utils::wildcard_string_match(network::ip_address(pl->first), parameters)) {
-                                       kicked = true;
-                                       out << "Kicked " << pl->second.name() 
<< ".\n";
+                                       if (kicked) out << "\n";
+                                       else kicked = true;
+                                       out << "Kicked " << pl->second.name() 
<< " ("
+                                               << 
network::ip_address(pl->first) << ").";
                                        send_error(pl->first, "You have been 
kicked.");
                                        network::queue_disconnect(pl->first);
                                }
@@ -1382,22 +1410,16 @@
                                pl != players_.end(); ++pl)
                        {
                                if 
(utils::wildcard_string_match(pl->second.name(), parameters)) {
-                                       kicked = true;
+                                       if (kicked) out << "\n";
+                                       else kicked = true;
                                        out << "Kicked " << pl->second.name() 
<< " ("
-                                               << 
network::ip_address(pl->first) << ").\n";
+                                               << 
network::ip_address(pl->first) << ").";
                                        send_error(pl->first, "You have been 
kicked.");
                                        network::queue_disconnect(pl->first);
                                }
                        }
                }
-               if (!kicked) out << "No user matched '" << parameters << "'.\n";
-       } else if(command == "sample") {
-               request_sample_frequency = atoi(parameters.c_str());
-               if(request_sample_frequency <= 0) {
-                       out << "Sampling turned off";
-               } else {
-                       out << "Sampling every " << request_sample_frequency << 
" requests\n";
-               }
+               if (!kicked) out << "No user matched '" << parameters << "'.";
        } else if (command == "motd") {
                if (parameters == "") {
                        if (motd_ != "") {
@@ -1410,7 +1432,10 @@
                motd_ = parameters;
                out << "Message of the day set to: " << motd_;
        } else if (command == "searchlog") {
-               out << "IP/NICK LOG\n";
+               if (parameters.empty()) {
+                       return "You must enter a mask to search for.";
+               }
+               out << "IP/NICK LOG";
 
                bool found_something = false;
 
@@ -1421,7 +1446,7 @@
                                        i != ip_log_.end(); i++) {
                                if (utils::wildcard_string_match(i->second, 
parameters)) {
                                        found_something = true;
-                                       out << i->first << "@" << i->second << 
"\n";
+                                       out << "\n" << i->first << "@" << 
i->second;
                                }
                        }
                } else {
@@ -1429,11 +1454,11 @@
                                        i != ip_log_.end(); i++) {
                                if (utils::wildcard_string_match(i->first, 
parameters)) {
                                        found_something = true;
-                                       out << "'" << i->first << "' @ " << 
i->second << "\n";
+                                       out << "\n'" << i->first << "' @ " << 
i->second;
                                }
                        }
                }
-               if(!found_something) out << "No results found for '" << 
parameters << "'.\n";
+               if (!found_something) out << "No results found for '" << 
parameters << "'.";
        } else {
                out << "Command '" << command << "' is not recognized.\n" << 
help_msg;
        }


_______________________________________________
Wesnoth-commits mailing list
[email protected]
https://mail.gna.org/listinfo/wesnoth-commits

Reply via email to