#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 */
struct udphdr {
  int dest;
};
struct tcphdr {
  int dest;
};
typedef int /*<<< orphan*/ __u64;
typedef int __u16;

/* Variables and functions */
#define IPPROTO_TCP 129
#define IPPROTO_UDP 128
__u16 get_dst_port(void *pkt, void *end, __u64 offset, int protocol) {
  struct udphdr *udp;
  struct tcphdr *tcp;
  switch (protocol) {
  case IPPROTO_UDP:
    udp = pkt + offset;
    if (pkt + offset + sizeof(*udp) > end)
      return 0;
    return (((udp->dest & 0xFF00) >> 8) | ((udp->dest & 0x00FF) << 8));
  case IPPROTO_TCP:
    tcp = pkt + offset;
    if (pkt + offset + sizeof(*tcp) > end)
      return 0;
    return (((tcp->dest & 0xFF00) >> 8) | ((tcp->dest & 0x00FF) << 8));
  }
  return 0;
}


x86 assembly (clang, -O0):
<get_dst_port>:
push   %rbp
mov    %rsp,%rbp
mov    %rdi,-0x10(%rbp)
mov    %rsi,-0x18(%rbp)
mov    %edx,-0x1c(%rbp)
mov    %ecx,-0x20(%rbp)
mov    -0x20(%rbp),%eax
mov    %eax,-0x34(%rbp)
sub    $0x80,%eax
je     113b <get_dst_port+0x3b>
jmp    1128 <get_dst_port+0x28>
mov    -0x34(%rbp),%eax
sub    $0x81,%eax
je     1196 <get_dst_port+0x96>
jmp    11f1 <get_dst_port+0xf1>
mov    -0x10(%rbp),%rax
movslq -0x1c(%rbp),%rcx
add    %rcx,%rax
mov    %rax,-0x28(%rbp)
mov    -0x10(%rbp),%rax
movslq -0x1c(%rbp),%rcx
add    %rcx,%rax
add    $0x4,%rax
cmp    -0x18(%rbp),%rax
jbe    116f <get_dst_port+0x6f>
movl   $0x0,-0x4(%rbp)
jmp    11f8 <get_dst_port+0xf8>
mov    -0x28(%rbp),%rax
mov    (%rax),%eax
and    $0xff00,%eax
sar    $0x8,%eax
mov    -0x28(%rbp),%rcx
mov    (%rcx),%ecx
and    $0xff,%ecx
shl    $0x8,%ecx
or     %ecx,%eax
mov    %eax,-0x4(%rbp)
jmp    11f8 <get_dst_port+0xf8>
mov    -0x10(%rbp),%rax
movslq -0x1c(%rbp),%rcx
add    %rcx,%rax
mov    %rax,-0x30(%rbp)
mov    -0x10(%rbp),%rax
movslq -0x1c(%rbp),%rcx
add    %rcx,%rax
add    $0x4,%rax
cmp    -0x18(%rbp),%rax
jbe    11ca <get_dst_port+0xca>
movl   $0x0,-0x4(%rbp)
jmp    11f8 <get_dst_port+0xf8>
mov    -0x30(%rbp),%rax
mov    (%rax),%eax
and    $0xff00,%eax
sar    $0x8,%eax
mov    -0x30(%rbp),%rcx
mov    (%rcx),%ecx
and    $0xff,%ecx
shl    $0x8,%ecx
or     %ecx,%eax
mov    %eax,-0x4(%rbp)
jmp    11f8 <get_dst_port+0xf8>
movl   $0x0,-0x4(%rbp)
mov    -0x4(%rbp),%eax
pop    %rbp
ret

x86 assembly (clang, -O2):
<get_dst_port>:
xor    %eax,%eax
cmp    $0x81,%ecx
je     1112 <get_dst_port+0x12>
cmp    $0x80,%ecx
jne    112f <get_dst_port+0x2f>
movslq %edx,%rcx
lea    (%rdi,%rcx,1),%rdx
add    $0x4,%rdx
cmp    %rsi,%rdx
ja     112f <get_dst_port+0x2f>
add    %rcx,%rdi
movzwl (%rdi),%eax
rol    $0x8,%ax
movzwl %ax,%eax
ret

