Change >> behaviour in LLVM to prevent stupid UB

This commit is contained in:
gingerBill
2021-03-30 20:38:04 +01:00
parent fbd01660ee
commit 3359d0323a
2 changed files with 7 additions and 8 deletions

View File

@@ -6600,7 +6600,6 @@ handle_op:
LLVMValueRef bits = rhs.value;
LLVMValueRef bit_size = LLVMConstInt(lb_type(p->module, rhs.type), 8*type_size_of(lhs.type), false);
LLVMValueRef max = LLVMConstInt(lb_type(p->module, rhs.type), 8*type_size_of(lhs.type)-1, false);
LLVMValueRef width_test = LLVMBuildICmp(p->builder, LLVMIntULT, bits, bit_size, "");
@@ -6617,16 +6616,17 @@ handle_op:
bool is_unsigned = is_type_unsigned(type);
LLVMValueRef bit_size = LLVMConstInt(lb_type(p->module, rhs.type), 8*type_size_of(lhs.type), false);
LLVMValueRef max = LLVMConstInt(lb_type(p->module, rhs.type), 8*type_size_of(lhs.type)-1, false);
LLVMValueRef width_test = LLVMBuildICmp(p->builder, LLVMIntULT, bits, bit_size, "");
bits = LLVMBuildSelect(p->builder, width_test, bits, max, "");
if (is_unsigned) {
res.value = LLVMBuildLShr(p->builder, lhs.value, bits, "");
res.value = LLVMBuildLShr(p->builder, lhsval, bits, "");
} else {
res.value = LLVMBuildAShr(p->builder, lhsval, bits, "");
}
LLVMValueRef zero = LLVMConstNull(lb_type(p->module, lhs.type));
res.value = LLVMBuildSelect(p->builder, width_test, res.value, zero, "");
return res;
}
case Token_AndNot: