Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix kernel message deserialization #14721

Merged
merged 3 commits into from
Jun 22, 2023

Conversation

davidbrochart
Copy link
Contributor

References

Closes #14720

Code changes

Prior to this change, an empty buffer could be inserted.
This PR checks if the buffer is empty before inserting it.

User-facing changes

None (but fixes a bug).

Backwards-incompatible changes

None.

@jupyterlab-probot
Copy link

Thanks for making a pull request to jupyterlab!
To try out this branch on binder, follow this link: Binder

Comment on lines 79 to 81
buffer = new DataView(binMsg.slice(offsets[i]));
} else {
buffers.push(new DataView(binMsg.slice(offsets[i], offsets[i + 1])));
buffer = new DataView(binMsg.slice(offsets[i], offsets[i + 1]));
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is it worth not even creating the DataView? We could compare the last offset with the message length instead.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

That's what I did in 110e530.

buffers.push(new DataView(binMsg.slice(offsets[i], offsets[i + 1])));
buffer = new DataView(binMsg.slice(offsets[i], offsets[i + 1]));
}
if (buffer.byteLength > 0) {

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is this safe? Seems like you could in fact push empty arrays (for whatever reason) and those should be deserialized correctly. In Panel I tried a test case where I pushed an empty array and that (correctly) got deserialized as a zero-length DataView (which appeared in addition to the bogus additional DataView).

Copy link

@philippjfr philippjfr Jun 20, 2023

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Here's what I mean, this is a message where I've sent an empty array:

Screen Shot 2023-06-20 at 14 32 49

You can see that we get two DataViews with length 0, one being the real empty array and then the bogus empty DataView that is always added. Dropping both would be incorrect.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes I can see a need for inserting empty buffers. Could you try this PR with your Panel example?

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Never built JupyterLab myself so give me a few minutes.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Basically, after checking out this branch:

micromamba install -c conda-forge python nodejs
pip install -e .

Then:

jupyter lab --dev-mode --extensions-in-dev-mode

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Maybe run jlpm run build?

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

... and jlpm install to install the npm deps.

https://jupyterlab.readthedocs.io/en/latest/developer/contributing.html#setting-up-a-local-development-environment should contain the ordered list of commands to run, which is documented on https://jupyterlab.readthedocs.io/en/latest/developer/contributing.html#installing-jupyterlab

git clone https://github.com/<your-github-username>/jupyterlab.git
cd jupyterlab
pip install -e ".[dev,test]"
jlpm install
jlpm run build  # Build the dev mode assets (optional)

I haven't tried those simple commands in a fresh env (cc/ @ericsnekbytes)

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actually, according to this comment:

offsetN == len(binMsg) (i.e., the last offset is the length of the file, so that the last buffer is binMsg[offset[N-1], offset[N]]?

I don't think we need to process the last offset at all. That's what I did in ce3c98a.
Maybe @jasongrout can confirm?

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

That was my feeling too, it seems like the simplest fix. In all the messages I've manually stepped through it also would have produced the correct result. For example in my screenshot above where I sent an empty array you can see that the accumulated offsets ends with ..., 547, 547] i.e. it correctly records the offset for the empty array and then adds an extra one that should not be processed.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Maybe @jasongrout can confirm?

I think that is right (see also jupyter-server/jupyter_server#657 (comment)). IIRC, the reason we made the last offset the length of the message is so that the decoding would be very simple, and it provides a sanity check on the message size, and lets us possibly include content after the last buffer in the future.

Copy link
Contributor

@jasongrout jasongrout left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. Yay for simpler code! Thanks!

Copy link
Member

@fcollonval fcollonval left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks @davidbrochart

CI failures are not related

@fcollonval fcollonval merged commit 8428835 into jupyterlab:main Jun 22, 2023
@fcollonval
Copy link
Member

@meeseeksdev please backport to 4.0.x

meeseeksmachine pushed a commit to meeseeksmachine/jupyterlab that referenced this pull request Jun 22, 2023
fcollonval pushed a commit that referenced this pull request Jun 22, 2023
Co-authored-by: David Brochart <david.brochart@gmail.com>
@github-actions github-actions bot locked as resolved and limited conversation to collaborators Jun 22, 2024
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

All comm messages contain empty DataView in buffers
6 participants