diff --git a/src/main/javassist/expr/NewExpr.java b/src/main/javassist/expr/NewExpr.java index 3171fc3f..ee065167 100644 --- a/src/main/javassist/expr/NewExpr.java +++ b/src/main/javassist/expr/NewExpr.java @@ -193,6 +193,14 @@ public void replace(String statement) throws CannotCompileException { */ int codeSize = canReplace(); int end = pos + codeSize; + //check isStoreBeforeInit ,such as : new xx/xx ; dup;[astoreN];invokespecial xx/xx; + int beforeStoreOp = 0; + int preOp = iterator.byteAt(currentPos - 1); + if (iterator.byteAt(newPos + 3) == Opcode.DUP + && (preOp >= Opcode.ASTORE_0 + && preOp <= Opcode.ASTORE_3) && currentPos - newPos == 5) { + beforeStoreOp = preOp; + } for (int i = pos; i < end; ++i) iterator.writeByte(NOP, i); @@ -230,7 +238,12 @@ public void replace(String statement) throws CannotCompileException { if (codeSize > 3) // if the original code includes DUP. bytecode.addAload(retVar); - replace0(pos, bytecode, bytecodeSize); + if (beforeStoreOp >= Opcode.ASTORE_0) { + bytecode.addOpcode(beforeStoreOp); + replace0(pos - 1, bytecode, bytecodeSize + 1); + } else { + replace0(pos, bytecode, bytecodeSize); + } } catch (CompileError e) { throw new CannotCompileException(e); } catch (NotFoundException e) { throw new CannotCompileException(e); }