Question

我正在尝试使用一种设备的API，但是它使用的是带有强制执行的Origin标头的WS接口，这给我带来了麻烦。

在Chrome中，我可以在加载具有正确Origin的页面时打开控制台，创建WS连接，并轻松发送/接收消息：请注意，服务器始终会确认发送的消息（绿色）。

作为参考，如果我在另一个页面上创建连接会导致Origin标头不匹配（报告为404），则会发生这种情况：

为了避免这个问题，我转向了C语言，因为我的程序的其余部分还是以这种方式编写的。这是我目前使用的代码，主要基于this answer：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <signal.h>
#include <libwebsockets.h>

#define KGRN "\033[0;32;32m"
#define KCYN "\033[0;36m"
#define KRED "\033[0;32;31m"
#define KYEL "\033[1;33m"
#define KBLU "\033[0;32;34m"
#define KCYN_L "\033[1;36m"
#define KBRN "\033[0;33m"
#define RESET "\033[0m"

static int destroy_flag = 0;
static int connection_flag = 0;
static int writeable_flag = 0;

static void INT_HANDLER(int signo) {
    destroy_flag = 1;
}

struct session_data {
    int fd;
};

struct pthread_routine_tool {
    struct lws_context *context;
    struct lws *wsi;
};

static int websocket_write_back(struct lws *wsi_in, char *str, int str_size_in) 
{
    if (str == NULL || wsi_in == NULL)
        return -1;

    int n;
    int len;
    char *out = NULL;

    if (str_size_in < 1) 
        len = strlen(str);
    else
        len = str_size_in;

    out = (char *)malloc(sizeof(char)*(LWS_SEND_BUFFER_PRE_PADDING + len + LWS_SEND_BUFFER_POST_PADDING));
    //* setup the buffer*/
    memcpy (out + LWS_SEND_BUFFER_PRE_PADDING, str, len );
    //* write out*/
    n = lws_write(wsi_in, out + LWS_SEND_BUFFER_PRE_PADDING, len, LWS_WRITE_TEXT);

    printf(KBLU"[websocket_write_back] %s\n"RESET, str);
    //* free the buffer*/
    free(out);

    return n;
}


static int ws_service_callback(
                         struct lws *wsi,
                         enum lws_callback_reasons reason, void *user,
                         void *in, size_t len)
{

    switch (reason) {

        case LWS_CALLBACK_CLIENT_ESTABLISHED:
            printf(KYEL"[Main Service] Connect with server success.\n"RESET);
            connection_flag = 1;
            break;

        case LWS_CALLBACK_CLIENT_CONNECTION_ERROR:
            printf(KRED"[Main Service] Connect with server error.\n"RESET);
            destroy_flag = 1;
            connection_flag = 0;
            break;

        case LWS_CALLBACK_CLOSED:
            printf(KYEL"[Main Service] LWS_CALLBACK_CLOSED\n"RESET);
            destroy_flag = 1;
            connection_flag = 0;
            break;

        case LWS_CALLBACK_CLIENT_RECEIVE:
            printf(KCYN_L"[Main Service] Client recvived:%s\n"RESET, (char *)in);

            if (writeable_flag)
                destroy_flag = 1;

            break;
        case LWS_CALLBACK_CLIENT_WRITEABLE :
            printf(KYEL"[Main Service] On writeable is called. send byebye message\n"RESET);
            websocket_write_back(wsi, "{\"command\":\"subscribe\",\"identifier\":\"{\\\"channel\\\":\\\"DevicesChannel\\\",\\\"share_token\\\":\\\"D0E91\\\"}\"}", -1);
            websocket_write_back(wsi, "{\"command\":\"message\",\"identifier\":\"{\\\"channel\\\":\\\"DevicesChannel\\\",\\\"share_token\\\":\\\"D0E91\\\"}\",\"data\":\"{\\\"value\\\":100,\\\"action\\\":\\\"set_buzz\\\"}\"}", -1);
            writeable_flag = 1;
            break;

        default:
            break;
    }

    return 0;
}

static void *pthread_routine(void *tool_in)
{
    struct pthread_routine_tool *tool = tool_in;

    printf(KBRN"[pthread_routine] Good day. This is pthread_routine.\n"RESET);

    //* waiting for connection with server done.*/
    while(!connection_flag)
        usleep(1000*20);

    //*Send greeting to server*/ 
    lws_callback_on_writable(tool->wsi);

}

