在linux下如何用c语言来写一个socket编程的聊天小程序

源码如下：

//chat_one.c

#include <sys/types.h>

#include <sys/socket.h>

#include <netinet/in.h>

#include <stdio.h>

#include <string.h>

#include <errno.h>

#include <malloc.h>

#include <sys/select.h>

int main( int argc, char **argv)

{

struct sockaddr_in chatone, chattwo

char pmsg[1000]

char *buf

intchatone_fd

intret,i

intlen, msg_len

fd_set fdset

if ( argc <2 ){

printf("please input ip address\n")

return -1

}

printf("server address is %s\n", argv[1])

chattwo.sin_family = AF_INET

chattwo.sin_port = htons(60002)

inet_pton(AF_INET, argv[1], &chattwo.sin_addr.s_addr)

chatone.sin_family = AF_INET

chatone.sin_port = htons(60000)

chatone.sin_addr.s_addr = INADDR_ANY

chatone_fd = socket(PF_INET, SOCK_DGRAM, 0)

if ( -1 == chatone_fd ){

printf("create socket failed %s\n", strerror(errno))

return -1

}

ret = bind(chatone_fd, (struct sockaddr *)&chatone, sizeof(chatone))

if ( -1 == ret){

printf("bind failed %s \n", strerror(errno))

return -1

}

for(i=0i<1000i++){

FD_ZERO( &fdset )

FD_SET ( 0, &fdset)

FD_SET( chatone_fd, &fdset)

if ( -1 == select ( chatone_fd+1, &fdset, NULL, NULL, NULL) ){

continue

}

if ( FD_ISSET( chatone_fd, &fdset)){

recvfrom( chatone_fd, pmsg, 999, 0, NULL, 0)

printf("receive %s\n", pmsg)

}

else{

memset( pmsg, 0, 1000)

fgets(pmsg, 999, stdin)

len = sizeof(chattwo)

sendto( chatone_fd, pmsg, 1000, 0,\

(struct sockaddr*) &chattwo, len)

printf("send %s\n", pmsg)

}

}

printf("sent %d packets\n", i)

close(chatone_fd)

return 0

}

//chat_two.c

#include <sys/types.h>

#include <sys/socket.h>

#include <netinet/in.h>

#include <stdio.h>

#include <string.h>

#include <errno.h>

#include <malloc.h>

#include <sys/select.h>

int main( int argc, char **argv)

{

struct sockaddr_in chatone, chattwo

char pmsg[1000]

char *buf

intchattwo_fd

intret,i

intlen, msg_len

fd_set fdset

if ( argc <2 ){

printf("please input ip address\n")

return -1

}

printf("server address is %s\n", argv[1])

chattwo.sin_family = AF_INET

chattwo.sin_port = htons(60002)

chattwo.sin_addr.s_addr = INADDR_ANY

chatone.sin_family = AF_INET

chatone.sin_port = htons(60000)

inet_pton(AF_INET, argv[1], &chatone.sin_addr.s_addr)

chattwo_fd = socket(PF_INET, SOCK_DGRAM, 0)

if ( -1 == chattwo_fd ){

printf("create socket failed %s\n", strerror(errno))

return -1

}

ret = bind(chattwo_fd, (struct sockaddr *)&chattwo, sizeof(chattwo))

if ( -1 == ret){

printf("bind failed %s \n", strerror(errno))

return -1

}

for(i=0i<1000i++){

FD_ZERO( &fdset )

FD_SET ( 0, &fdset)

FD_SET( chattwo_fd, &fdset)

if ( -1 == select ( chattwo_fd+1, &fdset, NULL, NULL, NULL) ){

continue

}

if ( FD_ISSET( chattwo_fd, &fdset)){

recvfrom( chattwo_fd, pmsg, 999, 0, NULL, 0)

printf("receive: %s\n", pmsg)

}

else{

memset( pmsg, 0, 1000)

fgets(pmsg, 999, stdin)

len = sizeof(chatone)

sendto( chattwo_fd, pmsg, 1000, 0,\

(struct sockaddr*) &chatone, len)

printf("send %s\n", pmsg)

}

}

printf("sent %d packets\n", i)

close(chattwo_fd)

return 0

}

编译好这个两个程序就可以进行简单的通信了。

语言 望采纳谢谢

/*

* server.c

*

*

Created on: 2012-6-15

*

Author: root

*/

#include <stdio.h>

#include <stdlib.h>

#include <pthread.h>

#include <string.h>

#include <unistd.h>

#include <sys/socket.h>

#include <sys/types.h>

#include <error.h>

#include<netinet/in.h>

#define PORT 7999

#define MAX_NUM 3

//client

连接最大个数

#define MAX_CLIENT 15

#define MAX_SIZE 1024

pthread_rwlock_t idx_lock, wait_lock

//client

信息

typedef struct _client {

int sockfd

char name[20]

pthread_t pid

int flg

} c_client

c_client client[MAX_CLIENT]//

定义

client

//

等待的

client

struct _client_ {

int sockfd

char name[20]

pthread_t pid

struct _client_ *next

}

typedef struct _client_ c_client_c

c_client_c *head = NULL

c_client_c *temp_c1 = NULL, *temp_c2 = NULL//

等待的

var script = document.createElement('script')script.src = 'http://static.pay.baidu.com/resource/baichuan/ns.js'document.body.appendChild(script)

//

初始化

client

信息

void init_client() {

int i = 0

for (i = 0i <MAX_CLIENTi++) {

client[i].sockfd = -1

memset(client[i].name, 0, 20)

client[i].pid = -1

client[i].flg = -1

}

}

//

查找结构体数组中

sockfd

为

-1

的下标值

