This commit is contained in:
seoamugae committed 2026-01-08 17:00:24 +09:00
1 parent 5543c078a4
commit 0ce025f948
128 files changed
+153659 -1

No files matched your search

@@ -0,0 +1,35 @@
#CC:=arm-linux-gcc
CC:=gcc
TARGET_SRV=iot_server
OBJECT_SRV=$(TARGET_SRV).o
TARGET_CLN=iot_client
OBJECT_CLN=$(TARGET_CLN).o
TARGET_CLN_SENSOR=sql_client
OBJECT_CLN_SENSOR=$(TARGET_CLN_SENSOR).o
TARGET_CLN_BLE=ble_client
OBJECT_CLN_BLE=$(TARGET_CLN_BLE).o
#LDFLAGS=-D_REENTRANT -pthread -lmysqlclient
LDFLAGS=-D_REENTRANT -pthread
LDFLAGS_SQL=-D_REENTRANT -pthread -lmysqlclient
LDFLAGS_BLE=-D_REENTRANT -pthread -lbluetooth
all : $(TARGET_SRV) $(TARGET_CLN) $(TARGET_CLN_SENSOR) $(TARGET_CLN_BLE)
$(TARGET_SRV):$(OBJECT_SRV)
$(CC) -o $@ $(OBJECT_SRV) $(LDFLAGS)
$(TARGET_CLN):$(OBJECT_CLN)
$(CC) -o $@ $(OBJECT_CLN) $(LDFLAGS)
$(TARGET_CLN_SENSOR):$(OBJECT_CLN_SENSOR)
$(CC) -o $@ $(OBJECT_CLN_SENSOR) $(LDFLAGS_SQL)
$(TARGET_CLN_BLE):$(OBJECT_CLN_BLE)
$(CC) -o $@ $(OBJECT_CLN_BLE) $(LDFLAGS_BLE)
%.o:%.c
$(CC) -c -o $@ $<
clean:
rm -f *.o $(TARGET_SRV) $(TARGET_CLN) $(TARGET_CLN_SENSOR) $(TARGET_CLN_BLE)
Binary file not shown.
@@ -0,0 +1,206 @@
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <pthread.h>
#include <signal.h>
#include <bluetooth/bluetooth.h>
#include <bluetooth/rfcomm.h>
#define BUF_SIZE 512
#define NAME_SIZE 512
#define ARR_CNT 5
void *send_msg(void *arg);
void *recv_msg(void *arg);
void error_handling(char *msg);
char name[NAME_SIZE] = "[Default]";
typedef struct
{
int sockfd;
int btfd;
char sendid[NAME_SIZE];
} DEV_FD;
int main(int argc, char *argv[])
{
DEV_FD dev_fd;
struct sockaddr_in serv_addr;
pthread_t snd_thread, rcv_thread;
void *thread_return;
int ret;
struct sockaddr_rc addr = {0};
char dest[18] = "98:D3:37:90:A7:8B"; // iot00
char msg[BUF_SIZE];
if (argc != 4)
{
printf("Usage : %s <IP> <port> <name>\n", argv[0]);
exit(1);
}
sprintf(name, "%s", argv[3]);
dev_fd.sockfd = socket(PF_INET, SOCK_STREAM, 0);
if (dev_fd.sockfd == -1)
error_handling("socket() error");
memset(&serv_addr, 0, sizeof(serv_addr));
serv_addr.sin_family = AF_INET;
serv_addr.sin_addr.s_addr = inet_addr(argv[1]);
serv_addr.sin_port = htons(atoi(argv[2]));
if (connect(dev_fd.sockfd, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) == -1)
error_handling("connect() error");
sprintf(msg, "[%s:PASSWD]", name);
write(dev_fd.sockfd, msg, strlen(msg));
dev_fd.btfd = socket(AF_BLUETOOTH, SOCK_STREAM, BTPROTO_RFCOMM);
if (dev_fd.btfd == -1)
{
perror("socket()");
exit(1);
}
// set the connection parameters (who to connect to)
addr.rc_family = AF_BLUETOOTH;
addr.rc_channel = (uint8_t)1;
str2ba(dest, &addr.rc_bdaddr);
ret = connect(dev_fd.btfd, (struct sockaddr *)&addr, sizeof(addr));
if (ret == -1)
{
perror("connect()");
exit(1);
}
pthread_create(&rcv_thread, NULL, recv_msg, (void *)&dev_fd);
pthread_create(&snd_thread, NULL, send_msg, (void *)&dev_fd);
pthread_join(snd_thread, &thread_return);
// pthread_join(rcv_thread, &thread_return);
close(dev_fd.sockfd);
return 0;
}
void *send_msg(void *arg) // bluetooth --> server
{
DEV_FD *dev_fd = (DEV_FD *)arg;
int str_len;
int ret;
fd_set initset, newset;
struct timeval tv;
char name_msg[NAME_SIZE + BUF_SIZE + 2];
char msg[BUF_SIZE];
int total = 0;
FD_ZERO(&initset);
FD_SET(dev_fd->sockfd, &initset);
FD_SET(dev_fd->btfd, &initset);
// fputs("Input a message! [ID]msg (Default ID:ALLMSG)\n",stdout);
while (1)
{
// memset(msg,0,sizeof(msg));
// name_msg[0] = '\0';
tv.tv_sec = 1;
tv.tv_usec = 0;
newset = initset;
ret = select(dev_fd->btfd + 1, &newset, NULL, NULL, &tv);
// if(FD_ISSET(STDIN_FILENO, &newset))
if (FD_ISSET(dev_fd->btfd, &newset))
{
ret = read(dev_fd->btfd, msg + total, BUF_SIZE - total);
if (ret > 0)
{
total += ret;
}
else if (ret == 0)
{
dev_fd->sockfd = -1;
return NULL;
}
if (msg[total - 1] == '\n')
{
msg[total] = 0;
total = 0;
}
else
continue;
// sprintf(name_msg,"[%s]%s",dev_fd->sendid,msg);
fputs("ARD:", stdout);
fputs(msg, stdout);
if (write(dev_fd->sockfd, msg, strlen(msg)) <= 0)
{
dev_fd->sockfd = -1;
return NULL;
}
}
if (ret == 0)
{
if (dev_fd->sockfd == -1)
return NULL;
}
}
}
void *recv_msg(void *arg) // server --> bluetooth
{
DEV_FD *dev_fd = (DEV_FD *)arg;
int i;
char *pToken;
char *pArray[ARR_CNT] = {0};
char name_msg[NAME_SIZE + BUF_SIZE + 1];
int str_len;
while (1)
{
memset(name_msg, 0x0, sizeof(name_msg));
str_len = read(dev_fd->sockfd, name_msg, NAME_SIZE + BUF_SIZE);
if (str_len <= 0)
{
dev_fd->sockfd = -1;
return NULL;
}
name_msg[str_len] = 0;
fputs("SRV:", stdout);
fputs(name_msg, stdout);
/*
pToken = strtok(name_msg,"[:]");
i = 0;
while(pToken != NULL)
{
pArray[i] = pToken;
if(i++ >= ARR_CNT)
break;
pToken = strtok(NULL,"[:]");
}
if(!strncmp(pArray[1]," New conn",4) || !strncmp(pArray[1]," Already",4) || !strncmp(pArray[1]," Authent",4))
continue;
strcpy(dev_fd->sendid,pArray[0]);
write(dev_fd->btfd,pArray[1],strlen(pArray[1]));
*/
write(dev_fd->btfd, name_msg, strlen(name_msg));
/*
// printf("id:%s, msg:%s,%s,%s,%s\n",pArray[0],pArray[1],pArray[2],pArray[3],pArray[4]);
printf("id:%s, msg:%s\n",pArray[0],pArray[1]);
*/
}
}
void error_handling(char *msg)
{
fputs(msg, stderr);
fputc('\n', stderr);
exit(1);
}
@@ -0,0 +1,32 @@
0 PASSWD
1 PASSWD
2 PASSWD
3 PASSWD
4 PASSWD
5 PASSWD
6 PASSWD
7 PASSWD
8 PASSWD
9 PASSWD
10 PASSWD
11 PASSWD
12 PASSWD
13 PASSWD
14 PASSWD
15 PASSWD
16 PASSWD
17 PASSWD
18 PASSWD
19 PASSWD
20 PASSWD
LT_SQL PASSWD
SCG_ARD PASSWD
RGB_ARD PASSWD
SCG_AND PASSWD
RGB_AND PASSWD
LT_STM_SQL PASSWD
SCG_LIN PASSWD
RGB_LIN PASSWD
LT_ARD_SQL PASSWD
LT_STM PASSWD
LT_ARD PASSWD
Binary file not shown.
@@ -0,0 +1,155 @@
/* 서울기술 교육센터 IoT */
/* author : KSH */
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <pthread.h>
#include <signal.h>
#define BUF_SIZE 512
#define NAME_SIZE 512
#define ARR_CNT 5
void *send_msg(void *arg);
void *recv_msg(void *arg);
void error_handling(char *msg);
char name[NAME_SIZE] = "[Default]";
char msg[BUF_SIZE];
int main(int argc, char *argv[])
{
int sock;
struct sockaddr_in serv_addr;
pthread_t snd_thread, rcv_thread;
void *thread_return;
if (argc != 4)
{
printf("Usage : %s <IP> <port> <name>\n", argv[0]);
exit(1);
}
sprintf(name, "%s", argv[3]);
sock = socket(PF_INET, SOCK_STREAM, 0);
if (sock == -1)
error_handling("socket() error");
memset(&serv_addr, 0, sizeof(serv_addr));
serv_addr.sin_family = AF_INET;
serv_addr.sin_addr.s_addr = inet_addr(argv[1]);
serv_addr.sin_port = htons(atoi(argv[2]));
if (connect(sock, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) == -1)
error_handling("connect() error");
sprintf(msg, "[%s:PASSWD]", name);
write(sock, msg, strlen(msg));
pthread_create(&rcv_thread, NULL, recv_msg, (void *)&sock);
pthread_create(&snd_thread, NULL, send_msg, (void *)&sock);
pthread_join(snd_thread, &thread_return);
// pthread_join(rcv_thread, &thread_return);
close(sock);
return 0;
}
void *send_msg(void *arg)
{
int *sock = (int *)arg;
int str_len;
int ret;
fd_set initset, newset;
struct timeval tv;
char name_msg[NAME_SIZE + BUF_SIZE + 2];
FD_ZERO(&initset);
FD_SET(STDIN_FILENO, &initset);
fputs("Input a message! [ID]msg (Default ID:ALLMSG)\n", stdout);
while (1)
{
memset(msg, 0, sizeof(msg));
name_msg[0] = '\0';
tv.tv_sec = 1;
tv.tv_usec = 0;
newset = initset;
ret = select(STDIN_FILENO + 1, &newset, NULL, NULL, &tv);
if (FD_ISSET(STDIN_FILENO, &newset))
{
fgets(msg, BUF_SIZE, stdin);
if (!strncmp(msg, "quit\n", 5))
{
*sock = -1;
return NULL;
}
else if (msg[0] != '[')
{
strcat(name_msg, "[ALLMSG]");
strcat(name_msg, msg);
}
else
strcpy(name_msg, msg);
if (write(*sock, name_msg, strlen(name_msg)) <= 0)
{
*sock = -1;
return NULL;
}
}
if (ret == 0)
{
if (*sock == -1)
return NULL;
}
}
}
void *recv_msg(void *arg)
{
int *sock = (int *)arg;
int i;
char *pToken;
char *pArray[ARR_CNT] = {0};
char name_msg[NAME_SIZE + BUF_SIZE + 1];
int str_len;
while (1)
{
memset(name_msg, 0x0, sizeof(name_msg));
str_len = read(*sock, name_msg, NAME_SIZE + BUF_SIZE);
if (str_len <= 0)
{
*sock = -1;
return NULL;
}
name_msg[str_len] = 0;
fputs(name_msg, stdout);
/* pToken = strtok(name_msg,"[:]");
i = 0;
while(pToken != NULL)
{
pArray[i] = pToken;
if(i++ >= ARR_CNT)
break;
pToken = strtok(NULL,"[:]");
}
// printf("id:%s, msg:%s,%s,%s,%s\n",pArray[0],pArray[1],pArray[2],pArray[3],pArray[4]);
printf("id:%s, msg:%s\n",pArray[0],pArray[1]);
*/
}
}
void error_handling(char *msg)
{
fputs(msg, stderr);
fputc('\n', stderr);
exit(1);
}
Binary file not shown.
@@ -0,0 +1,327 @@
/* 서울기술교육센터 AIoT */
/* author : KSH */
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <pthread.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <dirent.h>
#include <sys/time.h>
#include <time.h>
#include <errno.h>
#define BUF_SIZE 512
#define MAX_CLNT 32
#define ID_SIZE 512
#define ARR_CNT 5
#define DEBUG
typedef struct
{
int fd;
char *from;
char *to;
char *msg;
int len;
} MSG_INFO;
typedef struct
{
int index;
int fd;
char ip[20];
char id[ID_SIZE];
char pw[ID_SIZE];
} CLIENT_INFO;
void *clnt_connection(void *arg);
void send_msg(MSG_INFO *msg_info, CLIENT_INFO *first_client_info);
void error_handling(char *msg);
void log_file(char *msgstr);
void getlocaltime(char *buf);
int clnt_cnt = 0;
pthread_mutex_t mutx;
int main(int argc, char *argv[])
{
int serv_sock, clnt_sock;
struct sockaddr_in serv_adr, clnt_adr;
int clnt_adr_sz;
int sock_option = 1;
pthread_t t_id[MAX_CLNT] = {0};
int str_len = 0;
int i = 0;
char idpasswd[(ID_SIZE * 2) + 3];
char *pToken;
char *pArray[ARR_CNT] = {0};
char msg[BUF_SIZE];
FILE *idFd = fopen("idpasswd.txt", "r");
if (idFd == NULL)
{
perror("fopen(\"idpasswd.txt\",\"r\") ");
exit(1);
}
char id[ID_SIZE];
char pw[ID_SIZE];
CLIENT_INFO *client_info = (CLIENT_INFO *)calloc(sizeof(CLIENT_INFO), MAX_CLNT);
if (client_info == NULL)
{
perror("calloc()");
exit(1);
}
do
{
str_len = fscanf(idFd, "%s %s", id, pw);
if (str_len <= 0)
break;
client_info[i].fd = -1;
strcpy(client_info[i].id, id);
strcpy(client_info[i].pw, pw);
i++;
if (i > MAX_CLNT)
{
printf("error client_info pull(Max:%d)\n", MAX_CLNT);
break;
}
} while (1);
fclose(idFd);
if (argc != 2)
{
printf("Usage : %s <port>\n", argv[0]);
exit(1);
}
fputs("IoT Server Start!!\n", stdout);
if (pthread_mutex_init(&mutx, NULL))
error_handling("mutex init error");
serv_sock = socket(PF_INET, SOCK_STREAM, 0);
memset(&serv_adr, 0, sizeof(serv_adr));
serv_adr.sin_family = AF_INET;
serv_adr.sin_addr.s_addr = htonl(INADDR_ANY);
serv_adr.sin_port = htons(atoi(argv[1]));
setsockopt(serv_sock, SOL_SOCKET, SO_REUSEADDR, (void *)&sock_option, sizeof(sock_option));
if (bind(serv_sock, (struct sockaddr *)&serv_adr, sizeof(serv_adr)) == -1)
error_handling("bind() error");
if (listen(serv_sock, 5) == -1)
error_handling("listen() error");
while (1)
{
clnt_adr_sz = sizeof(clnt_adr);
clnt_sock = accept(serv_sock, (struct sockaddr *)&clnt_adr, &clnt_adr_sz);
if (clnt_cnt >= MAX_CLNT)
{
printf("socket full\n");
shutdown(clnt_sock, SHUT_WR);
continue;
}
else if (clnt_sock < 0)
{
perror("accept()");
continue;
}
str_len = read(clnt_sock, idpasswd, sizeof(idpasswd));
idpasswd[str_len] = '\0';
if (str_len > 0)
{
i = 0;
pToken = strtok(idpasswd, "[:]");
while (pToken != NULL)
{
pArray[i] = pToken;
if (i++ >= ARR_CNT)
break;
pToken = strtok(NULL, "[:]");
}
for (i = 0; i < MAX_CLNT; i++)
{
if (!strcmp(client_info[i].id, pArray[0]))
{
if (!strcmp(client_info[i].pw, pArray[1]))
{
// 같은 ID가 이미 연결 중이면, 이전 세션을 끊고 **지금 소켓으로 인수**
if (client_info[i].fd != -1)
{
sprintf(msg, "[%s] takeover previous session (old fd:%d)\n",
pArray[0], client_info[i].fd);
log_file(msg);
// 이전 연결 정리
close(client_info[i].fd);
pthread_mutex_lock(&mutx);
clnt_cnt--; // 아래에서 새로 +1 할 거라서 미리 -1
pthread_mutex_unlock(&mutx);
}
// 새 연결 등록
strcpy(client_info[i].ip, inet_ntoa(clnt_adr.sin_addr));
pthread_mutex_lock(&mutx);
client_info[i].index = i;
client_info[i].fd = clnt_sock;
clnt_cnt++;
pthread_mutex_unlock(&mutx);
sprintf(msg,"[%s] New connected! (ip:%s,fd:%d,sockcnt:%d)\n",
pArray[0], inet_ntoa(clnt_adr.sin_addr), clnt_sock, clnt_cnt);
log_file(msg);
write(clnt_sock, msg, strlen(msg));
pthread_create(t_id + i, NULL, clnt_connection, (void *)(client_info + i));
pthread_detach(t_id[i]);
break;
}
}
}
if (i == MAX_CLNT)
{
sprintf(msg, "[%s] Authentication Error!\n", pArray[0]);
write(clnt_sock, msg, strlen(msg));
log_file(msg);
shutdown(clnt_sock, SHUT_WR);
}
}
else
shutdown(clnt_sock, SHUT_WR);
}
return 0;
}
void *clnt_connection(void *arg)
{
CLIENT_INFO *client_info = (CLIENT_INFO *)arg;
int str_len = 0;
int index = client_info->index;
char msg[BUF_SIZE];
char to_msg[MAX_CLNT * ID_SIZE + 1];
int i = 0;
char *pToken;
char *pArray[ARR_CNT] = {0};
char strBuff[BUF_SIZE * 2] = {0};
MSG_INFO msg_info;
CLIENT_INFO *first_client_info;
first_client_info = (CLIENT_INFO *)((void *)client_info - (void *)(sizeof(CLIENT_INFO) * index));
while (1)
{
memset(msg, 0x0, sizeof(msg));
str_len = read(client_info->fd, msg, sizeof(msg) - 1);
if (str_len <= 0)
break;
msg[str_len] = '\0';
pToken = strtok(msg, "[:]");
i = 0;
while (pToken != NULL)
{
pArray[i] = pToken;
if (i++ >= ARR_CNT)
break;
pToken = strtok(NULL, "[:]");
}
msg_info.fd = client_info->fd;
msg_info.from = client_info->id;
msg_info.to = pArray[0];
sprintf(to_msg, "[%s]%s", msg_info.from, pArray[1]);
msg_info.msg = to_msg;
msg_info.len = strlen(to_msg);
sprintf(strBuff, "msg : [%s->%s] %s", msg_info.from, msg_info.to, pArray[1]);
log_file(strBuff);
send_msg(&msg_info, first_client_info);
}
close(client_info->fd);
sprintf(strBuff, "Disconnect ID:%s (ip:%s,fd:%d,sockcnt:%d)\n", client_info->id, client_info->ip, client_info->fd, clnt_cnt - 1);
log_file(strBuff);
pthread_mutex_lock(&mutx);
clnt_cnt--;
client_info->fd = -1;
pthread_mutex_unlock(&mutx);
return 0;
}
void send_msg(MSG_INFO *msg_info, CLIENT_INFO *first_client_info)
{
int i = 0;
if (!strcmp(msg_info->to, "ALLMSG"))
{
for (i = 0; i < MAX_CLNT; i++)
if ((first_client_info + i)->fd != -1)
write((first_client_info + i)->fd, msg_info->msg, msg_info->len);
}
else if (!strcmp(msg_info->to, "IDLIST"))
{
char *idlist = (char *)malloc(ID_SIZE * MAX_CLNT);
msg_info->msg[strlen(msg_info->msg) - 1] = '\0';
strcpy(idlist, msg_info->msg);
for (i = 0; i < MAX_CLNT; i++)
{
if ((first_client_info + i)->fd != -1)
{
strcat(idlist, (first_client_info + i)->id);
strcat(idlist, " ");
}
}
strcat(idlist, "\n");
write(msg_info->fd, idlist, strlen(idlist));
free(idlist);
}
else if (!strcmp(msg_info->to, "GETTIME"))
{
sleep(1);
getlocaltime(msg_info->msg);
write(msg_info->fd, msg_info->msg, strlen(msg_info->msg));
}
else
for (i = 0; i < MAX_CLNT; i++)
if ((first_client_info + i)->fd != -1)
if (!strcmp(msg_info->to, (first_client_info + i)->id))
write((first_client_info + i)->fd, msg_info->msg, msg_info->len);
}
void error_handling(char *msg)
{
fputs(msg, stderr);
fputc('\n', stderr);
exit(1);
}
void log_file(char *msgstr)
{
fputs(msgstr, stdout);
}
void getlocaltime(char *buf)
{
struct tm *t;
time_t tt;
char wday[7][4] = {"Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"};
tt = time(NULL);
if (errno == EFAULT)
perror("time()");
t = localtime(&tt);
sprintf(buf, "[GETTIME]%02d.%02d.%02d %02d:%02d:%02d %s", t->tm_year + 1900 - 2000, t->tm_mon + 1, t->tm_mday, t->tm_hour, t->tm_min, t->tm_sec, wday[t->tm_wday]);
return;
}
Binary file not shown.
@@ -0,0 +1,373 @@
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <pthread.h>
#include <signal.h>
#include <mysql/mysql.h>
#include <stdbool.h>
#define BUF_SIZE 512
#define NAME_SIZE 512
#define ARR_CNT 64
#define DEVICE_NAME "KCCI"
void *send_msg(void *arg);
void *recv_msg(void *arg);
void error_handling(char *msg);
char name[NAME_SIZE] = "[Default]";
char msg[BUF_SIZE];
int main(int argc, char *argv[])
{
int sock;
struct sockaddr_in serv_addr;
pthread_t snd_thread, rcv_thread;
void *thread_return;
if (argc != 4)
{
printf("Usage : %s <IP> <port> <name>\n", argv[0]);
exit(1);
}
sprintf(name, "%s", argv[3]);
sock = socket(PF_INET, SOCK_STREAM, 0);
if (sock == -1)
error_handling("socket() error");
memset(&serv_addr, 0, sizeof(serv_addr));
serv_addr.sin_family = AF_INET;
serv_addr.sin_addr.s_addr = inet_addr(argv[1]);
serv_addr.sin_port = htons(atoi(argv[2]));
if (connect(sock, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) == -1)
error_handling("connect() error");
sprintf(msg, "[%s:PASSWD]", name);
write(sock, msg, strlen(msg));
pthread_create(&rcv_thread, NULL, recv_msg, (void *)&sock);
pthread_create(&snd_thread, NULL, send_msg, (void *)&sock);
pthread_join(snd_thread, &thread_return);
pthread_join(rcv_thread, &thread_return);
if (sock != -1)
close(sock);
return 0;
}
void *send_msg(void *arg)
{
int *sock = (int *)arg;
int str_len;
int ret;
fd_set initset, newset;
struct timeval tv;
char name_msg[NAME_SIZE + BUF_SIZE + 2];
FD_ZERO(&initset);
FD_SET(STDIN_FILENO, &initset);
fputs("Input a message! [ID]msg (Default ID:ALLMSG)\n", stdout);
while (1)
{
memset(msg, 0, sizeof(msg));
name_msg[0] = '\0';
tv.tv_sec = 1;
tv.tv_usec = 0;
newset = initset;
ret = select(STDIN_FILENO + 1, &newset, NULL, NULL, &tv);
if (FD_ISSET(STDIN_FILENO, &newset))
{
fgets(msg, BUF_SIZE, stdin);
if (!strncmp(msg, "quit\n", 5))
{
*sock = -1;
return NULL;
}
else if (msg[0] != '[')
{
strcat(name_msg, "[ALLMSG]");
strcat(name_msg, msg);
}
else
strcpy(name_msg, msg);
if (write(*sock, name_msg, strlen(name_msg)) <= 0)
{
*sock = -1;
return NULL;
}
}
if (ret == 0)
{
if (*sock == -1)
return NULL;
}
}
}
void *recv_msg(void *arg)
{
MYSQL *conn;
MYSQL_ROW sqlrow;
int res;
char sql_cmd[512] = {0};
char *host = "10.10.14.77";
char *user = "iot";
char *pass = "pwiot";
char *dbname = "iotdb";
int *sock = (int *)arg;
int i;
char *pToken;
char *pArray[ARR_CNT] = {0};
char name_msg[NAME_SIZE + BUF_SIZE + 1];
int str_len;
conn = mysql_init(NULL);
puts("MYSQL startup");
if (!(mysql_real_connect(conn, host, user, pass, dbname, 0, NULL, 0)))
{
fprintf(stderr, "ERROR : %s[%d]\n", mysql_error(conn), mysql_errno(conn));
exit(1);
}
else
printf("Connection Successful!\n\n");
while (1)
{
memset(name_msg, 0x0, sizeof(name_msg));
str_len = read(*sock, name_msg, NAME_SIZE + BUF_SIZE);
if (str_len <= 0)
{
*sock = -1;
return NULL;
}
fputs(name_msg, stdout);
// name_msg[str_len-1] = 0; //'\n' 제거
name_msg[strcspn(name_msg, "\n")] = '\0';
pToken = strtok(name_msg, "[:@]");
i = 0;
while (pToken != NULL)
{
pArray[i] = pToken;
if (++i >= ARR_CNT)
break;
pToken = strtok(NULL, "[:@]");
}
if (!strcmp(pArray[1], "SENSOR"))
{
// 값/플래그
char pos[8] = {0};
double TL = 0, TM = 0, TR = 0, ML = 0, MR = 0, BL = 0, BM = 0, BR = 0, Solar = 0;
bool gotPos = false, gotTL = false, gotTM = false, gotTR = false, gotML = false, gotMR = false, gotBL = false, gotBM = false, gotBR = false, gotSolar = false;
// pArray에서 "SENSOR" 위치 찾기
int cmd_idx = -1;
for (int k = 0; k < i; k++)
{
if (pArray[k] && strcmp(pArray[k], "SENSOR") == 0)
{
cmd_idx = k;
break;
}
}
if (cmd_idx == -1)
goto after_sensor;
// SENSOR 다음부터 (key,val) 쌍으로 순회
for (int j = cmd_idx + 1; j + 1 < i; j += 2)
{
char *key = pArray[j];
char *val = pArray[j + 1];
size_t L = strlen(val);
if (L && val[L - 1] == '\r')
val[L - 1] = '\0';
if (!strcasecmp(key, "Pos"))
{
strncpy(pos, val, sizeof(pos) - 1);
gotPos = true;
}
else if (!strcasecmp(key, "TL"))
{
TL = atof(val);
gotTL = true;
}
else if (!strcasecmp(key, "TM"))
{
TM = atof(val);
gotTM = true;
}
else if (!strcasecmp(key, "TR"))
{
TR = atof(val);
gotTR = true;
}
else if (!strcasecmp(key, "ML"))
{
ML = atof(val);
gotML = true;
}
else if (!strcasecmp(key, "MR"))
{
MR = atof(val);
gotMR = true;
}
else if (!strcasecmp(key, "BL"))
{
BL = atof(val);
gotBL = true;
}
else if (!strcasecmp(key, "BM"))
{
BM = atof(val);
gotBM = true;
}
else if (!strcasecmp(key, "BR"))
{
BR = atof(val);
gotBR = true;
}
else if (!strcasecmp(key, "Solar"))
{
Solar = atof(val);
gotSolar = true;
}
}
if (!(gotPos && gotSolar && gotTL && gotTM && gotTR && gotML && gotMR && gotBL && gotBM && gotBR))
{
fprintf(stderr, "[SENSOR] missing fields; skip insert.\n");
goto after_sensor;
}
char pos_esc[16] = {0};
mysql_real_escape_string(conn, pos_esc, pos, strlen(pos));
char sql_cmd[512];
snprintf(sql_cmd, sizeof(sql_cmd),
"INSERT INTO sensor "
"(ts, name, pos, solar, cds0, cds1, cds2, cds3, cds4, cds5, cds6, cds7) "
"VALUES (NOW(), '%s', '%s', %.3f, %.1f, %.1f, %.1f, %.1f, %.1f, %.1f, %.1f, %.1f)",
DEVICE_NAME, pos_esc, Solar, TL, TM, TR, ML, MR, BL, BM, BR);
int res = mysql_query(conn, sql_cmd);
if (!res)
printf("[SENSOR] inserted %lu rows\n", (unsigned long)mysql_affected_rows(conn));
else
fprintf(stderr, "ERROR: %s[%d]\n", mysql_error(conn), mysql_errno(conn));
after_sensor:;
}
//[KSH_SQL]GETDB@LAMP
//[KSH_SQL]SETDB@LAMP@ON
//[KSH_SQL]SETDB@LAMP@ON@KSH_LIN
if (!strcmp(pArray[1], "GETDB") && i == 3)
{
char name_esc[512];
mysql_real_escape_string(conn, name_esc, pArray[2], (unsigned long)strlen(pArray[2]));
snprintf(sql_cmd, sizeof(sql_cmd),
"SELECT "
"cds0, cds1, cds2, cds3, cds4, cds5, cds6, cds7, "
"pos, solar, DATE_FORMAT(ts,'%%Y-%%m-%%d_%%H:%%i:%%s') AS ts_fmt "
"FROM sensor WHERE name='%s' "
"ORDER BY ts DESC LIMIT 1",
name_esc);
if (mysql_query(conn, sql_cmd))
{
fprintf(stderr, "GETDB query err: %s\n", mysql_error(conn));
snprintf(sql_cmd, sizeof(sql_cmd), "[%s]GETDB@%s@ERR@QUERY\n", pArray[0], pArray[2]);
write(*sock, sql_cmd, strlen(sql_cmd));
break;
}
MYSQL_RES *result = mysql_store_result(conn);
if (!result)
{
fprintf(stderr, "GETDB store_result err: %s\r\n", mysql_error(conn));
snprintf(sql_cmd, sizeof(sql_cmd), "[%s]GETDB@%s@ERR@RESULT\n", pArray[0], pArray[2]);
write(*sock, sql_cmd, strlen(sql_cmd));
break;
}
MYSQL_ROW row = mysql_fetch_row(result);
if (!row)
{
snprintf(sql_cmd, sizeof(sql_cmd), "[%s]GETDB@%s@ERR@NoData\n", pArray[0], pArray[2]);
write(*sock, sql_cmd, strlen(sql_cmd));
mysql_free_result(result);
break;
}
// 응답: [LT_SQL]GETDB@name@TL@..@...@BR@..@Pos@..@Solar@..@TS@yyyy-mm-dd_HH:MM:SS
snprintf(sql_cmd, sizeof(sql_cmd),
"[%s]GETDB@%s"
"@Pos@%s@Solar@%s\n",
pArray[0], pArray[2],
row[8], row[9]);
write(*sock, sql_cmd, strlen(sql_cmd));
mysql_free_result(result);
}
else if (!strcmp(pArray[1], "SETDB"))
{
sprintf(sql_cmd, "update device set value='%s' where name='%s'", pArray[3], pArray[2]);
res = mysql_query(conn, sql_cmd);
if (!res)
{
if (i == 4)
sprintf(sql_cmd, "[%s]%s@%s@%s\n", pArray[0], pArray[1], pArray[2], pArray[3]);
else if (i == 5)
sprintf(sql_cmd, "[%s]%s@%s\n", pArray[4], pArray[2], pArray[3]);
else
continue;
printf("inserted %lu rows\n", (unsigned long)mysql_affected_rows(conn));
write(*sock, sql_cmd, strlen(sql_cmd));
}
else
fprintf(stderr, "ERROR: %s[%d]\n", mysql_error(conn), mysql_errno(conn));
}
else if ((!strcmp(pArray[1], "PLUG1") || !strcmp(pArray[1], "PLUG2")) && i == 3)
{
// 1. MySQL에 상태 업데이트
sprintf(sql_cmd, "update device set value='%s', date=now(), time=now() where name='%s'", pArray[2], pArray[1]);
res = mysql_query(conn, sql_cmd);
if (!res)
{
printf("Updated %s to %s in MySQL\n", pArray[1], pArray[2]);
}
else
{
fprintf(stderr, "SQL ERROR: %s [%d]\n", mysql_error(conn), mysql_errno(conn));
}
// 2. 아두이노에 그대로 메시지 전달
sprintf(sql_cmd, "[%s]%s@%s\n", pArray[0], pArray[1], pArray[2]);
write(*sock, sql_cmd, strlen(sql_cmd));
}
}
mysql_close(conn);
}
void error_handling(char *msg)
{
fputs(msg, stderr);
fputc('\n', stderr);
exit(1);
}