Exemple #1
0
 def verify_login(self, text=""):
     self.draw_ui(text if text else "Login to Chatterbox")
     self.username = raw_input("Enter username: "******"\n"
     pwd = raw_input("Enter password: "******"\n"
     message.send_msg(message.USER, self.username, self.sock)
     message.send_msg(message.PASS, md5_pwd, self.sock)
     res, text = message.receive_msg_from(self.sock)
     if res == message.REJECT:
         self.verify_login(text)
Exemple #2
0
 def direct_message(self, data, client_socket):
     match = re.match(r"^/direct\s(\w+)\s(.*)", data)
     user = self.client_users[client_socket]
     if match:
         target = match.group(1)
         msg = match.group(2)
         out_message = "{} <direct>: {}".format(user.name, msg)
         send_to = self.find_socket_by_name(target)
         self.log_message(
             "CLIENT",
             "{} to {} <direct>: {}".format(user.name, target, msg))
         if send_to:
             message.send_msg(message.DIRECT, out_message + "\n", send_to)
         else:
             message.send_msg(message.NORMAL,
                              "User {} is not online\n".format(target),
                              client_socket)
Exemple #3
0
 def register_user(self, data, client_socket):
     username = data.split(" ")[0]
     match = False
     with open("users.txt", "r+a") as f:
         for line in f:
             user, _ = line.strip().split(":")
             if username.strip() == user:
                 match = True
                 message.send_msg(message.REJECT,
                                  "Username already registered\n",
                                  client_socket)
                 break
         if not match:
             f.write("\n")
             f.write(data)
             self.log_message(
                 "SERVER", "User {} registered to server".format(username))
             message.send_msg(message.OK, "\n", client_socket)
Exemple #4
0
 def user_login(self, user, pwd, client_socket):
     for log_user in self.client_users.values():
         if user == log_user.name:
             message.send_msg(message.REJECT, "User is already logged in\n",
                              client_socket)
             self.log_message("SERVER",
                              "User {} is already logged in".format(user))
             return 0
     if self.authenticate_user(user, pwd):
         new_user = u.User(user, self.user_is_admin(user), client_socket)
         self.chatrooms['default'].add_user(new_user)
         self.client_users[client_socket] = new_user
         self.log_message("SERVER", "User {} is now online".format(user))
         self.broadcast(message.NORMAL,
                        "User {} is now online".format(user))
     else:
         message.send_msg(message.REJECT,
                          "Couldn't authenticate user {}\n".format(user),
                          client_socket)
         self.log_message("SERVER", "Failed login for user {}".format(user))
Exemple #5
0
    def register(self, err=""):
        if err:
            self.draw_ui(err)
        else:
            self.draw_ui("Registering new user")
        self.username = raw_input("Enter username: "******"Enter password: "******"Confirm password: "******"Passwords don't match!")
            pwd = raw_input("Enter password: "******"Confirm password: "******"\n"
        message.send_msg(message.REGISTER, ":".join([self.username, md5_pwd]),
                         self.sock)
        msg_type, text = message.receive_msg_from(self.sock)

        if msg_type == message.REJECT:
            self.register(text)
        else:
            self.verify_login(
                "Registration successful. Login to your new account")
Exemple #6
0
 def parse_input(self, user_input):
     direct_match = re.match(r"^/direct\s(\w+)\s(.*)", user_input)
     if re.search(r"^/join", user_input):
         message.send_msg(message.JOIN, user_input, self.sock)
     elif direct_match:
         message.send_msg(message.DIRECT, user_input, self.sock)
         self.pretty_print_message("You <direct to {}>: {}".format(
             direct_match.group(1), direct_match.group(2)))
     elif re.search(r"^/quit", user_input):
         self.sock.close()
         sys.exit(0)
     elif re.search(r"^/", user_input):
         message.send_msg(message.COMMAND, user_input, self.sock)
     else:
         message.send_msg(message.NORMAL, user_input, self.sock)
         self.pretty_print_message("You: " + user_input.strip())
Exemple #7
0
 def kick_user(self, client_socket, args):
     user = self.client_users[client_socket]
     if user.is_admin:
         d_parse = re.match(r"^(\w+)\s(\w+)", args)
         if d_parse:
             target = self.chatrooms[user.chatroom].get_user(
                 d_parse.group(2))
             try:
                 if d_parse.group(1) == 'server':
                     message.send_msg(
                         message.NORMAL,
                         "You have been kicked from the server\n",
                         target.sock)
                     self.close_connection(target.sock)
                     self.chatrooms[user.chatroom].broadcast(
                         "User {} was kicked from the server".format(
                             target))
                     self.log_message(
                         "SERVER",
                         "User {} was kicked from the server by {}".format(
                             target, user))
                 else:
                     message.send_msg(
                         message.NORMAL,
                         "You have been kicked from the room\n",
                         target.sock)
                     self.chatrooms['default'].add_user(target)
                     self.chatrooms[user.chatroom].remove_user(target, True)
                     self.log_message(
                         "SERVER",
                         "User {} was kicked from the room by {}".format(
                             target, user))
             except AttributeError:
                 message.send_msg(
                     message.NORMAL,
                     "You need to enter the same room as the user to kick\n",
                     client_socket)
Exemple #8
0
 def broadcast(self, msg_type, msg_text, client_socket=None):
     broad_range = [x for x in self.client_sockets if x != client_socket]
     for sock in broad_range:
         message.send_msg(msg_type, msg_text + "\n", sock)