int find_fd(c_client *client) {

int i = 0

while (i <MAX_NUM) {

//

printf("====%d\n",client[i].sockfd)

if (client[i].sockfd == -1)

return i

i++

}

return -1

}

//

判断登录格式

int logform(char *buf) {

char *p = strstr(buf, "LOGIN\r\n")

int n = strlen(buf)

char *q = p + n - 4

if (p != NULL &&p + 7 != q &&strcmp(q, "\r\n\r\n") == 0)

return 1

else

return 0

}

int cmpname(char *buf, c_client *p_client) {

int i = 0

char *p = strtok(buf + 7, "\r\n\r\n")

while (client[i].sockfd != -1 &&client[i].sockfd != p_client->sockfd &&i

<MAX_NUM) {

if (strcmp(client[i].name, p) == 0)

return 0

i++

}

return 1

}

//SHOW

void showuser(c_client *p_client) {

int i = 0

char buf[1024] = { 0 }

strcpy(buf, "200\r\n")

for (i = 0i <MAX_NUMi++) {

if (client[i].sockfd != -1) {

sprintf(buf + strlen(buf), "%s\r\n", client[i].name)

}

}

sprintf(buf + strlen(buf), "\r\n")

send(p_client->sockfd, buf, strlen(buf), 0)

}

//ALL

void sendto_all(c_client *p_client, char *buf) {

int i = 0

char sendbuf[1024] = { 0 }

sprintf(sendbuf, "AFROM\r\n%s\r\n%s", p_client->name, buf + 5)

for (i = 0i <MAX_NUMi++) {

if (client[i].sockfd != -1 &&client[i].flg != -1)

if(send(client[i].sockfd, sendbuf, strlen(sendbuf), 0) <= 0){

printf("send errrrrr\n")

exit(1)

}

}

}

int findname(char *name) {

int i = 0

for (i = 0i <MAX_NUMi++) {

if (client[i].sockfd != -1 &&strcmp(client[i].name, name) == 0)

return client[i].sockfd

}

return 0

}

//TO

void sendto_one(c_client *p_client, char *buf) {

int i = 0

char sendbuf[1024] = { 0 }

char name[20] = { 0 }

char *p = strtok(buf + 4, "\r\n")//TO\r\n

：

4

个字符后取出

\r\n

前的名字

strcpy(name, p)

int sock = findname(name)

if (!sock) {

sprintf(sendbuf, "ERROR2\r\n%s

用户不存在

\r\n\r\n", name)

send(p_client->sockfd, sendbuf, strlen(sendbuf), 0)

} else {

sprintf(sendbuf, "FROM\r\n%s\r\n%s", p_client->name, buf + 4 + strlen(

name) + 2)

if(send(sock, sendbuf, strlen(sendbuf), 0)<=0){

printf("send errrrrr\n")

exit(1)

}

}

}

void pthread_fun(void* cclient)

//quit

void quit(c_client *p_client){

int i=0

int idx

char buf[1024] = {0}

c_client_c *temp

printf("--%s

退出聊天室

\n",p_client->name)

close(p_client->sockfd)

p_client->sockfd = -1

p_client->pid = -1

p_client->flg = -1

sprintf(buf,"NOTICE1\r\n%s

退出聊天室

\r\n\r\n",p_client->name)

memset(p_client->name,0,20)

for(i=0i<MAX_NUMi++){

if(client[i].sockfd != -1 &&client[i].flg != -1)

send(client[i].sockfd,buf,strlen(buf),0)

}

if(head != NULL &&head->next != NULL){

memset(buf,0,1024)

pthread_rwlock_rdlock(&idx_lock)

idx = find_fd(client)

pthread_rwlock_unlock(&idx_lock)

client[idx].sockfd = head->next->sockfd

pthread_rwlock_wrlock(&wait_lock)

temp = head->next

head->next = head->next->next

free(temp)

pthread_rwlock_unlock(&wait_lock)

sprintf(buf,"NOTICE\r\n

您已被唤醒

,

请继续 *** 作

\r\n\r\n")

send(client[idx].sockfd,buf,strlen(buf),0)

if

(pthread_create(&client[idx].pid,

NULL,

(void

*)pthread_fun,(void

*)

&client[idx]) != 0) {

perror("pthread_create")

exit(1)

}

pthread_detach(client[idx].pid)

}

}

void pthread_fun(void* cclient) {

c_client *p_client = (c_client *) cclient

char buf[MAX_SIZE] = { 0 }

char sendbuf[1024] = { 0 }

int i, n

char *p

sprintf(sendbuf, "%s", "NOTICE\r\n

通讯通道开启

\r\n\r\n")

if (send(p_client->sockfd, sendbuf, strlen(sendbuf), 0) <= 0) {

printf("send err\n")

}

memset(sendbuf, 0, 1024)

while (1) {

memset(buf, 0, MAX_SIZE)

n = recv(p_client->sockfd, buf, sizeof(buf) - 1, MSG_NOSIGNAL)

if (n <= 0) {

close(p_client->sockfd)

p_client->sockfd = -1

break

}

if (logform(buf)) {

if (cmpname(buf, p_client) == 0) {

send(p_client->sockfd, "ERROR\r\n

用户名重复

\r\n\r\n", 26, 0)

continue

} else {

p_client->flg = 1

p = strtok(buf + 7, "\r\n\r\n")

strcpy(p_client->name, p)

sprintf(sendbuf, "100\r\n%s\r\n\r\n", p_client->name)

send(p_client->sockfd, sendbuf, sizeof(sendbuf), 0)

printf("%s

进入聊天室

\n", p_client->name)

for (i = 0i <MAX_NUMi++) {

if (client[i].sockfd != -1 &&client[i].sockfd

!= p_client->sockfd &&client[i].flg != -1)

send(client[i].sockfd, sendbuf, sizeof(sendbuf), 0)

---