add lang items, update typeck/trans and add tests

Author: Jorge Aparicio

src/librustc/middle/expr_use_visitor.rs

@@ -584,10 +584,15 @@ impl<'d,'t,'tcx,TYPER:mc::Typer<'tcx>> ExprUseVisitor<'d,'t,'tcx,TYPER> {
             }
 
             ast::ExprAssignOp(_, ref lhs, ref rhs) => {
-                // This will have to change if/when we support
-                // overloaded operators for `+=` and so forth.
                 self.mutate_expr(expr, &**lhs, WriteAndRead);
-                self.consume_expr(&**rhs);
+
+                if self.typer.is_method_call(expr.id) {
+                    let r = ty::ReScope(region::CodeExtent::from_node_id(expr.id));
+                    self.borrow_expr(rhs, r, ty::ImmBorrow, OverloadedOperator);
+                } else {
+                    // built-in assignment operations consume the RHS
+                    self.consume_expr(&**rhs);
+                }
             }
 
             ast::ExprRepeat(ref base, ref count) => {

src/librustc/middle/lang_items.rs

@@ -265,6 +265,17 @@ lets_do_this! {
     RangeToStructLangItem,           "range_to",                range_to_struct;
     RangeFullStructLangItem,         "range_full",              range_full_struct;
 
+    AddAssignTraitLangItem,          "add_assign",              add_assign_trait;
+    BitAndAssignTraitLangItem,       "bitand_assign",           bitand_assign_trait;
+    BitOrAssignTraitLangItem,        "bitor_assign",            bitor_assign_trait;
+    BitXorAssignTraitLangItem,       "bitxor_assign",           bitxor_assign_trait;
+    DivAssignTraitLangItem,          "div_assign",              div_assign_trait;
+    MulAssignTraitLangItem,          "mul_assign",              mul_assign_trait;
+    RemAssignTraitLangItem,          "rem_assign",              rem_assign_trait;
+    ShlAssignTraitLangItem,          "shl_assign",              shl_assign_trait;
+    ShrAssignTraitLangItem,          "shr_assign",              shr_assign_trait;
+    SubAssignTraitLangItem,          "sub_assign",              sub_assign_trait;
+
     UnsafeCellTypeLangItem,          "unsafe_cell",             unsafe_cell_type;
 
     DerefTraitLangItem,              "deref",                   deref_trait;

src/librustc_trans/trans/expr.rs

@@ -1005,7 +1005,15 @@ fn trans_rvalue_stmt_unadjusted<'blk, 'tcx>(bcx: Block<'blk, 'tcx>,
             }
         }
         ast::ExprAssignOp(op, ref dst, ref src) => {
-            trans_assign_op(bcx, expr, op, &**dst, &**src)
+            if ty::is_binopable(bcx.tcx(), node_id_type(bcx, dst.id), op) {
+                trans_assign_op(bcx, expr, op, &**dst, &**src)
+            } else {
+                let dst = unpack_datum!(bcx, trans(bcx, &**dst));
+                let src_datum = unpack_datum!(bcx, trans(bcx, &**src));
+                trans_overloaded_op(bcx, expr, MethodCall::expr(expr.id), dst,
+                                    vec![(src_datum, src.id)], None,
+                                    true).bcx
+            }
         }
         ast::ExprInlineAsm(ref a) => {
             asm::trans_inline_asm(bcx, a)
@@ -1203,8 +1211,9 @@ fn trans_rvalue_dps_unadjusted<'blk, 'tcx>(bcx: Block<'blk, 'tcx>,
                                         "expr_cast of non-trait");
             }
         }
