You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@zookeeper.apache.org by rg...@apache.org on 2015/11/08 22:36:17 UTC
svn commit: r1713300 - in /zookeeper/trunk: CHANGES.txt
src/java/main/org/apache/zookeeper/server/NIOServerCnxn.java
src/java/main/org/apache/zookeeper/server/NettyServerCnxn.java
src/java/test/org/apache/zookeeper/test/FourLetterWordsTest.java
Author: rgs
Date: Sun Nov 8 21:36:17 2015
New Revision: 1713300
URL: http://svn.apache.org/viewvc?rev=1713300&view=rev
Log:
ZOOKEEPER-2227: stmk four-letter word fails execution at server while
reading trace mask argument (Chris Nauroth via rgs)
Modified:
zookeeper/trunk/CHANGES.txt
zookeeper/trunk/src/java/main/org/apache/zookeeper/server/NIOServerCnxn.java
zookeeper/trunk/src/java/main/org/apache/zookeeper/server/NettyServerCnxn.java
zookeeper/trunk/src/java/test/org/apache/zookeeper/test/FourLetterWordsTest.java
Modified: zookeeper/trunk/CHANGES.txt
URL: http://svn.apache.org/viewvc/zookeeper/trunk/CHANGES.txt?rev=1713300&r1=1713299&r2=1713300&view=diff
==============================================================================
--- zookeeper/trunk/CHANGES.txt (original)
+++ zookeeper/trunk/CHANGES.txt Sun Nov 8 21:36:17 2015
@@ -202,6 +202,9 @@ BUGFIXES:
ZOOKEEPER-2211: PurgeTxnLog does not correctly purge when snapshots and
logs are at different locations (Arshad Mohammad via rgs)
+ ZOOKEEPER-2227: stmk four-letter word fails execution at server while reading
+ trace mask argument (Chris Nauroth via rgs)
+
IMPROVEMENTS:
ZOOKEEPER-1660 Documentation for Dynamic Reconfiguration (Reed Wanderman-Milne via shralex)
Modified: zookeeper/trunk/src/java/main/org/apache/zookeeper/server/NIOServerCnxn.java
URL: http://svn.apache.org/viewvc/zookeeper/trunk/src/java/main/org/apache/zookeeper/server/NIOServerCnxn.java?rev=1713300&r1=1713299&r2=1713300&view=diff
==============================================================================
--- zookeeper/trunk/src/java/main/org/apache/zookeeper/server/NIOServerCnxn.java (original)
+++ zookeeper/trunk/src/java/main/org/apache/zookeeper/server/NIOServerCnxn.java Sun Nov 8 21:36:17 2015
@@ -506,6 +506,7 @@ public class NIOServerCnxn extends Serve
final PrintWriter pwriter = new PrintWriter(
new BufferedWriter(new SendBufferWriter()));
if (len == FourLetterCommands.setTraceMaskCmd) {
+ incomingBuffer = ByteBuffer.allocate(8);
int rc = sock.read(incomingBuffer);
if (rc < 0) {
throw new IOException("Read error");
Modified: zookeeper/trunk/src/java/main/org/apache/zookeeper/server/NettyServerCnxn.java
URL: http://svn.apache.org/viewvc/zookeeper/trunk/src/java/main/org/apache/zookeeper/server/NettyServerCnxn.java?rev=1713300&r1=1713299&r2=1713300&view=diff
==============================================================================
--- zookeeper/trunk/src/java/main/org/apache/zookeeper/server/NettyServerCnxn.java (original)
+++ zookeeper/trunk/src/java/main/org/apache/zookeeper/server/NettyServerCnxn.java Sun Nov 8 21:36:17 2015
@@ -279,10 +279,9 @@ public class NettyServerCnxn extends Ser
final PrintWriter pwriter = new PrintWriter(
new BufferedWriter(new SendBufferWriter()));
if (len == FourLetterCommands.setTraceMaskCmd) {
- ByteBuffer mask = ByteBuffer.allocate(4);
+ ByteBuffer mask = ByteBuffer.allocate(8);
message.readBytes(mask);
-
- bb.flip();
+ mask.flip();
long traceMask = mask.getLong();
ZooTrace.setTextTraceLevel(traceMask);
SetTraceMaskCommand setMask = new SetTraceMaskCommand(pwriter, this, traceMask);
Modified: zookeeper/trunk/src/java/test/org/apache/zookeeper/test/FourLetterWordsTest.java
URL: http://svn.apache.org/viewvc/zookeeper/trunk/src/java/test/org/apache/zookeeper/test/FourLetterWordsTest.java?rev=1713300&r1=1713299&r2=1713300&view=diff
==============================================================================
--- zookeeper/trunk/src/java/test/org/apache/zookeeper/test/FourLetterWordsTest.java (original)
+++ zookeeper/trunk/src/java/test/org/apache/zookeeper/test/FourLetterWordsTest.java Sun Nov 8 21:36:17 2015
@@ -19,18 +19,23 @@
package org.apache.zookeeper.test;
import java.io.BufferedReader;
+import java.io.ByteArrayOutputStream;
+import java.io.DataOutputStream;
import java.io.IOException;
import java.io.StringReader;
import java.util.regex.Pattern;
import org.apache.zookeeper.TestableZooKeeper;
import org.apache.zookeeper.ZooKeeper;
+import org.apache.zookeeper.common.IOUtils;
import org.apache.zookeeper.common.X509Exception.SSLContextException;
import static org.apache.zookeeper.client.FourLetterWordMain.send4LetterWord;
import org.junit.Assert;
+import org.junit.Rule;
import org.junit.Test;
+import org.junit.rules.Timeout;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
@@ -38,6 +43,9 @@ public class FourLetterWordsTest extends
protected static final Logger LOG =
LoggerFactory.getLogger(FourLetterWordsTest.class);
+ @Rule
+ public Timeout timeout = new Timeout(30000);
+
/** Test the various four letter words */
@Test
public void testFourLetterWords() throws Exception {
@@ -195,4 +203,43 @@ public class FourLetterWordsTest extends
String resp = sendRequest("isro", 2000);
Assert.assertTrue(resp.contains("rw"));
}
+
+ @Test
+ public void testSetTraceMask() throws Exception {
+ String gtmkResp = sendRequest("gtmk");
+ Assert.assertNotNull(gtmkResp);
+ gtmkResp = gtmkResp.trim();
+ Assert.assertFalse(gtmkResp.isEmpty());
+ long formerMask = Long.valueOf(gtmkResp);
+ try {
+ verify(buildSetTraceMaskRequest(0), "0");
+ verify("gtmk", "0");
+ } finally {
+ // Restore former value.
+ sendRequest(buildSetTraceMaskRequest(formerMask));
+ }
+ }
+
+ /**
+ * Builds a SetTraceMask request to be sent to the server, consisting of
+ * "stmk" followed by the 8-byte long representation of the trace mask.
+ *
+ * @param mask trace mask to set
+ * @return built request
+ * @throws IOException if there is an I/O error
+ */
+ private String buildSetTraceMaskRequest(long mask) throws IOException {
+ ByteArrayOutputStream baos = null;
+ DataOutputStream dos = null;
+ try {
+ baos = new ByteArrayOutputStream();
+ dos = new DataOutputStream(baos);
+ dos.writeBytes("stmk");
+ dos.writeLong(mask);
+ } finally {
+ IOUtils.closeStream(dos);
+ IOUtils.closeStream(baos);
+ }
+ return new String(baos.toByteArray());
+ }
}