基于C/S架构的智能家居状态监测系统 - 服务器端代码及优化
基于C/S架构的智能家居状态监测系统 - 服务器端代码及优化
这篇文章展示了一个简单的智能家居状态监测系统的服务器端代码,该系统使用C/S架构,服务器端使用C语言编写,并结合SQLite数据库实现。
服务器端代码c#include <stdio.h>#include <stdlib.h>#include <string.h>#include <unistd.h>#include <sys/socket.h>#include <sys/select.h>#include <arpa/inet.h>#include <sqlite3.h>#include <netinet/in.h>
#define MAX_BUFFER_SIZE 1024
typedef struct { int uid; char device_name[10]; char device_state[10]; char value[10]; char mode[10];} Status;
typedef struct { int sockfd; struct sockaddr_in client_addr; socklen_t client_addr_len; sqlite3 *db;} ServerContext;
void handleClientRequest(ServerContext *context) { char buffer[MAX_BUFFER_SIZE]; int userid;
// 接收客户端发送的userid ssize_t recvSize = recv(context->sockfd, &userid, sizeof(int), 0);
if (recvSize == -1) { perror('userid接受失败
recv'); return; } else if (recvSize == 0) { perror('客户端已关闭连接'); return; } else if (recvSize != sizeof(int)) { perror('接收到的字节数不正确'); return; } userid = ntohl(userid); // 将网络字节序转换为主机字节序 printf('客户端已连接 '); printf('userid:%d',userid); // 查询数据库获取设备状态信息 char sql[100]; snprintf(sql, sizeof(sql), 'SELECT * FROM Status WHERE uid = %d', userid); sqlite3_stmt *stmt; int rc = sqlite3_prepare_v2(context->db, sql, -1, &stmt, NULL); if (rc != SQLITE_OK) { fprintf(stderr, '无法执行的语句: %s ', sqlite3_errmsg(context->db)); return; }
Status status; memset(&status, 0, sizeof(Status)); // 清空status结构体 while (sqlite3_step(stmt) == SQLITE_ROW) { status.uid = sqlite3_column_int(stmt, 1); strncpy(status.device_name, sqlite3_column_text(stmt, 2), sizeof(status.device_name)-1); strncpy(status.device_state, sqlite3_column_text(stmt, 3), sizeof(status.device_state)-1); strncpy(status.value, sqlite3_column_text(stmt, 4), sizeof(status.value)-1); strncpy(status.mode, sqlite3_column_text(stmt, 5), sizeof(status.mode)-1); } if (rc == SQLITE_DONE) { fprintf(stderr, 'No data found for user %d
', userid);
sqlite3_finalize(stmt); } int columnCount = sqlite3_column_count(stmt); printf('列数:%d
', columnCount); if (sqlite3_column_count(stmt) == 0) { fprintf(stderr, '查询结果为空 '); return; } sqlite3_finalize(stmt);
// **此处存在潜在的段错误风险,将在下文分析和解决**
// 分析设备状态并生成建议 char suggestion[MAX_BUFFER_SIZE]; if (strcmp(status.device_name, '空调') == 0) { int temperature = atoi(status.value); if (temperature < 24) { snprintf(suggestion, sizeof(suggestion), '空调温度过低,建议提高温度至26℃'); } else { snprintf(suggestion, sizeof(suggestion), '空调温度正常'); } } else if (strcmp(status.device_name, '加湿器') == 0) { int humidity = atoi(status.value); if (humidity < 40 || humidity > 70) { snprintf(suggestion, sizeof(suggestion), '加湿器湿度过高或过低,建议调整加湿器湿度'); } else { snprintf(suggestion, sizeof(suggestion), '加湿器湿度正常'); } } else { snprintf(suggestion, sizeof(suggestion), '设备状态未知'); }
// 向客户端发送建议 ssize_t sendSize = send(context->sockfd, suggestion, strlen(suggestion), 0); if (sendSize == -1) { perror('发送失败
send'); return; } printf('%s',suggestion);}
int main() { printf('正在连接中... '); int sockfd = socket(AF_INET, SOCK_STREAM, 0); if (sockfd == -1) { perror('连接失败 socket'); return 1; }
struct sockaddr_in server_addr; server_addr.sin_family = AF_INET; server_addr.sin_addr.s_addr = INADDR_ANY; server_addr.sin_port = htons(12345);
if (bind(sockfd, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) { perror('连接失败
bind'); close(sockfd); return 1; }
if (listen(sockfd, 5) == -1) { perror('连接失败
listen'); close(sockfd); return 1; }
sqlite3 *db; int rc = sqlite3_open('/mnt/g/Qt/Client/Smarthome_Client/database/database.db', &db); if (rc != SQLITE_OK) { fprintf(stderr, '数据库打开失败: %s
', sqlite3_errmsg(db)); return 1; }
ServerContext context; context.sockfd = sockfd; context.client_addr_len = sizeof(context.client_addr); context.db = db;
fd_set readfds; int maxfd = sockfd;
while (1) { FD_ZERO(&readfds); FD_SET(sockfd, &readfds);
int activity = select(maxfd + 1, &readfds, NULL, NULL, NULL); if (activity == -1) { perror('select'); break; }
if (FD_ISSET(sockfd, &readfds)) { int clientSockfd = accept(sockfd, (struct sockaddr *)&context.client_addr, &context.client_addr_len); if (clientSockfd == -1) { perror('套接字接收失败
accept'); break; }
context.sockfd = clientSockfd; handleClientRequest(&context);
close(clientSockfd); } } sqlite3_close(db); close(sockfd); printf('服务器已关闭
');
return 0;}
客户端代码cpp#include 'procession.h'#include 'ui_procession.h'#include #include
Procession::Procession(int userid,QWidget *parent) : QWidget(parent), ui(new Ui::Procession), userid(userid){ ui->setupUi(this); processionWidget(); m_socket = new QTcpSocket(this); //connect(ui->connectBtn, &QPushButton::clicked, this, &Procession::on_connectBtn_clicked); connect(m_socket, &QTcpSocket::readyRead, this, &Procession::readyRead); connect(m_socket, &QTcpSocket::connected, this, &Procession::connected); connect(m_socket, &QTcpSocket::disconnected,this,&Procession::disconnected); connect(m_socket, QOverloadQAbstractSocket::SocketError::of(&QTcpSocket::error), this, &Procession::displayError);}
Procession::~Procession(){ delete ui;}
void Procession::processionWidget(){ setWindowTitle('服务器通信'); setAutoFillBackground(true); QPalette palette=this->palette(); QPixmap pixmap(':/user/image/image/net.jpg'); palette.setBrush(QPalette::Window, QBrush(pixmap)); setPalette(palette); setFixedSize(600,400);}
void Procession::connected(){ ui->message->append('连接成功'); QByteArray block; QDataStream out(&block, QIODevice::WriteOnly); out << static_cast
void Procession::disconnected(){ ui->message->append('连接失败'); m_socket->close();}
void Procession::readyRead(){ QByteArray data = m_socket->readAll(); if (data.isEmpty()) { ui->message->append('读取的数据为空'); return; } // 解析服务器返回的数据 QDataStream in(&data, QIODevice::ReadOnly); QString suggestion; in >> suggestion; // 使用重载的 >> 运算符来读取数据 if (in.status() != QDataStream::Ok) { ui->message->append('解析服务器返回的数据失败'); return; } ui->message->append(suggestion); m_socket->close();}
void Procession::displayError(QAbstractSocket::SocketError error){ QString errorMessage; switch (error) { case QAbstractSocket::ConnectionRefusedError: errorMessage = '连接被拒绝'; break; case QAbstractSocket::RemoteHostClosedError: errorMessage = '远程主机关闭'; break; case QAbstractSocket::HostNotFoundError: errorMessage = '未找到主机'; break; case QAbstractSocket::SocketTimeoutError: errorMessage = '连接超时'; break; case QAbstractSocket::NetworkError: errorMessage = '网络错误'; break; default: errorMessage = '未知错误'; break; } ui->message->append('连接失败:' + errorMessage);}
void Procession::on_connectBtn_clicked(){ if (m_socket->state() == QAbstractSocket::ConnectedState) { m_socket->disconnectFromHost(); // 先关闭套接字 qDebug()<<'1'; }
ui->message->append('已经连接上服务器'); QString ip = ui->IP->text(); QString port = ui->port->text();
if (ip.isEmpty() || port.isEmpty()) { ui->message->clear(); ui->message->append('请输入有效的IP地址和端口号'); return; }
ui->message->clear(); ui->message->append('正在连接中...'); m_socket->connectToHost(ip, static_cast<quint16>(ui->port->text().toInt()));}
#ifndef PROCESSION_H#define PROCESSION_H
#include
namespace Ui {class Procession;}
class Procession : public QWidget{ Q_OBJECT
public: explicit Procession(int userid,QWidget *parent = nullptr); ~Procession(); void processionWidget();
private slots: //void on_pushButton_clicked(); void connected(); void disconnected(); void readyRead(); void displayError(QAbstractSocket::SocketError error);
void on_connectBtn_clicked();
private: Ui::Procession ui; int userid; QTcpSocket m_socket;};
#endif // PROCESSION_H
代码分析
服务器端代码主要功能如下:
- 连接到SQLite数据库,并读取设备状态信息。2. 接收客户端发送的userid。3. 根据userid查询数据库获取设备状态信息。4. 根据设备状态信息分析并生成建议信息。5. 将建议信息发送给客户端。
段错误问题分析及解决
在使用 Valgrind 工具对服务器端代码进行内存泄漏检测时,发现程序在运行过程中出现了 Invalid read of size 1 的错误,并最终导致段错误。
经过分析,该错误的原因是在 handleClientRequest 函数中,程序尝试使用 strncpy 函数将数据库查询结果复制到 status 结构体中的成员变量时,由于 strncpy 函数不会自动在目标字符串末尾添加 � 结束符,如果源字符串长度大于等于目标缓冲区大小,目标字符串将不会以 � 结尾。这可能会导致后续对字符串的操作出现问题,例如 strcmp 函数在比较字符串时,会一直读取内存,直到遇到 � 结束符,如果字符串没有正确终止,就可能访问到非法的内存地址,从而导致程序崩溃。
解决方法:
为了解决这个问题,可以使用 strlcpy 函数代替 strncpy 函数。strlcpy 函数会保证目标字符串以 � 结尾,即使源字符串长度大于等于目标缓冲区大小。
将 handleClientRequest 函数中所有使用 strncpy 函数的地方替换为 strlcpy 函数,即可解决段错误问题。
**修改后的代码:**c// ... 其他代码 ...
#include <bsd/string.h> // 包含 strlcpy 函数
// ... 其他代码 ...
void handleClientRequest(ServerContext *context) { // ... 其他代码 ...
Status status; memset(&status, 0, sizeof(Status)); // 清空status结构体 while (sqlite3_step(stmt) == SQLITE_ROW) { status.uid = sqlite3_column_int(stmt, 1); strlcpy(status.device_name, sqlite3_column_text(stmt, 2), sizeof(status.device_name)); strlcpy(status.device_state, sqlite3_column_text(stmt, 3), sizeof(status.device_state)); strlcpy(status.value, sqlite3_column_text(stmt, 4), sizeof(status.value)); strlcpy(status.mode, sqlite3_column_text(stmt, 5), sizeof(status.mode)); }
// ... 其他代码 ...}
// ... 其他代码 ...
注意:strlcpy 函数并非C标准库函数,需要包含 bsd/string.h 头文件才能使用。
总结
通过使用 strlcpy 函数替换 strncpy 函数,可以有效地解决代码中潜在的段错误问题,提高代码的健壮性和安全性。在进行字符串操作时,应尽量选择安全可靠的函数,避免出现类似的错误。
原文地址: https://www.cveoy.top/t/topic/fCAG 著作权归作者所有。请勿转载和采集!