LLVM IR (-O0):
%struct.udphdr = type { i32 }
%struct.tcphdr = type { i32 }
define i32 @get_dst_port(i8* %0, i8* %1, i32 %2, i32 %3) {
%5 = alloca i32
%6 = alloca i8*
%7 = alloca i8*
%8 = alloca i32
%9 = alloca i32
%10 = alloca %struct.udphdr*
%11 = alloca %struct.tcphdr*
store i8* %0, i8** %6
store i8* %1, i8** %7
store i32 %2, i32* %8
store i32 %3, i32* %9
%12 = load i32, i32* %9
switch i32 %12, label %65 [
i32 128, label %13
i32 129, label %39
]
13: ; preds = %4
%14 = load i8*, i8** %6
%15 = load i32, i32* %8
%16 = sext i32 %15 to i64
%17 = getelementptr i8, i8* %14, i64 %16
%18 = bitcast i8* %17 to %struct.udphdr*
store %struct.udphdr* %18, %struct.udphdr** %10
%19 = load i8*, i8** %6
%20 = load i32, i32* %8
%21 = sext i32 %20 to i64
%22 = getelementptr i8, i8* %19, i64 %21
%23 = getelementptr i8, i8* %22, i64 4
%24 = load i8*, i8** %7
%25 = icmp ugt i8* %23, %24
br i1 %25, label %26, label %27
26: ; preds = %13
store i32 0, i32* %5
br label %66
27: ; preds = %13
%28 = load %struct.udphdr*, %struct.udphdr** %10
%29 = getelementptr inbounds %struct.udphdr, %struct.udphdr* %28, i32 0, i32 0
%30 = load i32, i32* %29
%31 = and i32 %30, 65280
%32 = ashr i32 %31, 8
%33 = load %struct.udphdr*, %struct.udphdr** %10
%34 = getelementptr inbounds %struct.udphdr, %struct.udphdr* %33, i32 0, i32 0
%35 = load i32, i32* %34
%36 = and i32 %35, 255
%37 = shl i32 %36, 8
%38 = or i32 %32, %37
store i32 %38, i32* %5
br label %66
39: ; preds = %4
%40 = load i8*, i8** %6
%41 = load i32, i32* %8
%42 = sext i32 %41 to i64
%43 = getelementptr i8, i8* %40, i64 %42
%44 = bitcast i8* %43 to %struct.tcphdr*
store %struct.tcphdr* %44, %struct.tcphdr** %11
%45 = load i8*, i8** %6
%46 = load i32, i32* %8
%47 = sext i32 %46 to i64
%48 = getelementptr i8, i8* %45, i64 %47
%49 = getelementptr i8, i8* %48, i64 4
%50 = load i8*, i8** %7
%51 = icmp ugt i8* %49, %50
br i1 %51, label %52, label %53
52: ; preds = %39
store i32 0, i32* %5
br label %66
53: ; preds = %39
%54 = load %struct.tcphdr*, %struct.tcphdr** %11
%55 = getelementptr inbounds %struct.tcphdr, %struct.tcphdr* %54, i32 0, i32 0
%56 = load i32, i32* %55
%57 = and i32 %56, 65280
%58 = ashr i32 %57, 8
%59 = load %struct.tcphdr*, %struct.tcphdr** %11
%60 = getelementptr inbounds %struct.tcphdr, %struct.tcphdr* %59, i32 0, i32 0
%61 = load i32, i32* %60
%62 = and i32 %61, 255
%63 = shl i32 %62, 8
%64 = or i32 %58, %63
store i32 %64, i32* %5
br label %66
65: ; preds = %4
store i32 0, i32* %5
br label %66
66: ; preds = %65, %53, %52, %27, %26
%67 = load i32, i32* %5
ret i32 %67
}

LLVM IR (-O2):
define i32 @get_dst_port(i8* %0, i8* readnone %1, i32 %2, i32 %3) {
switch i32 %3, label %22 [
i32 128, label %5
i32 129, label %10
]
5: ; preds = %4
%6 = sext i32 %2 to i64
%7 = getelementptr i8, i8* %0, i64 %6
%8 = getelementptr i8, i8* %7, i64 4
%9 = icmp ugt i8* %8, %1
br i1 %9, label %22, label %15
10: ; preds = %4
%11 = sext i32 %2 to i64
%12 = getelementptr i8, i8* %0, i64 %11
%13 = getelementptr i8, i8* %12, i64 4
%14 = icmp ugt i8* %13, %1
br i1 %14, label %22, label %15
15: ; preds = %10, %5
%16 = phi i8* [ %7, %5 ], [ %12, %10 ]
%17 = bitcast i8* %16 to i32*
%18 = load i32, i32* %17
%19 = trunc i32 %18 to i16
%20 = tail call i16 @llvm.bswap.i16(i16 %19)
%21 = zext i16 %20 to i32
br label %22
22: ; preds = %15, %4, %10, %5
%23 = phi i32 [ 0, %5 ], [ 0, %10 ], [ 0, %4 ], [ %21, %15 ]
ret i32 %23
}