tree.h (wi::fits_to_tree_p): Accept only 0 and 1 for boolean types.

	* tree.h (wi::fits_to_tree_p): Accept only 0 and 1 for boolean types.
	* tree.c (int_fits_type_p): Likewise.  Adjust head comment.

From-SVN: r241529

gcc/ChangeLog

+2016-10-25  Eric Botcazou  <ebotcazou@adacore.com>
+
+	* tree.h (wi::fits_to_tree_p): Accept only 0 and 1 for boolean types.
+	* tree.c (int_fits_type_p): Likewise.  Adjust head comment.
+
 2016-10-25  David Malcolm  <dmalcolm@redhat.com>
 
 	* ggc-tests.c (forcibly_ggc_collect): Rename to...

gcc/testsuite/ChangeLog

+2016-10-25  Eric Botcazou  <ebotcazou@adacore.com>
+
+	* gnat.dg/opt59.adb: New test.
+	* gnat.dg/opt59_pkg.ad[sb]: New helper.
+
+	* gnat.dg/opt60.adb: New test.
+
 2016-10-25  Andre Vehreschild  <vehre@gcc.gnu.org>
 
 	PR fortran/72770

gcc/testsuite/gnat.dg/opt59.adb

+-- { dg-do run }
+-- { dg-options "-O" }
+
+with Opt59_Pkg; use Opt59_Pkg;
+
+procedure Opt59 is
+
+  type Enum is (Zero, One, Two);
+
+  function Has_True (V : Boolean_Vector) return Boolean is
+  begin
+     for I in V'Range loop
+        if V (I) then
+           return True;
+        end if;
+     end loop;
+     return False;
+  end;
+
+  Data1  : constant Boolean_Vector := Get_BV1;
+  Data2  : constant Boolean_Vector := Get_BV2;
+  Result : Boolean_Vector;
+
+  function F return Enum is
+    Res  : Enum := Zero;
+    Set1 : constant Boolean := Has_True (Data1);
+    Set2 : constant Boolean := Has_True (Data2);
+  begin
+    if Set1 then
+      Res := Two;
+    elsif Set2 then
+      Res := One;
+    end if;
+    return Res;
+  end;
+
+  Val : constant Enum := F;
+
+begin
+
+  for I in Result'Range loop
+    Result (I) := Data1 (I) or Data2 (I);
+  end loop;
+
+  if Val /= Zero then
+    Test (Val = Two);
+  end if;
+
+end;

gcc/testsuite/gnat.dg/opt59_pkg.adb

+package body Opt59_Pkg is
+
+  function Get_BV1 return Boolean_Vector is
+  begin
+    return (others => True);
+  end;
+
+  function Get_BV2 return Boolean_Vector is
+  begin
+    return (others => False);
+  end;
+
+  procedure Test (B : Boolean) is
+  begin
+    if not B then
+      raise Program_Error;
+    end if;
+  end;
+
+end Opt59_Pkg;

gcc/testsuite/gnat.dg/opt59_pkg.ads

+package Opt59_Pkg is
+
+  type Boolean_Vector is array (1 .. 8) of Boolean;
+
+  function Get_BV1 return Boolean_Vector;
+
+  function Get_BV2 return Boolean_Vector;
+
+  procedure Test (B : Boolean);
+
+end Opt59_Pkg;

gcc/testsuite/gnat.dg/opt60.adb

+-- { dg-do compile }
+-- { dg-options "-gnatws -O2 -fdump-tree-optimized" }
+
+with System; use System;
+with System.CRTL; use System.CRTL;
+
+function Opt60 (Size : size_t) return System.Address is
+  Result : System.Address;
+begin
+  Result := malloc (Size);
+  if Result = System.Null_Address then
+    raise Program_Error;
+  end if;
+  return Result;
+end;
+
+-- { dg-final { scan-tree-dump "== 0B" "optimized" } }

gcc/tree.c

@@ -9065,8 +9065,8 @@ get_narrower (tree op, int *unsignedp_ptr)
   return win;
 }
 
-/* Returns true if integer constant C has a value that is permissible
-   for type TYPE (an INTEGER_TYPE).  */
+/* Return true if integer constant C has a value that is permissible
+   for TYPE, an integral type.  */
 
 bool
 int_fits_type_p (const_tree c, const_tree type)
@@ -9075,6 +9075,11 @@ int_fits_type_p (const_tree c, const_tree type)
   bool ok_for_low_bound, ok_for_high_bound;
   signop sgn_c = TYPE_SIGN (TREE_TYPE (c));
 
+  /* Short-circuit boolean types since various transformations assume that
+     they can only take values 0 and 1.  */
+  if (TREE_CODE (type) == BOOLEAN_TYPE)
+    return integer_zerop (c) || integer_onep (c);
+
 retry:
   type_low_bound = TYPE_MIN_VALUE (type);
   type_high_bound = TYPE_MAX_VALUE (type);

gcc/tree.h

@@ -5295,6 +5295,11 @@ template <typename T>
 bool
 wi::fits_to_tree_p (const T &x, const_tree type)
 {
+  /* Short-circuit boolean types since various transformations assume that
+     they can only take values 0 and 1.  */
+  if (TREE_CODE (type) == BOOLEAN_TYPE)
+    return eq_p (x, 0) || eq_p (x, 1);
+
   if (TYPE_SIGN (type) == UNSIGNED)
     return eq_p (x, zext (x, TYPE_PRECISION (type)));
   else