#define NULL ((void *)0)
typedef unsigned long size_t; // Customize by platform.
typedef long intptr_t;
typedef unsigned long uintptr_t;
typedef long scalar_t__; // Either arithmetic or pointer type.
/* By default, we understand bool (as a convenience). */
typedef int bool;
#define false 0
#define true 1

/* Forward declarations */

/* Type definitions */

/* Variables and functions */
size_t rws_check_recv_frame_size(const void *data, const size_t data_size) {
  if (data && data_size >= 2) {
    const unsigned char *udata = (const unsigned char *)data;
    const unsigned int is_masked = (udata[1] >> 7) & 0x01;
    const unsigned int payload = udata[1] & 0x7f;
    unsigned int header_size = is_masked ? 6 : 2;
    unsigned int expected_size = 0;
    switch (payload) {
    case 126:
      header_size += 2;
      break;
    case 127:
      header_size += 8;
      break;
    default:
      break;
    }
    if (data_size < header_size) {
      return 0;
    }
    switch (payload) {
    case 126:
      expected_size |= ((unsigned int)udata[2]) << 8;
      expected_size |= (unsigned int)udata[3];
      break;
    case 127:
      expected_size |= ((unsigned int)udata[6]) << 24;
      expected_size |= ((unsigned int)udata[7]) << 16;
      expected_size |= ((unsigned int)udata[8]) << 8;
      expected_size |= (unsigned int)udata[9];
      break;
    default:
      if (payload <= 125) {
        expected_size = payload;
      }
      break;
    }
    const unsigned int nPackSize = expected_size + header_size;
    return (nPackSize <= data_size) ? nPackSize : 0;
  }
  return 0;
}


x86 assembly (clang, -O0):
<rws_check_recv_frame_size>:
push   %rbp
mov    %rsp,%rbp
mov    %rdi,-0x10(%rbp)
mov    %rsi,-0x18(%rbp)
cmpq   $0x0,-0x10(%rbp)
je     12a4 <rws_check_recv_frame_size+0x1a4>
cmpq   $0x2,-0x18(%rbp)
jb     12a4 <rws_check_recv_frame_size+0x1a4>
mov    -0x10(%rbp),%rax
mov    %rax,-0x20(%rbp)
mov    -0x20(%rbp),%rax
movzbl 0x1(%rax),%eax
shr    $0x7,%eax
mov    %eax,-0x24(%rbp)
mov    -0x20(%rbp),%rax
movzbl 0x1(%rax),%eax
and    $0x7f,%eax
mov    %eax,-0x28(%rbp)
mov    -0x24(%rbp),%eax
test   %eax,%eax
setne  %al
movzbl %al,%ecx
mov    %ecx,%eax
lea    0x2(,%rax,4),%eax
mov    %eax,-0x2c(%rbp)
movl   $0x0,-0x30(%rbp)
mov    -0x28(%rbp),%eax
mov    %eax,-0x38(%rbp)
sub    $0x7e,%eax
je     1189 <rws_check_recv_frame_size+0x89>
jmp    1178 <rws_check_recv_frame_size+0x78>
mov    -0x38(%rbp),%eax
sub    $0x7f,%eax
je     1197 <rws_check_recv_frame_size+0x97>
jmp    11a5 <rws_check_recv_frame_size+0xa5>
mov    -0x2c(%rbp),%eax
add    $0x2,%eax
mov    %eax,-0x2c(%rbp)
jmp    11aa <rws_check_recv_frame_size+0xaa>
mov    -0x2c(%rbp),%eax
add    $0x8,%eax
mov    %eax,-0x2c(%rbp)
jmp    11aa <rws_check_recv_frame_size+0xaa>
jmp    11aa <rws_check_recv_frame_size+0xaa>
mov    -0x18(%rbp),%rax
mov    -0x2c(%rbp),%ecx
cmp    %rcx,%rax
jae    11c7 <rws_check_recv_frame_size+0xc7>
movq   $0x0,-0x8(%rbp)
jmp    12ac <rws_check_recv_frame_size+0x1ac>
mov    -0x28(%rbp),%eax
mov    %eax,-0x3c(%rbp)
sub    $0x7e,%eax
je     11ec <rws_check_recv_frame_size+0xec>
jmp    11db <rws_check_recv_frame_size+0xdb>
mov    -0x3c(%rbp),%eax
sub    $0x7f,%eax
je     1210 <rws_check_recv_frame_size+0x110>
jmp    1256 <rws_check_recv_frame_size+0x156>
mov    -0x20(%rbp),%rax
movzbl 0x2(%rax),%eax
shl    $0x8,%eax
or     -0x30(%rbp),%eax
mov    %eax,-0x30(%rbp)
mov    -0x20(%rbp),%rax
movzbl 0x3(%rax),%eax
or     -0x30(%rbp),%eax
mov    %eax,-0x30(%rbp)
jmp    126b <rws_check_recv_frame_size+0x16b>
mov    -0x20(%rbp),%rax
movzbl 0x6(%rax),%eax
shl    $0x18,%eax
or     -0x30(%rbp),%eax
mov    %eax,-0x30(%rbp)
mov    -0x20(%rbp),%rax
movzbl 0x7(%rax),%eax
shl    $0x10,%eax
or     -0x30(%rbp),%eax
mov    %eax,-0x30(%rbp)
mov    -0x20(%rbp),%rax
movzbl 0x8(%rax),%eax
shl    $0x8,%eax
or     -0x30(%rbp),%eax
mov    %eax,-0x30(%rbp)
mov    -0x20(%rbp),%rax
movzbl 0x9(%rax),%eax
or     -0x30(%rbp),%eax
mov    %eax,-0x30(%rbp)
jmp    126b <rws_check_recv_frame_size+0x16b>
cmpl   $0x7d,-0x28(%rbp)
ja     1266 <rws_check_recv_frame_size+0x166>
mov    -0x28(%rbp),%eax
mov    %eax,-0x30(%rbp)
jmp    126b <rws_check_recv_frame_size+0x16b>
mov    -0x30(%rbp),%eax
add    -0x2c(%rbp),%eax
mov    %eax,-0x34(%rbp)
mov    -0x34(%rbp),%eax
cmp    -0x18(%rbp),%rax
ja     128c <rws_check_recv_frame_size+0x18c>
mov    -0x34(%rbp),%eax
mov    %eax,-0x40(%rbp)
jmp    1296 <rws_check_recv_frame_size+0x196>
xor    %eax,%eax
mov    %eax,-0x40(%rbp)
jmp    1296 <rws_check_recv_frame_size+0x196>
mov    -0x40(%rbp),%eax
mov    %eax,%eax
mov    %rax,-0x8(%rbp)
jmp    12ac <rws_check_recv_frame_size+0x1ac>
movq   $0x0,-0x8(%rbp)
mov    -0x8(%rbp),%rax
pop    %rbp
ret

x86 assembly (clang, -O2):
<rws_check_recv_frame_size>:
xor    %eax,%eax
test   %rdi,%rdi
je     116e <rws_check_recv_frame_size+0x6e>
cmp    $0x2,%rsi
jb     116e <rws_check_recv_frame_size+0x6e>
movzbl 0x1(%rdi),%eax
mov    %eax,%ecx
and    $0x7f,%ecx
shr    $0x7,%al
movzbl %al,%eax
lea    0x2(,%rax,4),%eax
cmp    $0x7f,%ecx
je     1132 <rws_check_recv_frame_size+0x32>
cmp    $0x7e,%ecx
jne    1135 <rws_check_recv_frame_size+0x35>
add    $0x2,%eax
jmp    1135 <rws_check_recv_frame_size+0x35>
or     $0x8,%eax
mov    %eax,%edx
cmp    %rsi,%rdx
jbe    113f <rws_check_recv_frame_size+0x3f>
xor    %eax,%eax
ret
cmp    $0x7f,%ecx
je     1154 <rws_check_recv_frame_size+0x54>
cmp    $0x7e,%ecx
jne    115b <rws_check_recv_frame_size+0x5b>
movzwl 0x2(%rdi),%edx
shl    $0x10,%edx
bswap  %edx
jmp    1163 <rws_check_recv_frame_size+0x63>
mov    0x6(%rdi),%edx
bswap  %edx
jmp    1163 <rws_check_recv_frame_size+0x63>
xor    %edx,%edx
cmp    $0x7e,%cl
cmovb  %ecx,%edx
add    %eax,%edx
xor    %eax,%eax
cmp    %rsi,%rdx
cmovbe %rdx,%rax
ret

LLVM IR (-O0):
define i64 @rws_check_recv_frame_size(i8* %0, i64 %1) {
%3 = alloca i64
%4 = alloca i8*
%5 = alloca i64
%6 = alloca i8*
%7 = alloca i32
%8 = alloca i32
%9 = alloca i32
%10 = alloca i32
%11 = alloca i32
store i8* %0, i8** %4
store i64 %1, i64* %5
%12 = load i8*, i8** %4
%13 = icmp ne i8* %12, null
br i1 %13, label %14, label %112
14: ; preds = %2
%15 = load i64, i64* %5
%16 = icmp uge i64 %15, 2
br i1 %16, label %17, label %112
17: ; preds = %14
%18 = load i8*, i8** %4
store i8* %18, i8** %6
%19 = load i8*, i8** %6
%20 = getelementptr inbounds i8, i8* %19, i64 1
%21 = load i8, i8* %20
%22 = zext i8 %21 to i32
%23 = ashr i32 %22, 7
%24 = and i32 %23, 1
store i32 %24, i32* %7
%25 = load i8*, i8** %6
%26 = getelementptr inbounds i8, i8* %25, i64 1
%27 = load i8, i8* %26
%28 = zext i8 %27 to i32
%29 = and i32 %28, 127
store i32 %29, i32* %8
%30 = load i32, i32* %7
%31 = icmp ne i32 %30, 0
%32 = zext i1 %31 to i64
%33 = select i1 %31, i32 6, i32 2
store i32 %33, i32* %9
store i32 0, i32* %10
%34 = load i32, i32* %8
switch i32 %34, label %41 [
i32 126, label %35
i32 127, label %38
]
35: ; preds = %17
%36 = load i32, i32* %9
%37 = add i32 %36, 2
store i32 %37, i32* %9
br label %42
38: ; preds = %17
%39 = load i32, i32* %9
%40 = add i32 %39, 8
store i32 %40, i32* %9
br label %42
41: ; preds = %17
br label %42
42: ; preds = %41, %38, %35
%43 = load i64, i64* %5
%44 = load i32, i32* %9
%45 = zext i32 %44 to i64
%46 = icmp ult i64 %43, %45
br i1 %46, label %47, label %48
47: ; preds = %42
store i64 0, i64* %3
br label %113
48: ; preds = %42
%49 = load i32, i32* %8
switch i32 %49, label %92 [
i32 126, label %50
i32 127, label %64
]
50: ; preds = %48
%51 = load i8*, i8** %6
%52 = getelementptr inbounds i8, i8* %51, i64 2
%53 = load i8, i8* %52
%54 = zext i8 %53 to i32
%55 = shl i32 %54, 8
%56 = load i32, i32* %10
%57 = or i32 %56, %55
store i32 %57, i32* %10
%58 = load i8*, i8** %6
%59 = getelementptr inbounds i8, i8* %58, i64 3
%60 = load i8, i8* %59
%61 = zext i8 %60 to i32
%62 = load i32, i32* %10
%63 = or i32 %62, %61
store i32 %63, i32* %10
br label %98
64: ; preds = %48
%65 = load i8*, i8** %6
%66 = getelementptr inbounds i8, i8* %65, i64 6
%67 = load i8, i8* %66
%68 = zext i8 %67 to i32
%69 = shl i32 %68, 24
%70 = load i32, i32* %10
%71 = or i32 %70, %69
store i32 %71, i32* %10
%72 = load i8*, i8** %6
%73 = getelementptr inbounds i8, i8* %72, i64 7
%74 = load i8, i8* %73
%75 = zext i8 %74 to i32
%76 = shl i32 %75, 16
%77 = load i32, i32* %10
%78 = or i32 %77, %76
store i32 %78, i32* %10
%79 = load i8*, i8** %6
%80 = getelementptr inbounds i8, i8* %79, i64 8
%81 = load i8, i8* %80
%82 = zext i8 %81 to i32
%83 = shl i32 %82, 8
%84 = load i32, i32* %10
%85 = or i32 %84, %83
store i32 %85, i32* %10
%86 = load i8*, i8** %6
%87 = getelementptr inbounds i8, i8* %86, i64 9
%88 = load i8, i8* %87
%89 = zext i8 %88 to i32
%90 = load i32, i32* %10
%91 = or i32 %90, %89
store i32 %91, i32* %10
br label %98
92: ; preds = %48
%93 = load i32, i32* %8
%94 = icmp ule i32 %93, 125
br i1 %94, label %95, label %97
95: ; preds = %92
%96 = load i32, i32* %8
store i32 %96, i32* %10
br label %97
97: ; preds = %95, %92
br label %98
98: ; preds = %97, %64, %50
%99 = load i32, i32* %10
%100 = load i32, i32* %9
%101 = add i32 %99, %100
store i32 %101, i32* %11
%102 = load i32, i32* %11
%103 = zext i32 %102 to i64
%104 = load i64, i64* %5
%105 = icmp ule i64 %103, %104
br i1 %105, label %106, label %108
106: ; preds = %98
%107 = load i32, i32* %11
br label %109
108: ; preds = %98
br label %109
109: ; preds = %108, %106
%110 = phi i32 [ %107, %106 ], [ 0, %108 ]
%111 = zext i32 %110 to i64
store i64 %111, i64* %3
br label %113
112: ; preds = %14, %2
store i64 0, i64* %3
br label %113
113: ; preds = %112, %109, %47
%114 = load i64, i64* %3
ret i64 %114
}