int main(void)
{
    //* register the signal SIGINT handler */
    struct sigaction act;
    act.sa_handler = INT_HANDLER;
    act.sa_flags = 0;
    sigemptyset(&act.sa_mask);
    sigaction( SIGINT, &act, 0);


    struct lws_context *context = NULL;
    struct lws_context_creation_info info;
    struct lws *wsi = NULL;
    struct lws_protocols protocol;

    memset(&info, 0, sizeof info);
    info.port = CONTEXT_PORT_NO_LISTEN;
    info.iface = NULL;
    info.protocols = &protocol;
    info.ssl_cert_filepath = NULL;
    info.ssl_private_key_filepath = NULL;
    info.extensions = lws_get_internal_extensions();
    info.gid = -1;
    info.uid = -1;
    info.options = 0;

    protocol.name  = "websockets";
    protocol.callback = &ws_service_callback;
    protocol.per_session_data_size = sizeof(struct session_data);
    protocol.rx_buffer_size = 0;
    protocol.id = 0;
    protocol.user = NULL;

    context = lws_create_context(&info);
    printf(KRED"[Main] context created.\n"RESET);

    if (context == NULL) {
        printf(KRED"[Main] context is NULL.\n"RESET);
        return -1;
    }


    wsi = lws_client_connect(context, "mobu1.herokuapp.com", 443, 1,
        "/cable", "mobu1.herokuapp.com", "link.motorbunny.com",
    if (wsi == NULL) {
        printf(KRED"[Main] wsi create error.\n"RESET);
        return -1;
    }

    printf(KGRN"[Main] wsi create success.\n"RESET);

    struct pthread_routine_tool tool;
    tool.wsi = wsi;
    tool.context = context;

    pthread_t pid;
    pthread_create(&pid, NULL, pthread_routine, &tool);
    pthread_detach(pid);

    while(!destroy_flag)
    {
        lws_service(context, 50);
    }

    lws_context_destroy(context);

    return 0;
}

运行上述程序的结果是：

如您所见，从服务器到我的客户端的定期ping被接收，但是lws_callback_on_writable(wsi);似乎没有任何作用，因为从未调用LWS_CALLBACK_CLIENT_WRITEABLE回调。另外，如果我直接在其他任何地方调用websocket_write_back()，则似乎没有向服务器发送任何内容，也没有确认。

有什么明显的地方我做错了吗？

编辑1： 我找到了整洁的wscat，可以在其中复制Chrome的结果：现在的问题是，如何将它与此C程序进行接口连接，使其可以等待来自服务器的Welcome消息，然后发送两条消息？更好的是，如何保持连接状态，以便我的程序可以在不同的时间点发送多个命令，而不必一直进行握手？

Answer 1

我发现一个丑陋的骇客，使我的C程序通过wsta程序将WebSocket消息发送到服务器。

它需要一个文本文件，只要要将程序发送到服务器，我的程序就会在其中添加文本文件。然后，新行由tail -f在后台拾取，并通过管道传递到wsta，以维护连接。可以将输出重定向到/dev/null，以使wsta的输出不会污染程序的输出，如果需要解析服务器的响应，也可以将其发送到文件。整个脚本如下所示（或者您可以将FIFO管道与cat一起使用，而不是与tail的文件一起使用）：

#!/bin/bash
touch commands.txt
tail commands.txt -f -n 0 | wsta --header "Origin: https://link.motorbunny.com" "wss://mobu1.herokuapp.com/cable" &> /dev/null &
./program

在C程序中，我只需要写入commands.txt文件：

FILE* cmd;
char sync_str[6];
void mb_connect()
{
  fprintf (cmd, "{\"command\":\"subscribe\",\"identifier\":\"{\\\"channel\\\":\\\"DevicesChannel\\\",\\\"share_token\\\":\\\"%s\\\"}\"}\n",sync_str);
  fflush(cmd);
}
void mb_send(int power, char* type)
{
  fprintf (cmd, "{\"command\":\"message\",\"identifier\":\"{\\\"channel\\\":\\\"DevicesChannel\\\",\\\"share_token\\\":\\\"%s\\\"}\",\"data\":\"{\\\"value\\\":%d,\\\"action\\\":\\\"set_%s\\\"}\"}\n",sync_str,power,type);
  fflush(cmd);
}

int main()
{
  cmd = fopen ("commands.txt","w");
  ...
  mb_connect();
  ...
  mb_send(200,"buzz");
  ...
  mb_send(0,"buzz");
}

Answer 2

从未调用LWS_CALLBACK_CLIENT_WRITEABLE回调的原因是因为此特定服务器使用非标准握手。因此，为了绕过这个问题，我forked的libwsclient分叉，并修改了握手检查功能，以确保不失配不会失败。我还添加了一个可选的Origin标头。

现在，我在原始程序中要做的就是

wsclient *client;
char sync_str[6];
void mb_send(int power, char* type)
{
  char cmd[2048];
  sprintf (cmd, "{\"command\":\"message\",\"identifier\":\"{\\\"channel\\\":\\\"DevicesChannel\\\",\\\"share_token\\\":\\\"%s\\\"}\",\"data\":\"{\\\"value\\\":%d,\\\"action\\\":\\\"set_%s\\\"}\"}",sync_str,power,type);
  libwsclient_send(client,cmd);
}
void mb_connect()
{
  char cmd[2048];
  sprintf (cmd, "{\"command\":\"subscribe\",\"identifier\":\"{\\\"channel\\\":\\\"DevicesChannel\\\",\\\"share_token\\\":\\\"%s\\\"}\"}",sync_str);
  libwsclient_send(client,cmd);
  mb_send(0,"buzz");
}
int nop()
{
  return 0;
}
int main()
{
  client = libwsclient_new_extra("wss://mobu1.herokuapp.com/cable","https://link.motorbunny.com");
  if(!client) {
    fprintf(stderr, "Unable to initialize new WS client.\n");
    exit(1);
  }

  libwsclient_onopen(client, &nop);
  libwsclient_onmessage(client, &nop);
  libwsclient_onerror(client, &nop);
  libwsclient_onclose(client, &nop);

  libwsclient_run(client);

  ...
  mb_connect();
  ...  
  mb_send(200,"buzz");
  mb_send(40,"twirl");
  ...
  mb_send(0,"buzz");
  mb_send(0,"twirl");
}

将WebSockets消息发送到服务器

2 个答案: