module/socket.cpp at main · Connecktail/module · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
#include <Arduino.h>
#include <WiFi.h>
#include "socket.h"
#include "wifi_connect.h"
#include "actions.h"

extern rgb_led_t rgb_led;

void establish_socket()
{
  char message[255];
  IPAddress gateway = WiFi.gatewayIP();
  sprintf(message, "Establishing socket to server : %s, at port : %d", gateway.toString(), PORT_NUMBER);
  Serial.println(message);

  while (!client.connect(gateway, PORT_NUMBER, 5000))
  {
    digitalWrite(rgb_led.blue.pin, rgb_led.blue.value);
    delay(250);
    digitalWrite(rgb_led.blue.pin, 0);
    Serial.println("Can't connect");
    delay(250);
  }
  Serial.println("Connected to server !");
  enable_led(rgb_led);
  delay(2000);
  disable_led(rgb_led);
}

char *construct_pair_message()
{
  IPAddress ip_address = get_ip_address();
  String ip_address_string = ip_address.toString();
  unsigned int ip_address_length = ip_address_string.length() + 1;
  unsigned int nb = ip_address_length > 0 ? (int)log10((double)ip_address_length) + 1 : 1;

  char *ip_address_char = (char *)malloc((ip_address_length + 1) * sizeof(char));
  ip_address_string.toCharArray(ip_address_char, ip_address_length + 1);

  char *pair_message = (char *)malloc((63 + nb + ip_address_length) * sizeof(char));
  char *mac_address = get_mac_address();

  sprintf(pair_message, "*%d {\"action\":\"pair\",\"ip_address\":\"%s\",\"mac_address\":\"%s\"}", 61 + ip_address_length, ip_address_char, mac_address);
  return pair_message;
}

char *construct_bottle_taken_message()
{
  char *bottle_taken_message = (char *)malloc((29) * sizeof(char));

  strcpy(bottle_taken_message, "*25 {\"action\":\"bottle_taken\"}");
  return bottle_taken_message;
}

char *construct_set_battery_message(int battery_level)
{
  char battery_level_as_string[3];
  itoa(battery_level, battery_level_as_string, 10);
  unsigned int message_length = strlen(battery_level_as_string) + 41;
  char *set_battery_message = (char *)malloc((2 + 2 + message_length) * sizeof(char));

  sprintf(set_battery_message, "*%d {\"action\":\"set_battery\",\"battery_level\":%d}", message_length, battery_level);
  return set_battery_message;
}

bool check_protocol()
{
  char check = client.read();
  if (check == PROTOCOL_CHECK)
    return true;
  else
  {
    // The message is not processed
    while (client.available())
      client.read();
    return false;
  }
}

char *get_response_data()
{
  char carac;
  char length_as_string[5];
  int i = 0;

  // Get the length of the data
  while ((carac = client.read()) != ' ')
    length_as_string[i++] = carac;
  length_as_string[i] = '\0';
  int length = atoi(length_as_string);

  // Retrieve data
  char *content = (char *)malloc((length + 1) * sizeof(char));
  for (int i = 0; i < length; i++)
  {
    content[i] = client.read();
  }
  content[length] = '\0';
  return content;
}

char *get_action(char *data)
{
  int i = 11, j = 0;
  while (data[i] != '"')
    i++;
  char *action = (char *)malloc((i - 11) * sizeof(char));
  for (j = 0; j < i - 11; j++)
    action[j] = data[11 + j];
  action[j] = '\0';
  return action;
}