• 日期: 2028-03-21
  • 标签: 常规

智能家居设备状态建议：基于SQLite和TCP套接字的服务器端实现

本示例展示了一个基于SQLite数据库和TCP套接字的智能家居设备状态建议服务器的实现。服务器接收来自客户端的userid，查询数据库以获取设备状态信息，并根据设备状态生成建议发送回客户端。

服务器端代码：

#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[20]; // 将字段大小增加到20
    char device_state[20]; // 将字段大小增加到20
    char value[20]; // 将字段大小增加到20
    char mode[20]; // 将字段大小增加到20
} 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接受失败\nrecv");
        return;
    } else if (recvSize == 0) {
        perror("客户端已关闭连接");
        return;
    } else if (recvSize != sizeof(int)) {
        perror("接收到的字节数不正确");
        return;
    }
    userid = ntohl(userid); // 将网络字节序转换为主机字节序
    printf("客户端已连接\n");
    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\n", 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, (char *)sqlite3_column_text(stmt, 2), sizeof(status.device_name)-1); // 添加强制类型转换
        strncpy(status.device_state, (char *)sqlite3_column_text(stmt, 3), sizeof(status.device_state)-1); // 添加强制类型转换
        strncpy(status.value, (char *)sqlite3_column_text(stmt, 4), sizeof(status.value)-1); // 添加强制类型转换
        strncpy(status.mode, (char *)sqlite3_column_text(stmt, 5), sizeof(status.mode)-1); // 添加强制类型转换
    }

    // 检查查询结果是否为空
    if (sqlite3_step(stmt) == SQLITE_DONE) {
        fprintf(stderr, "No data found for user %d\n", userid);
        sqlite3_finalize(stmt);
        return; // 如果查询结果为空，则返回
    }

    int columnCount = sqlite3_column_count(stmt);
    printf("列数：%d\n", columnCount);
    if (sqlite3_column_count(stmt) == 0) {
        fprintf(stderr, "查询结果为空\n");
        sqlite3_finalize(stmt);
        return; // 如果查询结果为空，则返回
    }
    sqlite3_finalize(stmt);

    if (strlen(status.device_name) == 0) {
        perror("接收设备状态信息失败");
        return;
    }

    // 分析设备状态并生成建议
    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("发送失败\nsend");
        return;
    }
    printf("%s",suggestion);
}

int main() {
    printf("正在连接中...\n");
    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd == -1) {
        perror("连接失败\nsocket");
        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("连接失败\nbind");
        close(sockfd);
        return 1;
    }

    if (listen(sockfd, 5) == -1) {
        perror("连接失败\nlisten");
        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\n", 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("套接字接收失败\naccept");
                break;
            }

            context.sockfd = clientSockfd;
            handleClientRequest(&context);

            close(clientSockfd);
        }
    }
    sqlite3_close(db);
    close(sockfd);
    printf("服务器已关闭\n");

    return 0;
}



下面是客户端的代码
#include "procession.h"
#include "ui_procession.h"
#include <QAbstractSocket>
#include <QDebug>

Procession::Procession(int userid,QWidget *parent) : QWidget(parent), ui(new Ui::Procession), userid(userid) {
    ui->setupUi(this);
    processionWidget();
    m_socket = new QTcpSocket(this);
    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, QOverload<QAbstractSocket::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<quint32>(this->userid);
    if (m_socket->write(block) == -1) {
        ui->message->append("发送userid失败");
        m_socket->close();
    }
}

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(); // 先关闭套接字
        }

    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 <QWidget>
#include <QTcpSocket>
#include <QDebug>
#include <QByteArray>
#include <QTcpSocket>

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

代码修改说明：

  1. 增加Status结构体字段大小： 由于数据库查询结果可能包含超过10个字符的设备名称、状态、值和模式，将Status结构体中相应字段的大小从10增加到20，以确保能够容纳所有数据。
  2. 添加sqlite3_column_text强制类型转换： 在从sqlite3_column_text函数获取数据时，添加强制类型转换，确保返回的是一个有效的字符指针，避免无效读取。
  3. 检查查询结果是否为空： 在使用sqlite3_step函数获取数据后，添加判断语句，检查查询结果是否为空。如果为空，则使用sqlite3_finalize函数释放语句资源，并返回，避免访问无效数据。
  4. 增加错误处理： 在接收userid、发送建议和accept连接时，添加了错误处理机制，以确保程序在发生错误时能正常处理，避免程序崩溃。

代码优化建议：

  1. 使用更安全的字符串操作函数： 可以使用更安全的字符串操作函数，例如snprintf和strcpy_s来防止缓冲区溢出问题。
  2. 添加日志记录： 添加日志记录功能，方便调试和排查问题。
  3. 使用线程池： 使用线程池来处理多个客户端连接，提高服务器的并发性能。
  4. 优化数据库查询： 根据实际情况，优化数据库查询语句，提高查询效率。
  5. 增加安全性： 添加安全机制，例如使用SSL/TLS加密连接，防止数据被窃取。

总结：

通过上述代码修改和优化，服务器端代码更加健壮和安全，并且能有效地处理多个客户端连接，提供更加可靠的智能家居设备状态建议服务。

注意：

  • 这只是一个示例代码，实际使用中需要根据实际情况进行调整和完善。
  • 请确保数据库连接成功并正确地打开了数据库文件。
  • 请确保数据库查询的结果是否为空，以避免在结果为空时访问无效的数据。
  • 请检查是否有其他潜在的内存错误或逻辑错误导致程序崩溃。
智能家居设备状态建议：基于SQLite和TCP套接字的服务器端实现

原文地址: https://www.cveoy.top/t/topic/fCAj 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录