Skip to content

C
CShell

龙的传人科龙 / CShell

通知 4
Star 0
Fork 0
C
CShell

前往新版Gitcode，体验更适合开发者的 AI 搜索 >>

提交 8bc9a77f 编写于 作者: 龙的传人科龙's avatar 龙的传人科龙
浏览文件
操作

存档

上级 a9102815
隐藏空白更改
内联 并排
Showing with 825 addition and 0 deletion
CShell/Client 0 → 100755
浏览文件 @ 8bc9a77f
文件已添加
CShell/Client.c 0 → 100644
浏览文件 @ 8bc9a77f
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/select.h>
#include <sys/time.h>
#define buf_SIZE 1024
#define SIZE 1024 //每次发送文件的大小
#define NameSize 32
//服务器信息
int serv_sock;
struct sockaddr_in serv_adr;
//用于select的集合
fd_set reads;
fd_set cpy_reads;
int fd_max;
//缓存定义
char Recvbuf[buf_SIZE];
char Name[NameSize]; //名字
char Inputbuf[buf_SIZE / 2];
char Filename[NameSize];
int telnet_sock = -1;
void error_handling(char *message);
void Shell();
void Read_a_send(char *filename);
void Excute();
void ReturnFile(char *filename);
void RecvFile(char *filename);
int main(int argc, char *argv[])
{
if (argc < 4)
error_handling("Usage <ServIP> <Port> <Name>\n");
serv_sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
memset(&serv_adr, 0, sizeof(serv_adr));
serv_adr.sin_family = AF_INET;
serv_adr.sin_addr.s_addr = inet_addr(argv[1]);
// serv_adr.sin_addr.s_addr = inet_addr("100.2.169.8");
serv_adr.sin_port = htons(atoi(argv[2]));
if (connect(serv_sock, (struct sockaddr *)&serv_adr, sizeof(serv_adr)) == -1)
error_handling("connect() error\n");
strcpy(Name, argv[3]);
printf("Name:%s\n", Name);
printf("Connect Ok ServerIp %s \t%d\n", inet_ntoa(serv_adr.sin_addr), ntohs(serv_adr.sin_port));
FD_ZERO(&reads);
fd_max = serv_sock;
FD_SET(0, &reads);
FD_SET(serv_sock, &reads);
while (1)
{
cpy_reads = reads;
int ret = select(fd_max + 1, &cpy_reads, NULL, NULL, NULL);
if (ret == -1)
error_handling("select() error");
else
Shell();
}
return 0;
}
void Shell()
{
if (FD_ISSET(0, &cpy_reads)) //从键盘读取数据
{
memset(Recvbuf, 0, buf_SIZE);
read(0, Recvbuf, buf_SIZE);
char Tempbuf[buf_SIZE];
char telnet_char[32];
memcpy(Tempbuf, Recvbuf, sizeof(Recvbuf));
char *p = strtok(Tempbuf, " ");
if (p != NULL)
{
char *pnext = strtok(NULL, " ");
if (pnext != NULL)
{
memcpy(telnet_char, p + 1, pnext - p);
}
}
if (0 == strcmp(telnet_char, "telnet"))
{
printf("Telnet Open\n");
telnet_sock = atoi(p);
}
if (Recvbuf[0] == ':' || Recvbuf[0] == '$')
write(serv_sock, Recvbuf, strlen(Recvbuf));
else
{
char Messagebuf[buf_SIZE + NameSize + 2];
memset(Messagebuf, 0, sizeof(Messagebuf));
sprintf(Messagebuf, "[%s]%s", Name, Recvbuf);
write(serv_sock, Messagebuf, strlen(Messagebuf));
}
}
if (FD_ISSET(serv_sock, &cpy_reads)) //接受数据，输出到屏幕
{
memset(Recvbuf, 0, sizeof(Recvbuf));
int str_len = read(serv_sock, Recvbuf, buf_SIZE);
if (str_len == 0) //如果收到长度为0的数据，说明服务器没了
{
printf("Server Baboo over!\n");
close(serv_sock);
return;
}
//对字符串进行分解，分解之后，可以知道是「命令」还是「信息」
char deliver[buf_SIZE];
memcpy(deliver, Recvbuf, sizeof(Recvbuf));
char *p;
p = strtok(deliver, " ");
for (int i = 0; i < 5; i++)
{
if (p != NULL)
p = strtok(NULL, " ");
if (p == NULL)
break;
}
char cmd[32];
if (p != NULL)
sprintf(cmd, "%s", p);
if (Recvbuf[0] == '[') ////默认为聊天
write(1, Recvbuf, sizeof(Recvbuf));
else if (0 == strcmp(cmd, "fetch"))
{
p = strtok(NULL, " ");
if (p != NULL)
{
sprintf(Filename, "%s", p);
Filename[strlen(Filename) - 1] = '\0';
printf("fetch filename :%s\n", Filename);
ReturnFile(Filename);
}
}
else if (0 == strcmp(cmd, "return"))
{
write(1, "成功fetch文件", strlen("成功fetch文件"));
//接受文件
char create_cmd[64];
sprintf(create_cmd, "%s %s", "touch", "Hello.txt");
RecvFile(Filename);
}
else if (Recvbuf[0] == 'e')
Excute();
else if (Recvbuf[0] != '[')
write(1, Recvbuf, sizeof(Recvbuf));
}
}
void ReturnFile(char *filename)
{
// char cmd[32];
// memset(cmd, 0, sizeof(cmd));
// strcpy(cmd, "return fetch");
// write(serv_sock, cmd, strlen(cmd)); //返回命令
sleep(1);
char name[32];
// strcpy(name, filename);
sprintf(name, "%s", filename);
//write(serv_sock, name, 32);
Read_a_send(filename);
}
void Excute()
{
//system("echo \"\" > cmd.txt ");
system("touch cmd.txt");
puts(Recvbuf);
int ret = system(Recvbuf);
if (ret < 0)
printf("system() error\n");
//发送文件给Server
Read_a_send("cmd.txt");
system("cat cmd.txt");
}
void Read_a_send(char *p)
{
char filename[32];
sprintf(filename, "%s", p);
FILE *fp;
int totlen;
char Recvbuf[buf_SIZE];
if ((fp = fopen(filename, "rb")) == NULL)
{
puts("fopen() error\n");
return;
}
fseek(fp, 0, SEEK_END);
totlen = ftell(fp);
//发送文件长度，方便服务器
memset(Recvbuf, 0, sizeof(Recvbuf));
sprintf(Recvbuf, "%d", totlen);
send(serv_sock, Recvbuf, strlen(Recvbuf) + 1, 0);
int cnt = totlen / SIZE;
int len1 = SIZE;
int lenlast;
if (totlen % SIZE)
{
lenlast = totlen - cnt * SIZE;
cnt = cnt + 1;
}
rewind(fp);
for (int i = 0; i < cnt - 1; i++)
{
fread(Recvbuf, SIZE, 1, fp);
sleep(0.1);
send(serv_sock, Recvbuf, SIZE, 0);
memset(Recvbuf, 0, sizeof(Recvbuf));
}
fread(Recvbuf, lenlast, 1, fp);
fclose(fp);
send(serv_sock, Recvbuf, lenlast, 0);
}
void RecvFile(char *filename)
{
char recvBuf[buf_SIZE];
int ret = read(serv_sock, recvBuf, buf_SIZE);
int Total = atoi(recvBuf); //先接收，文件大小
//接受
FILE *fp;
fp = fopen("Hello.txt", "wb");
if (fp == NULL)
error_handling("fopen() error\n");
int cnt = 0;
sleep(0.5);
while (1)
{
memset(recvBuf, 0, sizeof(recvBuf));
ret = read(serv_sock, recvBuf, SIZE);
fwrite(recvBuf, ret, 1, fp);
if (recvBuf[0] == '\0')
{
printf("recvBuf[0] == 0\n");
printf("ret == %d\n", ret);
}
else
{
printf("recvBuf[0] = %c\n", recvBuf[0]);
}
cnt += ret;
if (cnt >= Total || ret == -1)
break;
}
fclose(fp);
puts("End of Recv\n");
}
void Read(char *p)
{
char filename[32];
sprintf(filename, "%s", p);
FILE *fp;
int totlen;
char Recvbuf[buf_SIZE];
if ((fp = fopen(filename, "rb")) == NULL)
{
puts("fopen() error\n");
return;
}
fseek(fp, 0, SEEK_END);
totlen = ftell(fp);
//发送文件长度，方便服务器
memset(Recvbuf, 0, sizeof(Recvbuf));
sprintf(Recvbuf, "%d", totlen);
send(serv_sock, Recvbuf, strlen(Recvbuf) + 1, 0);
int cnt = totlen / SIZE;
int len1 = SIZE;
int lenlast;
if (totlen % SIZE)
{
lenlast = totlen - cnt * SIZE;
cnt = cnt + 1;
}
rewind(fp);
for (int i = 0; i < cnt - 1; i++)
{
fread(Recvbuf, SIZE, 1, fp);
sleep(0.1);
send(serv_sock, Recvbuf, SIZE, 0);
memset(Recvbuf, 0, sizeof(Recvbuf));
}
fread(Recvbuf, lenlast, 1, fp);
fclose(fp);
send(serv_sock, Recvbuf, lenlast, 0);
}
void error_handling(char *message)
{
printf("%s", message);
exit(0);
}
CShell/Server 0 → 100755
浏览文件 @ 8bc9a77f
文件已添加
CShell/Server.c 0 → 100644
浏览文件 @ 8bc9a77f
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/select.h>
#include <sys/time.h>
#define Max_clnt_cnt 100
#define BUF_SIZE 1024
#define SIZE 1024
#define NameSize 32
#define Passwdsize 32
int clnt_cnt = 0; //客户端连接数
socklen_t socklen_IPV4;
int clnt_socks[Max_clnt_cnt]; //客户端的socket数组
struct sockaddr_in clnt_adrs[Max_clnt_cnt]; //客户端socket数组对应的ip地址
char clnt_Name[Max_clnt_cnt][NameSize]; //名字数组
int telnet_sock = -1; //远程连接的socket号
int self_sock = -1;
/*要连接的服务器信息*/
int serv_sock; //服务器socket
struct sockaddr_in serv_adr; //服务器地址
/*用于Select的全局变量*/
fd_set reads;
fd_set cpy_reads;
int fd_max;
/*全局缓冲区*/
char buf[BUF_SIZE];
char password[Passwdsize];
char Filename[NameSize];
void error_handling(char *message);
void socket_handling();
void Execute(int sock, int cmd_len);
void Message_handle(int sock);
void Server_ls(int sock);
void Recv_a_Send(int sock, char *filename);
void Fetch(int sock, int cmd_len);
void Sendfile(int sock, char *p);
int main(int argc, char *argv[])
{
if (argc < 3)
error_handling("Usage <IP> <Port>\n");
serv_sock = socket(AF_INET, SOCK_STREAM, 0);
memset(&serv_adr, 0, sizeof(serv_adr));
serv_adr.sin_family = AF_INET;
serv_adr.sin_addr.s_addr = inet_addr(argv[1]);
//serv_adr.sin_addr.s_addr = inet_addr("100.2.169.8");
serv_adr.sin_port = htons(atoi(argv[2]));
//serv_adr.sin_port = htons(9991);
if (bind(serv_sock, (struct sockaddr *)&serv_adr, sizeof(serv_adr)) == -1)
error_handling("bind() error\n");
printf("bind Ok ServerIp %s \t%d\n", inet_ntoa(serv_adr.sin_addr), ntohs(serv_adr.sin_port));
if (listen(serv_sock, Max_clnt_cnt) == -1)
error_handling("listen() error\n");
FD_ZERO(&reads);
fd_max = serv_sock;
FD_SET(serv_sock, &reads);
while (1)
{
/* SELECT 处理连接 */
cpy_reads = reads;
// struct timeval time_out;
// time_out.tv_sec = 2;
// time_out.tv_usec = 0;
int ret = select(fd_max + 1, &cpy_reads, NULL, NULL, NULL);
if (ret == -1)
error_handling("select() error\n");
// else if (ret == 0)
// continue;
else
socket_handling();
}
close(serv_sock);
return 0;
}
void socket_handling()
{
for (int i = 0; i < fd_max + 1; i++)
{
if (FD_ISSET(i, &cpy_reads))
{
if (i == serv_sock)
{
struct sockaddr_in tmp_cln_adr;
int tongxunsock = accept(serv_sock, (struct sockaddr *)&tmp_cln_adr, &socklen_IPV4);
struct timeval Recvtimeout; //测试了很多次，给接受设置个超时比较好
Recvtimeout.tv_sec = 3;
Recvtimeout.tv_usec = 0;
setsockopt(tongxunsock, SOL_SOCKET, SO_RCVTIMEO, &Recvtimeout, sizeof(Recvtimeout));
clnt_adrs[clnt_cnt++] = tmp_cln_adr;
clnt_socks[clnt_cnt - 1] = tongxunsock;
FD_SET(tongxunsock, &reads);
if (fd_max < tongxunsock)
fd_max = tongxunsock;
printf("connected client :%d\t %s %d\n", tongxunsock, inet_ntoa(tmp_cln_adr.sin_addr), ntohs(tmp_cln_adr.sin_port));
}
else
{
sleep(0.2); //睡0.2秒确保一次性收取所有信息
int str_len;
memset(buf, 0, BUF_SIZE);
str_len = read(i, buf, BUF_SIZE);
if (str_len == 0)
{
printf("client:%d 断开连接\n", i);
FD_CLR(i, &reads);
close(i);
int location;
for (int k = 0; k < sizeof(clnt_socks); k++) //TODO 这里只是将某个数组上的位置置0，可以做到移动数组
{
if (clnt_socks[k] == i)
{
clnt_socks[k] = -1;
clnt_adrs[k].sin_addr.s_addr = 0;
clnt_adrs[k].sin_port = 0;
}
}
}
else //对客户端的数据进行处理
{
printf("buf[0] = %c\n", buf[0]);
//如果客户端返回return fetch命令，我就可以提取第一个单词，看是不是return
char temp[BUF_SIZE];
memcpy(temp, buf, sizeof(buf));
char *p;
p = strtok(temp, " ");
char retcmd[32];
if (p != NULL)
{
sprintf(retcmd, "%s", p + 1);
printf("retcmd :%s\n", retcmd);
}
if (buf[0] == '$') //执行服务端命令
{
if (buf[1] == 'l' && buf[2] == 's')
Server_ls(i); //处理 服务端「$ls」 命令
}
else if (buf[0] == ':' && 0 == strcmp(retcmd, "fetch"))
{ //TODO 服务端要把文件接受下来，再发送
p = strtok(NULL, " ");
sprintf(Filename, "%s", p);
printf("filename :%s\n", Filename);
Fetch(i, str_len);
}
else if (buf[0] == ':' && 0 == strcmp(retcmd, "telnet"))
{
p = strtok(NULL, " ");
if (p != NULL)
telnet_sock = atoi(p);
self_sock = i;
printf("telsock = %d\n", telnet_sock);
}
else if (buf[0] == ':') //执行客户端命令
Execute(i, str_len);
else
Message_handle(i);
}
}
}
}
}
void Message_handle(int sock)
{
char temp[BUF_SIZE];
memcpy(temp, buf, sizeof(buf));
char *p;
char Name[32];
p = strtok(temp, "]");
if (p != NULL)
{
char *pre = p;
p = strtok(NULL, "]");
if (p != NULL)
memcpy(Name, pre + 1, p - pre - 1);
}
printf("Name:%s\n", Name);
memcpy(clnt_Name[sock], Name, strlen(Name));
printf("Clnt_Name[%d]:%s\n", sock, clnt_Name[sock]);
for (int j = serv_sock + 1; j < fd_max + 1; j++)
{
if (j != sock)
{
write(j, buf, strlen(buf));
printf("返回数据给 %s --> client:%d", buf, j);
}
}
}
void Execute(int sock, int cmd_len)
{
char prefix[] = "echo 8002904 | sudo -S";
//char backfix[] = "> cmd.txt 2>&1";
char backfix[] = " 2>&1";
char cmd_buf[BUF_SIZE + 40];
char tmp[BUF_SIZE];
memset(tmp, 0, sizeof(tmp));
strncpy(tmp, buf + 1, strlen(buf) - 1);
tmp[cmd_len - 1] = '\0';
printf("tmp == %s\n", tmp);
sprintf(cmd_buf, "%s %s %s", prefix, tmp, backfix);
puts(cmd_buf);
if (telnet_sock > -1)
write(telnet_sock, cmd_buf, strlen(cmd_buf));
int ret = system("touch Temp.txt");
if (ret == -1)
{
puts("system(\"touch Temp.txt\") error");
return;
}
Recv_a_Send(sock, "Temp.txt");
}
void Recv_a_Send(int sock, char *p)
{
char filename[32];
if (p != NULL)
sprintf(filename, "%s", p);
printf("filename:%s\n", filename);
char recvBuf[BUF_SIZE];
int ret;
if (telnet_sock > -1)
ret = recv(telnet_sock, recvBuf, BUF_SIZE, 0);
int Total = atoi(recvBuf); //先接收，文件大小
//接受
FILE *fp;
fp = fopen(filename, "wb");
if (fp == NULL)
error_handling("fopen() error\n");
int cnt = 0;
sleep(1);
while (1)
{
memset(recvBuf, 0, sizeof(recvBuf));
if (telnet_sock > -1)
ret = read(telnet_sock, recvBuf, SIZE);
fwrite(recvBuf, ret, 1, fp);
if (recvBuf[0] == '\0')
{
printf("recvBuf[0] == 0\n");
printf("ret == %d\n", ret);
}
else
{
printf("recvBuf[0] = %c\n", recvBuf[0]);
}
write(sock, recvBuf, ret); //接完就发
cnt += ret;
if (cnt >= Total || ret == -1)
break;
}
fclose(fp);
}
void Recvfile(int sock, char *p)
{
char filename[32];
if (p != NULL)
sprintf(filename, "%s", p);
printf("filename:%s\n", filename);
char recvBuf[BUF_SIZE];
int ret;
if (telnet_sock > -1)
ret = recv(telnet_sock, recvBuf, BUF_SIZE, 0);
int Total = atoi(recvBuf); //先接收，文件大小
//接受
FILE *fp;
fp = fopen(filename, "wb");
if (fp == NULL)
error_handling("fopen() error\n");
int cnt = 0;
sleep(1);
while (1)
{
memset(recvBuf, 0, sizeof(recvBuf));
if (telnet_sock > -1)
ret = read(telnet_sock, recvBuf, SIZE);
fwrite(recvBuf, ret, 1, fp);
if (recvBuf[0] == '\0')
{
printf("recvBuf[0] == 0\n");
printf("ret == %d\n", ret);
}
else
{
printf("recvBuf[0] = %c\n", recvBuf[0]);
}
cnt += ret;
if (cnt >= Total || ret == -1)
break;
}
fclose(fp);
}
void Fetch(int sock, int cmd_len)
{
char prefix[] = "echo 8002904 | sudo -S";
char backfix[] = " 2>&1";
char cmd_buf[BUF_SIZE + 40];
char tmp[BUF_SIZE];
memset(tmp, 0, sizeof(tmp));
strncpy(tmp, buf + 1, strlen(buf) - 1);
tmp[cmd_len - 1] = '\0';
printf("tmp == %s\n", tmp);
sprintf(cmd_buf, "%s %s %s", prefix, tmp, backfix);
puts(cmd_buf);
if (telnet_sock > -1)
write(telnet_sock, cmd_buf, strlen(cmd_buf));
char create_cmd[64];
sprintf(create_cmd, "%s %s", "touch", Filename);
int ret = system(create_cmd);
if (ret == -1)
{
puts(create_cmd);
puts("error");
return;
}
//先发命令给 主动的客户端
Recvfile(sock, "world.txt");
sleep(2);
Sendfile(sock, "world.txt");
}
void Sendfile(int sock, char *p)
{
char ret_cmd[64] = "echo return return return return return return";
write(sock, ret_cmd, strlen(ret_cmd));
write(1, ret_cmd, strlen(ret_cmd));
sleep(1);
char filename[32];
sprintf(filename, "%s", p);
FILE *fp;
int totlen;
char Recvbuf[BUF_SIZE];
if ((fp = fopen(filename, "rb")) == NULL)
{
puts("fopen() error\n");
return;
}
fseek(fp, 0, SEEK_END);
totlen = ftell(fp);
//发送文件长度，方便客户端
memset(Recvbuf, 0, sizeof(Recvbuf));
sprintf(Recvbuf, "%d", totlen);
write(sock, Recvbuf, strlen(Recvbuf) + 1);
sleep(1);
int cnt = totlen / SIZE;
int len1 = SIZE;
int lenlast;
if (totlen % SIZE)
{
lenlast = totlen - cnt * SIZE;
cnt = cnt + 1;
}
rewind(fp);
for (int i = 0; i < cnt - 1; i++)
{
fread(Recvbuf, SIZE, 1, fp);
sleep(0.1);
send(sock, Recvbuf, SIZE, 0);
memset(Recvbuf, 0, sizeof(Recvbuf));
}
fread(Recvbuf, lenlast, 1, fp);
fclose(fp);
send(sock, Recvbuf, lenlast, 0);
}
void Server_ls(int sock)
{
for (int k = 0; k < clnt_cnt; k++)
{
char hostMess[50];
char port[8];
char sockNum[8];
char Name[32];
strcpy(hostMess, inet_ntoa(clnt_adrs[k].sin_addr));
strcat(hostMess, "\t");
sprintf(port, "%d", ntohs(clnt_adrs[k].sin_port));
strcat(port, "\t");
strcat(hostMess, port);
strcat(hostMess, "sockNum\t");
sprintf(sockNum, "%d", clnt_socks[k]);
strcat(hostMess, sockNum);
strcat(hostMess, "\t");
strcat(hostMess, clnt_Name[k + 4]);
strcat(hostMess, "\n");
send(sock, hostMess, strlen(hostMess), 0);
}
}
void error_handling(char *message)
{
printf("%s", message);
exit(0);
}
CShell/Temp.txt 0 → 100644
浏览文件 @ 8bc9a77f
/home/coderon/文档/Inetprogram/TelRoom/Telroom/CShell
CShell/img_downloader.txt 0 → 100644
浏览文件 @ 8bc9a77f
#w!/bin/bash
if [ $# -ne 3 ];
then
 echo "Usage: $0 URL -d DIRECTORY"
 exit -1
fi
for i in {1..4}
do
case $1 in
-d) shift; directory=$1; shift;;
*) url=${url:-$1};shift;;
esac
done
mkdir -p $directory
baseurl=$(echo $url | egrep -o "https?://[a-z.]+")
echo Downloading $url
curl -s $url | egrep -o "<img [^>]* src=[^>]*>" |
egrep -o "http[^\"]*" > /tmp/$$.list
#sed 's/<img =\"\([^"]*\).*/\1/g' > /tmp/$$.list
#sed -i "s|^/|$baseurl/|" /tmp/$$.list
cd $directory;
while read fileurl
do
filename=$(echo $fileurl |sed "s/\//_/g")
#echo fileurl is $fileurl
#echo the "filename is" $filename
echo downloading $fileurl
curl -s -0 "$fileurl" --silent -o $filename
done < /tmp/$$.list
CShell/world.txt 0 → 100644
浏览文件 @ 8bc9a77f
#w!/bin/bash
if [ $# -ne 3 ];
then
 echo "Usage: $0 URL -d DIRECTORY"
 exit -1
fi
for i in {1..4}
do
case $1 in
-d) shift; directory=$1; shift;;
*) url=${url:-$1};shift;;
esac
done
mkdir -p $directory
baseurl=$(echo $url | egrep -o "https?://[a-z.]+")
echo Downloading $url
curl -s $url | egrep -o "<img [^>]* src=[^>]*>" |
egrep -o "http[^\"]*" > /tmp/$$.list
#sed 's/<img =\"\([^"]*\).*/\1/g' > /tmp/$$.list
#sed -i "s|^/|$baseurl/|" /tmp/$$.list
cd $directory;
while read fileurl
do
filename=$(echo $fileurl |sed "s/\//_/g")
#echo fileurl is $fileurl
#echo the "filename is" $filename
echo downloading $fileurl
curl -s -0 "$fileurl" --silent -o $filename
done < /tmp/$$.list
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或