UDP畅聊

简介：项目是在Linux系统中CentOS6.5的环境下编写完成的，可以实现一个基本的群聊功能。客户端向服务端发送消息，服务器接收这条消息并将它发送给所有客户端，用到json库对消息序列化以及反序列化。ncurse库绘制客户端窗口。

原理图 

 

部分代码展示如下（用到哪些模块）

服务器udp_server.h

#pragma once

#include <iostream>
#include <string>
#include <map>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include "data_pool.h"

#define SIZE 1024

class udp_server{
	public:
		udp_server(int _port);
		void init_server();
		void recv_data(std::string &out_string);
		void broadcast();
		~udp_server();

	private:
		void send_data(const std::string &in_string,const struct sockaddr_in &peer);

	private:
		int sock;
		int port;
		data_pool pool;
		std::map<uint32_t, struct sockaddr_in> online;
};

客户端udp_client.h

#pragma once

#include <iostream>
#include <string>
#include <stdlib.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#define SIZE 1024

class udp_client{
	public:
		udp_client(std::string server_ip,int server_port);
		void init_client();
		void recv_data(std::string &out_string);
		void send_data(const std::string &in_string);
		~udp_client();
	private:
		int sock;
		struct sockaddr_in server;
};

数据池data_pool.h

#pragma once

#include <iostream>
#include <string>
#include <vector>
#include <semaphore.h>

#define NUM 256

class data_pool{
	public:
		data_pool(int _cap = NUM);
		void get_message(std::string &out_message);
		void put_message(const std::string &in_message);
		~data_pool();
	private:
		std::vector<std::string> pool;
		int cap;
		int consume_step;
		int product_step;
		sem_t space_sem;
		sem_t message_sem;
};

序列化和反序列化data.h

 

#pragma once

#include <iostream>
#include <string>
#include <json/json.h>

class data{
	public:
		data();
		void serialize(std::string &out_string);
		void unserialize(std::string &in_string);
		~data();

	public:
		std::string nick_name;
		std::string school;
		std::string message;
		std::string type;
};

窗口window.h

#pragma once

#include <iostream>
#include <ncurses.h>

#define SIZE 1024

class window{
	public:
		window();
		void put_string_to_window(WINDOW *w, int y,\
				int x, std::string &message);
		void get_string_from_window(WINDOW *w, std::string &out_string);
		void draw_header();
		void draw_output();
		void draw_flist();
		void draw_input();
		WINDOW *get_header();
		WINDOW *get_output();
		WINDOW *get_flist();
		WINDOW *get_input();
		~window();
	private:
		WINDOW *header;
		WINDOW *output;
		WINDOW *flist;
		WINDOW *input;
};

具体结果