def getGrid2D(self, idx):
""" generated source for method getGrid2D """
if idx < 0 or idx >= shape[0]:
raise IllegalArgumentException("index out of range")
itemSize = shape[1] * shape[2] * dataType.getSize()
pos = idx * itemSize
newBuffer = buffer_.slice_()
newBuffer.position(pos)
newBuffer.limit(pos + itemSize)
newBuffer = newBuffer.slice_()
return Grid2D(newBuffer, dataType,
Arrays.copyOfRange(origin, 1, origin.length),
Arrays.copyOfRange(shape, 1, shape.length))
def replace_header_name(callbacks, rule, request):
"""
Finds header name and replaces it with a new value, provided by the user.
Args:
callbacks: the burp callbacks object.
rule: the rule that we should apply.
request: a byte[] object that should be modified.
Returns:
tuple: the first element is a boolean that indicates whether the request was modified and the second one is the modified request or None if we didn't modify the request.
"""
helpers = callbacks.helpers
analyzedRequest = helpers.analyzeRequest(request)
headers = analyzedRequest.headers
modified = False
newHeaders = ArrayList()
for header in headers:
splat = header.split(":")
if len(splat) >= 2:
name, value = splat[0], splat[1]
else:
newHeaders.add(header)
continue # First line of header doesn't have ":"
if name.lower().strip() == rule.search.lower().strip():
newHeaders.add("%s: %s" % (name, rule.replacement))
modified = True
else:
newHeaders.add(header)
if modified:
body = Arrays.copyOfRange(request, analyzedRequest.bodyOffset,
len(request))
return modified, helpers.buildHttpMessage(newHeaders, body)
else:
return modified, None
