mirror of https://github.com/OpenIPC/firmware.git
116 lines
2.5 KiB
C
116 lines
2.5 KiB
C
#include <stdbool.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#include <arpa/inet.h>
|
|
#include <fcntl.h>
|
|
#include <netinet/in.h>
|
|
#include <sys/socket.h>
|
|
#include <unistd.h>
|
|
|
|
#include "cjson/cJSON.h"
|
|
#include "netip.h"
|
|
#include "utils.h"
|
|
|
|
#define PACKED __attribute__((packed))
|
|
|
|
#define OP_LOGIN 1000
|
|
#define OP_SYSINFO 1020
|
|
|
|
#define RESULT_OK 100
|
|
#define RESULT_UNKNOWN_ERROR 101
|
|
#define RESULT_INCORRECT_PWD 203
|
|
|
|
typedef struct netip_preabmle {
|
|
uint8_t head;
|
|
uint8_t version;
|
|
uint16_t unused;
|
|
uint32_t session;
|
|
uint32_t sequence;
|
|
uint8_t total;
|
|
uint8_t cur;
|
|
uint16_t msgid;
|
|
uint32_t len_data;
|
|
char data[];
|
|
} PACKED netip_preabmle_t;
|
|
|
|
#define MAX_UDP_PACKET_SIZE 0xFFFF
|
|
|
|
typedef union netip_pkt {
|
|
char buf[MAX_UDP_PACKET_SIZE];
|
|
netip_preabmle_t header;
|
|
} netip_pkt_t;
|
|
|
|
#define NETIP_HSIZE sizeof(netip_preabmle_t)
|
|
#define NETIP_MAX_JSON sizeof(resp) - NETIP_HSIZE - 1
|
|
|
|
enum ConnectStatus netip_connect(const char *addr, uint16_t port) {
|
|
bool res = CONNECT_OK;
|
|
cJSON *json = NULL;
|
|
|
|
int s = socket(AF_INET, SOCK_STREAM, 0);
|
|
if (s == -1)
|
|
return false;
|
|
|
|
struct sockaddr_in srv;
|
|
srv.sin_addr.s_addr = inet_addr(addr);
|
|
srv.sin_family = AF_INET;
|
|
srv.sin_port = htons(port);
|
|
|
|
const int flags = fcntl(s, F_GETFL, 0);
|
|
fcntl(s, F_SETFL, flags | O_NONBLOCK);
|
|
(void)connect(s, (struct sockaddr *)&srv, sizeof(srv));
|
|
|
|
fd_set fdset;
|
|
FD_ZERO(&fdset);
|
|
FD_SET(s, &fdset);
|
|
struct timeval tv = {
|
|
.tv_sec = 2, /* 2 second timeout */
|
|
};
|
|
|
|
if (select(s + 1, NULL, &fdset, NULL, &tv) != 1) {
|
|
res = CONNECT_ERR;
|
|
goto quit;
|
|
}
|
|
fcntl(s, F_SETFL, flags);
|
|
|
|
netip_pkt_t msg;
|
|
memset(&msg.header, 0, sizeof(msg.header));
|
|
msg.header.head = 0xff;
|
|
msg.header.msgid = OP_LOGIN;
|
|
const char default_login[] =
|
|
"{\"EncryptType\": \"MD5\", \"LoginType\": \"DVRIP-Web\", \"PassWord\": "
|
|
"\"tlJwpbo6\", \"UserName\": \"admin\"}\n\0";
|
|
strcpy(msg.header.data, default_login);
|
|
msg.header.len_data = sizeof(default_login);
|
|
|
|
if (send(s, &msg, sizeof(default_login) + NETIP_HSIZE, 0) < 0) {
|
|
return CONNECT_ERR;
|
|
}
|
|
|
|
if (recv(s, &msg, sizeof(msg), 0) <= NETIP_HSIZE) {
|
|
puts("recv failed");
|
|
}
|
|
|
|
json = cJSON_Parse(msg.header.data);
|
|
if (!json) {
|
|
const char *error_ptr = cJSON_GetErrorPtr();
|
|
if (error_ptr != NULL) {
|
|
fprintf(stderr, "Error before: %s\n", error_ptr);
|
|
}
|
|
res = CONNECT_ERR;
|
|
goto quit;
|
|
}
|
|
const int retval = get_json_intval(json, "Ret", 0);
|
|
if (retval != RESULT_OK)
|
|
return CONNECT_PWDREQ;
|
|
|
|
quit:
|
|
if (json)
|
|
cJSON_Delete(json);
|
|
close(s);
|
|
|
|
return res;
|
|
}
|