-        ast::ExprAssignOp(op, ref dst, ref src) => {
-            trans_assign_op(bcx, expr, op, &**dst, &**src)
+        ast::ExprAssignOp(..) => {
+            bcx.tcx().sess.span_bug(expr.span,
+                                    "operator assignment (`+=`) should always be an rvalue_stmt");
         }
         _ => {
             bcx.tcx().sess.span_bug(

src/librustc_typeck/check/mod.rs

@@ -2988,19 +2988,7 @@ fn check_expr_with_unifier<'a, 'tcx, F>(fcx: &FnCtxt<'a, 'tcx>,
         let result_t = if is_binop_assignment == SimpleBinop {
             check_user_binop(fcx, expr, lhs, lhs_t, op, rhs)
         } else {
-            fcx.type_error_message(expr.span,
-                                   |actual| {
-                                        format!("binary assignment \
-                                                 operation `{}=` \
-                                                 cannot be applied to \
-                                                 type `{}`",
-                                                ast_util::binop_to_string(op.node),
-                                                actual)
-                                   },
-                                   lhs_t,
-                                   None);
-            check_expr(fcx, &**rhs);
-            fcx.tcx().types.err
+            check_user_binop_assign(fcx, expr, lhs, lhs_t, op, rhs)
         };
 
         fcx.write_ty(expr.id, result_t);
@@ -3049,6 +3037,42 @@ fn check_expr_with_unifier<'a, 'tcx, F>(fcx: &FnCtxt<'a, 'tcx>,
         }, if ast_util::is_by_value_binop(op.node) { AutorefArgs::No } else { AutorefArgs::Yes })
     }
 
+    fn check_user_binop_assign<'a, 'tcx>(fcx: &FnCtxt<'a, 'tcx>,
+                                         ex: &'tcx ast::Expr,
+                                         lhs_expr: &'tcx ast::Expr,
+                                         lhs_resolved_t: Ty<'tcx>,
+                                         op: ast::BinOp,
+                                         rhs: &'tcx P<ast::Expr>) -> Ty<'tcx> {
+        let tcx = fcx.ccx.tcx;
+        let lang = &tcx.lang_items;
+        let (name, trait_did) = match op.node {
+            ast::BiAdd => ("add_assign", lang.add_assign_trait()),
+            ast::BiSub => ("sub_assign", lang.sub_assign_trait()),
+            ast::BiMul => ("mul_assign", lang.mul_assign_trait()),
+            ast::BiDiv => ("div_assign", lang.div_assign_trait()),
+            ast::BiRem => ("rem_assign", lang.rem_assign_trait()),
+            ast::BiBitXor => ("bitxor_assign", lang.bitxor_assign_trait()),
+            ast::BiBitAnd => ("bitand_assign", lang.bitand_assign_trait()),
+            ast::BiBitOr => ("bitor_assign", lang.bitor_assign_trait()),
+            ast::BiShl => ("shl_assign", lang.shl_assign_trait()),
+            ast::BiShr => ("shr_assign", lang.shr_assign_trait()),
+            ast::BiLt | ast::BiLe | ast::BiGe | ast::BiGt | ast::BiEq | ast::BiNe | ast::BiAnd |
+                ast::BiOr =>
+            {
+                check_expr(fcx, &**rhs);
+                return tcx.types.err;
+            }
+        };
+        lookup_op_method(fcx, ex, lhs_resolved_t, token::intern(name),
+                         trait_did, lhs_expr, Some(rhs), || {
+            fcx.type_error_message(ex.span, |actual| {
+                format!("binary operation `{}=` cannot be applied to type `{}`",
+                        ast_util::binop_to_string(op.node),
+                        actual)
+            }, lhs_resolved_t, None)
+        }, AutorefArgs::Yes)
+    }
+
     fn check_user_unop<'a, 'tcx>(fcx: &FnCtxt<'a, 'tcx>,
                                  op_str: &str,
                                  mname: &str,
@@ -3480,10 +3504,6 @@ fn check_expr_with_unifier<'a, 'tcx, F>(fcx: &FnCtxt<'a, 'tcx>,
       ast::ExprAssignOp(op, ref lhs, ref rhs) => {
         check_binop(fcx, expr, op, &**lhs, rhs, BinopAssignment);
 
-        let lhs_t = fcx.expr_ty(&**lhs);
-        let result_t = fcx.expr_ty(expr);
-        demand::suptype(fcx, expr.span, result_t, lhs_t);
-
         let tcx = fcx.tcx();
         if !ty::expr_is_lval(tcx, &**lhs) {
             span_err!(tcx.sess, lhs.span, E0067, "illegal left-hand side expression");
@@ -3494,7 +3514,7 @@ fn check_expr_with_unifier<'a, 'tcx, F>(fcx: &FnCtxt<'a, 'tcx>,
         // Overwrite result of check_binop...this preserves existing behavior
         // but seems quite dubious with regard to user-defined methods
         // and so forth. - Niko
-        if !ty::type_is_error(result_t) {
+        if !ty::type_is_error(fcx.expr_ty(expr)) {
             fcx.write_nil(expr.id);
         }
       }

src/test/compile-fail/op-assign-borrow.rs

@@ -0,0 +1,31 @@
+// Copyright 2015 The Rust Project Developers. See the COPYRIGHT
+// file at the top-level directory of this distribution and at
+// http://rust-lang.org/COPYRIGHT.
+//
+// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
+// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
+// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
+// option. This file may not be copied, modified, or distributed
+// except according to those terms.
+
+use std::ops::AddAssign;
+
+struct Int(i32);
+
+impl AddAssign for Int {
+    fn add_assign(&mut self, rhs: &Int) {
+        self.0 += rhs.0
+    }
+}
+
+fn main() {
+    let mut x = Int(1);
+    x  //~ error: cannot borrow `x` as mutable because it is also borrowed as immutable
+    +=
+    x;  //~ note: previous borrow of `x` occurs here
+
+    let y = Int(2);
+    y  //~ error: cannot borrow immutable local variable `y` as mutable
+    +=
+    Int(1);
+}

src/test/run-pass/op-assign.rs

@@ -0,0 +1,135 @@
+// Copyright 2015 The Rust Project Developers. See the COPYRIGHT
+// file at the top-level directory of this distribution and at
+// http://rust-lang.org/COPYRIGHT.
+//
+// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
+// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
+// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
+// option. This file may not be copied, modified, or distributed
+// except according to those terms.
+
+use std::ops::{
+    AddAssign, BitAndAssign, BitOrAssign, BitXorAssign, DivAssign, MulAssign, RemAssign, ShlAssign,
+    ShrAssign, SubAssign,
+};
+
+#[derive(Debug, PartialEq)]
+struct Int(i32);
+
+fn main() {
+    let mut x = Int(1);
+
+    x += Int(2);
+    assert_eq!(x, Int(0b11));
+
+    x &= Int(0b01);
+    assert_eq!(x, Int(0b01));
+
+    x |= Int(0b10);
+    assert_eq!(x, Int(0b11));
+
+    x ^= Int(0b01);
+    assert_eq!(x, Int(0b10));
+
+    x /= Int(2);
+    assert_eq!(x, Int(1));
+
+    x *= Int(3);
+    assert_eq!(x, Int(3));
+
+    x %= Int(2);
+    assert_eq!(x, Int(1));
+
+    // overloaded RHS
+    x <<= 1u8;
+    assert_eq!(x, Int(2));
+
+    x <<= 1u16;
+    assert_eq!(x, Int(4));
+
+    x >>= 1u8;
+    assert_eq!(x, Int(2));
+
+    x >>= 1u16;
+    assert_eq!(x, Int(1));
+
+    x -= Int(1);
+    assert_eq!(x, Int(0));
+
+    // indexing
+    let mut v = vec![Int(1), Int(2)];
+    v[0] += Int(2);
+    assert_eq!(v[0], Int(3));
+}
+
+impl AddAssign for Int {
+    fn add_assign(&mut self, rhs: &Int) {
+        self.0 += rhs.0;
+    }
+}
+
+impl BitAndAssign for Int {
+    fn bitand_assign(&mut self, rhs: &Int) {
+        self.0 &= rhs.0;
+    }
+}
+
+impl BitOrAssign for Int {
+    fn bitor_assign(&mut self, rhs: &Int) {
+        self.0 |= rhs.0;
+    }
+}
+
+impl BitXorAssign for Int {
+    fn bitxor_assign(&mut self, rhs: &Int) {
+        self.0 ^= rhs.0;
+    }
+}
+
+impl DivAssign for Int {
+    fn div_assign(&mut self, rhs: &Int) {
+        self.0 /= rhs.0;
+    }
+}
+
+impl MulAssign for Int {
+    fn mul_assign(&mut self, rhs: &Int) {
+        self.0 *= rhs.0;
+    }
+}
+
+impl RemAssign for Int {
+    fn rem_assign(&mut self, rhs: &Int) {
+        self.0 %= rhs.0;
+    }
+}
+
+impl ShlAssign<u8> for Int {
+    fn shl_assign(&mut self, &rhs: &u8) {
+        self.0 <<= rhs;
+    }
+}
+
+impl ShlAssign<u16> for Int {
+    fn shl_assign(&mut self, &rhs: &u16) {
+        self.0 <<= rhs;
+    }
+}
+
+impl ShrAssign<u8> for Int {
+    fn shr_assign(&mut self, &rhs: &u8) {
+        self.0 >>= rhs;
+    }
+}
+
+impl ShrAssign<u16> for Int {
+    fn shr_assign(&mut self, &rhs: &u16) {
+        self.0 >>= rhs;
+    }
+}
+
+impl SubAssign for Int {
+    fn sub_assign(&mut self, rhs: &Int) {
+        self.0 -= rhs.0;
+    }
+}