<p>Moises Silva has uploaded this change for <strong>review</strong>.</p><p><a href="https://gerrit.asterisk.org/c/asterisk/+/14520">View Change</a></p><pre style="font-family: monospace,monospace; white-space: pre-wrap;">res_http_websocket: Add data masking to the websocket client<br><br>ASTERISK-28949<br><br>Change-Id: Id465030f2b1997b83d408933fdbabe01827469ca<br>---<br>M res/res_http_websocket.c<br>1 file changed, 29 insertions(+), 0 deletions(-)<br><br></pre><pre style="font-family: monospace,monospace; white-space: pre-wrap;">git pull ssh://gerrit.asterisk.org:29418/asterisk refs/changes/20/14520/1</pre><pre style="font-family: monospace,monospace; white-space: pre-wrap;"><span>diff --git a/res/res_http_websocket.c b/res/res_http_websocket.c</span><br><span>index 63fccdd..2bf04d7 100644</span><br><span>--- a/res/res_http_websocket.c</span><br><span>+++ b/res/res_http_websocket.c</span><br><span>@@ -283,6 +283,9 @@</span><br><span>       return 0;</span><br><span> }</span><br><span> </span><br><span style="color: hsl(120, 100%, 40%);">+/*! \brief Perform payload masking for client sessions */</span><br><span style="color: hsl(120, 100%, 40%);">+static void websocket_mask_payload(struct ast_websocket *session, char *frame, char *payload, uint64_t payload_size);</span><br><span style="color: hsl(120, 100%, 40%);">+</span><br><span> /*! \brief Close function for websocket session */</span><br><span> int AST_OPTIONAL_API_NAME(ast_websocket_close)(struct ast_websocket *session, uint16_t reason)</span><br><span> {</span><br><span>@@ -300,6 +303,8 @@</span><br><span>    /* If no reason has been specified assume 1000 which is normal closure */</span><br><span>    put_unaligned_uint16(&frame[2], htons(reason ? reason : 1000));</span><br><span> </span><br><span style="color: hsl(120, 100%, 40%);">+       websocket_mask_payload(session, frame, &frame[2], 2);</span><br><span style="color: hsl(120, 100%, 40%);">+</span><br><span>  session->closing = 1;</span><br><span>     session->close_sent = 1;</span><br><span> </span><br><span>@@ -341,6 +346,23 @@</span><br><span>       }</span><br><span> }</span><br><span> </span><br><span style="color: hsl(120, 100%, 40%);">+static void websocket_mask_payload(struct ast_websocket *session, char *frame, char *payload, uint64_t payload_size)</span><br><span style="color: hsl(120, 100%, 40%);">+{</span><br><span style="color: hsl(120, 100%, 40%);">+   /* RFC 6455 5.1 - clients MUST mask frame data */</span><br><span style="color: hsl(120, 100%, 40%);">+     if (session->client) {</span><br><span style="color: hsl(120, 100%, 40%);">+             uint64_t i;</span><br><span style="color: hsl(120, 100%, 40%);">+           uint8_t mask_key_idx;</span><br><span style="color: hsl(120, 100%, 40%);">+         uint32_t mask_key = ast_random();</span><br><span style="color: hsl(120, 100%, 40%);">+             uint8_t length = frame[1] & 0x7f;</span><br><span style="color: hsl(120, 100%, 40%);">+         frame[1] |= 0x80; /* set mask bit to 1 */</span><br><span style="color: hsl(120, 100%, 40%);">+             mask_key_idx = length == 126 ? 4 : length == 127 ? 10 : 2;</span><br><span style="color: hsl(120, 100%, 40%);">+            put_unaligned_uint32(&frame[mask_key_idx], mask_key);</span><br><span style="color: hsl(120, 100%, 40%);">+             for (i = 0; i < payload_size; i++) {</span><br><span style="color: hsl(120, 100%, 40%);">+                       payload[i] ^= ((char *)&mask_key)[i % 4];</span><br><span style="color: hsl(120, 100%, 40%);">+         }</span><br><span style="color: hsl(120, 100%, 40%);">+     }</span><br><span style="color: hsl(120, 100%, 40%);">+}</span><br><span style="color: hsl(120, 100%, 40%);">+</span><br><span> /*! \brief Write function for websocket traffic */</span><br><span> int AST_OPTIONAL_API_NAME(ast_websocket_write)(struct ast_websocket *session, enum ast_websocket_opcode opcode, char *payload, uint64_t payload_size)</span><br><span> {</span><br><span>@@ -364,6 +386,11 @@</span><br><span>            header_size += 8;</span><br><span>    }</span><br><span> </span><br><span style="color: hsl(120, 100%, 40%);">+ if (session->client) {</span><br><span style="color: hsl(120, 100%, 40%);">+             /* Additional 4 bytes for the client mask key */</span><br><span style="color: hsl(120, 100%, 40%);">+              header_size += 4;</span><br><span style="color: hsl(120, 100%, 40%);">+     }</span><br><span style="color: hsl(120, 100%, 40%);">+</span><br><span>  frame_size = header_size + payload_size;</span><br><span> </span><br><span>         frame = ast_alloca(frame_size + 1);</span><br><span>@@ -381,6 +408,8 @@</span><br><span> </span><br><span>        memcpy(&frame[header_size], payload, payload_size);</span><br><span> </span><br><span style="color: hsl(120, 100%, 40%);">+   websocket_mask_payload(session, frame, &frame[header_size], payload_size);</span><br><span style="color: hsl(120, 100%, 40%);">+</span><br><span>     ao2_lock(session);</span><br><span>   if (session->closing) {</span><br><span>           ao2_unlock(session);</span><br><span></span><br></pre><p>To view, visit <a href="https://gerrit.asterisk.org/c/asterisk/+/14520">change 14520</a>. To unsubscribe, or for help writing mail filters, visit <a href="https://gerrit.asterisk.org/settings">settings</a>.</p><div itemscope itemtype="http://schema.org/EmailMessage"><div itemscope itemprop="action" itemtype="http://schema.org/ViewAction"><link itemprop="url" href="https://gerrit.asterisk.org/c/asterisk/+/14520"/><meta itemprop="name" content="View Change"/></div></div>

<div style="display:none"> Gerrit-Project: asterisk </div>
<div style="display:none"> Gerrit-Branch: master </div>
<div style="display:none"> Gerrit-Change-Id: Id465030f2b1997b83d408933fdbabe01827469ca </div>
<div style="display:none"> Gerrit-Change-Number: 14520 </div>
<div style="display:none"> Gerrit-PatchSet: 1 </div>
<div style="display:none"> Gerrit-Owner: Moises Silva <moises.silva@gmail.com> </div>
<div style="display:none"> Gerrit-MessageType: newchange </div>