LLVM IR (-O2):
define i64 @rws_check_recv_frame_size(i8* %0, i64 %1) {
%3 = icmp ne i8* %0, null
%4 = icmp ugt i64 %1, 1
%5 = and i1 %3, %4
br i1 %5, label %6, label %59
6: ; preds = %2
%7 = getelementptr inbounds i8, i8* %0, i64 1
%8 = load i8, i8* %7
%9 = and i8 %8, 127
%10 = zext i8 %9 to i32
%11 = icmp sgt i8 %8, -1
%12 = select i1 %11, i32 2, i32 6
switch i32 %10, label %17 [
i32 126, label %13
i32 127, label %15
]
13: ; preds = %6
%14 = add nuw nsw i32 %12, 2
br label %17
15: ; preds = %6
%16 = or i32 %12, 8
br label %17
17: ; preds = %6, %15, %13
%18 = phi i32 [ %12, %6 ], [ %16, %15 ], [ %14, %13 ]
%19 = zext i32 %18 to i64
%20 = icmp ugt i64 %19, %1
br i1 %20, label %59, label %21
21: ; preds = %17
switch i32 %10, label %50 [
i32 126, label %22
i32 127, label %31
]
22: ; preds = %21
%23 = getelementptr inbounds i8, i8* %0, i64 2
%24 = load i8, i8* %23
%25 = zext i8 %24 to i32
%26 = shl nuw nsw i32 %25, 8
%27 = getelementptr inbounds i8, i8* %0, i64 3
%28 = load i8, i8* %27
%29 = zext i8 %28 to i32
%30 = or i32 %26, %29
br label %53
31: ; preds = %21
%32 = getelementptr inbounds i8, i8* %0, i64 6
%33 = load i8, i8* %32
%34 = zext i8 %33 to i32
%35 = shl nuw i32 %34, 24
%36 = getelementptr inbounds i8, i8* %0, i64 7
%37 = load i8, i8* %36
%38 = zext i8 %37 to i32
%39 = shl nuw nsw i32 %38, 16
%40 = or i32 %39, %35
%41 = getelementptr inbounds i8, i8* %0, i64 8
%42 = load i8, i8* %41
%43 = zext i8 %42 to i32
%44 = shl nuw nsw i32 %43, 8
%45 = or i32 %40, %44
%46 = getelementptr inbounds i8, i8* %0, i64 9
%47 = load i8, i8* %46
%48 = zext i8 %47 to i32
%49 = or i32 %45, %48
br label %53
50: ; preds = %21
%51 = icmp ult i8 %9, 126
%52 = select i1 %51, i32 %10, i32 0
br label %53
53: ; preds = %50, %31, %22
%54 = phi i32 [ %49, %31 ], [ %30, %22 ], [ %52, %50 ]
%55 = add i32 %54, %18
%56 = zext i32 %55 to i64
%57 = icmp ugt i64 %56, %1
%58 = select i1 %57, i64 0, i64 %56
br label %59
59: ; preds = %2, %53, %17
%60 = phi i64 [ %58, %53 ], [ 0, %17 ], [ 0, %2 ]
ret i64 %60
}